fix: Stop crash with domain without DNS

2024-01-03 02:05:48 +11:00
parent fa75ea3570
commit a5c0ffc4ba
2 changed files with 9 additions and 6 deletions
--- a/domainLookup.py
+++ b/domainLookup.py
@@ -143,15 +143,18 @@ def niami_info(domain: str):
    response = response.json()
    if response["data"]["owner_tx_data"] is not None:
        output = {
-            "owner": response["data"]["owner_tx_data"]["address"],
-            "dns": response["data"]["dnsData"]
+            "owner": response["data"]["owner_tx_data"]["address"]
        }
    else:
        output = {
-            "owner": None,
-            "dns": []
+            "owner": None
        }

+    if 'dnsData' in response["data"]:
+        output["dns"] = response["data"]["dnsData"]
+    else:
+        output["dns"] = []
+
    transactions = requests.get(f"https://api.niami.io/txs/{domain}")
    if transactions.status_code != 200:
        return False