Fixed an issue when location response was empty

TheHive-Project · Apr 25, 2019 · a271710 · a271710
1 parent a1b37f1
commit a271710
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/analyzers/TalosReputation/TalosReputation.py b/analyzers/TalosReputation/TalosReputation.py
@@ -57,7 +57,7 @@ def run(self):
                 if response_details.status_code == 200 | 201:
                     if response_location.status_code == 200 | 201:
                         result = response_details.json()
-                        result['country'] = response_location.json()['country']
+                        result['country'] = response_location.json().get('country', None)
                         self.report(result if len(result) > 0 else {})
                     else:
                         self.error('Failed to query Talos location. Status_code {}'.format(response_location.status_code))