diff --git a/misp_modules/modules/expansion/virustotal_public.py b/misp_modules/modules/expansion/virustotal_public.py index 12c73d5..1183c06 100644 --- a/misp_modules/modules/expansion/virustotal_public.py +++ b/misp_modules/modules/expansion/virustotal_public.py @@ -62,7 +62,7 @@ class VirusTotalParser(): vt_object.add_attribute('detection-ratio', type='text', value=detection_ratio) self.misp_event.add_object(**vt_object) - def query_result(self, query_type): + def get_query_result(self, query_type): params = {query_type: self.attribute.value, 'apikey': self.apikey} return requests.get(self.base_url, params=params) @@ -174,7 +174,7 @@ def handler(q=False): attribute = request['attribute'] query_type, to_call = misp_type_mapping[attribute['type']] parser = to_call(request['config']['apikey'], attribute) - query_result = parser.query_result(query_type) + query_result = parser.get_query_result(query_type) status_code = query_result.status_code if status_code == 200: parser.parse_report(query_result.json())