fix: [expansion:virus_total] Capture `ForbiddenError` from vr.APIError correctly

2024-09-03 11:13:41 +02:00 · 2024-09-03 11:13:41 +02:00 · 28390ac9ae
parent 55b0914185
commit 28390ac9ae
1 changed files with 4 additions and 1 deletions
--- 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()