CIRCL
/
lookyloo
mirror of https://github.com/CIRCL/lookyloo
2
0
Fork 0
Code Issues Releases Wiki Activity

fix: Exception when dumping WhistleBlowerDict to json

pull/852/head
Raphaël Vinot 2023-12-26 19:44:10 +01:00
parent c04f6566f3
commit a98c1d39ee
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
lookyloo/modules/vt.py
View File

@ -7,6 +7,7 @@ from typing import Any, Dict, Optional, TYPE_CHECKING
import vt # type: ignore import vt # type: ignore
from vt.error import APIError # type: ignore from vt.error import APIError # type: ignore
from vt.object import WhistleBlowerDict # type: ignore
from ..default import ConfigError, get_homedir from ..default import ConfigError, get_homedir
from ..helpers import get_cache_directory from ..helpers import get_cache_directory
@ -17,6 +18,11 @@ if TYPE_CHECKING:
from .abstractmodule import AbstractModule from .abstractmodule import AbstractModule
def jsonify_vt(obj: WhistleBlowerDict):
if isinstance(obj, WhistleBlowerDict):
return {k: v for k, v in obj.items()}
class VirusTotal(AbstractModule): class VirusTotal(AbstractModule):
def module_init(self) -> bool: def module_init(self) -> bool:
@ -92,7 +98,7 @@ class VirusTotal(AbstractModule):
try: try:
url_information = self.client.get_object(f"/urls/{url_id}") url_information = self.client.get_object(f"/urls/{url_id}")
with vt_file.open('w') as _f: with vt_file.open('w') as _f:
json.dump(url_information.to_dict(), _f) json.dump(url_information.to_dict(), _f, default=jsonify_vt)
break break
except APIError as e: except APIError as e:
if not self.autosubmit: if not self.autosubmit: