From 02497e167c47a14c2c11d4f8eb9a7365e17b8a0a Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Rapha=C3=ABl=20Vinot?= <raphael@vinot.info>
Date: Mon, 26 Dec 2022 23:30:00 +0100
Subject: [PATCH] fix: Catch QuotaExceededError in VT module

---
 lookyloo/modules/vt.py | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/lookyloo/modules/vt.py b/lookyloo/modules/vt.py
index 2c3fb5b..0e6aa73 100644
--- a/lookyloo/modules/vt.py
+++ b/lookyloo/modules/vt.py
@@ -81,7 +81,13 @@ class VirusTotal():
 
         scan_requested = False
         if self.autosubmit and force:
-            self.client.scan_url(url)
+            try:
+                self.client.scan_url(url)
+            except APIError as e:
+                if e.code == 'QuotaExceededError':
+                    self.logger.warning('VirusTotal quota exceeded, sry.')
+                    return
+                self.logger.exception('Something went poorly withi this query.')
             scan_requested = True
 
         if not force and vt_file.exists():