Minor Change

Author: dfirsec

utils/main_worker.py

@@ -447,16 +447,15 @@ def whois_ip(ip_addr):
             # ref: https://ipwhois.readthedocs.io/en/latest/RDAP.html
             obj = IPWhois(ip_addr)
             results = obj.lookup_rdap(depth=1)
+            if results["asn_description"] and "NA" not in results["asn_description"]:
+                contact = results["asn_description"]
         except (exceptions.ASNRegistryError, exceptions.WhoisLookupError):
             return "No results"
-        except Exception:
+        except KeyError:
             return None
         else:
             entity = results["entities"][0]
-            if results["asn_description"] and "NA" not in results["asn_description"]:
-                contact = results["asn_description"]
-            else:
-                contact = results["objects"][entity]["contact"]["address"][0]["value"].replace("\r\n", ", ")
+            contact = results["objects"][entity]["contact"]["address"][0]["value"].replace("\r\n", ", ")
             return contact.replace("\n", ", ")
 
     @staticmethod