Merge pull request #24 from Farom/master

use IPv6 DNS Servers
eferdman · Jul 11, 2018 · 9308092 · 9308092
2 parents cedf2e6 + bea6cda
commit 9308092
Showing 1 changed file with 8 additions and 5 deletions.
diff --git a/dehydrated-hook-ddns-tsig.py b/dehydrated-hook-ddns-tsig.py
@@ -152,10 +152,13 @@ def query_NS_record(domain_name):
     for i in range(0, len(name_list)):
         nameservers = []
         try:
-            for ns in [rdata.target.to_unicode()
-                       for rdata in dns.resolver.query('.'.join(name_list[i:]),
-                                                       'NS')]:
-                nameservers += [_.to_text() for _ in dns.resolver.query(ns)]
+            fqdn = '.'.join(name_list[i:])
+            for rdata in dns.resolver.query(fqdn, dns.rdatatype.NS):
+                ns = rdata.target.to_unicode()
+                nsL = [] 
+                nsL.extend([_.to_text() for _ in dns.resolver.query(ns)]) # default type: A
+                nsL.extend([_.to_text() for _ in dns.resolver.query(ns, rdtype=dns.rdatatype.AAAA)])
+                nameservers.append(nsL)
         except (dns.resolver.NoAnswer, dns.resolver.NXDOMAIN) as e:
             continue
         if nameservers:
@@ -197,7 +200,7 @@ def verify_record(domain_name,
                        rtype,
                        rdata if rdata is not None else "*",
                        ns))
-        resolver.nameservers = [ns]
+        resolver.nameservers = ns
         answer = []
         try:
             answer = [_.to_text().strip('"'+"'")