Merge pull request #63 from colinodell/fix/remove_key_error

fix: add safe get on dnsInfo to avoid raise TypeError
colinodell · Aug 13, 2021 · 5ea6625 · 5ea6625
2 parents 922cf66 + 981da39
commit 5ea6625
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/qnapstats/qnap_stats.py b/qnapstats/qnap_stats.py
@@ -276,8 +276,10 @@ def get_system_stats(self):
                 "err_packets": int(root["err_packet" + i])
             }
 
-        for dns in root["dnsInfo"]["DNS_LIST"]:
-            details["dns"].append(dns)
+        dnsInfo = root.get("dnsInfo")
+        if dnsInfo:
+            for dns in dnsInfo["DNS_LIST"]:
+                details["dns"].append(dns)
 
         return details