fix: [expansion:virus_total] Capture ForbiddenError from vr.APIErro…

…r correctly
MISP · Sep 3, 2024 · 28390ac · 28390ac
1 parent 55b0914
commit 28390ac
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/misp_modules/modules/expansion/virustotal.py b/misp_modules/modules/expansion/virustotal.py
@@ -286,7 +286,10 @@ def dict_handler(request: dict):
         parser = VirusTotalParser(client, int(event_limit) if event_limit else None)
         parser.query_api(attribute)
     except vt.APIError as ex:
-        misperrors['error'] = ex.message
+        if ex.code == 'ForbiddenError':
+            misperrors['error'] = 'ForbiddenError'
+        else:
+            misperrors['error'] = ex.message
         return misperrors
 
     return parser.get_result()