Преглед на файлове

Use catch-all for all dns query

Son NK преди 5 години
родител
ревизия
fc985d64d9
променени са 1 файла, в които са добавени 3 реда и са изтрити 3 реда
  1. 3 3
      app/dns_utils.py

+ 3 - 3
app/dns_utils.py

@@ -28,7 +28,7 @@ def get_spf_domain(hostname) -> [str]:
     """return all domains listed in *include:*"""
     """return all domains listed in *include:*"""
     try:
     try:
         answers = dns.resolver.query(hostname, "TXT")
         answers = dns.resolver.query(hostname, "TXT")
-    except dns.resolver.NoAnswer:
+    except Exception:
         return []
         return []
 
 
     ret = []
     ret = []
@@ -49,7 +49,7 @@ def get_spf_domain(hostname) -> [str]:
 def get_txt_record(hostname) -> [str]:
 def get_txt_record(hostname) -> [str]:
     try:
     try:
         answers = dns.resolver.query(hostname, "TXT")
         answers = dns.resolver.query(hostname, "TXT")
-    except dns.resolver.NoAnswer:
+    except Exception:
         return []
         return []
 
 
     ret = []
     ret = []
@@ -67,7 +67,7 @@ def get_dkim_record(hostname) -> str:
     """query the dkim._domainkey.{hostname} record and returns its value"""
     """query the dkim._domainkey.{hostname} record and returns its value"""
     try:
     try:
         answers = dns.resolver.query(f"dkim._domainkey.{hostname}", "TXT")
         answers = dns.resolver.query(f"dkim._domainkey.{hostname}", "TXT")
-    except dns.resolver.NoAnswer:
+    except Exception:
         return ""
         return ""
 
 
     ret = []
     ret = []