2nd fix for VT Public module

3 years ago · 036933ea14
parent 610c99ce7b
commit 036933ea14
1 changed files with 4 additions and 3 deletions
--- a/misp_modules/modules/expansion/virustotal_public.py
+++ b/misp_modules/modules/expansion/virustotal_public.py
@ -85,9 +85,10 @@ class DomainQuery(VirusTotalParser):
            whois_object = MISPObject(whois)
            whois_object.add_attribute('text', type='text', value=query_result[whois])
            self.misp_event.add_object(**whois_object)
-		if 'domain_siblings' in query_result['domain_siblings']:
-        	siblings = (self.parse_siblings(domain) for domain in query_result['domain_siblings'])
-        	self.parse_resolutions(query_result['resolutions'], query_result['subdomains'], siblings)
+        if 'domain_siblings' in query_result:
+            siblings = (self.parse_siblings(domain) for domain in query_result['domain_siblings'])
+            if 'subdomains' in query_result:
+                self.parse_resolutions(query_result['resolutions'], query_result['subdomains'], siblings)
        self.parse_urls(query_result)

    def parse_siblings(self, domain):