Initial version

lib/nsupdate.py

@@ -0,0 +1,37 @@
+import dns.update
+import dns.query
+import dns.tsigkeyring
+import ipaddress
+
+config = {}
+
+
+def add_records(records):
+    keyring = dns.tsigkeyring.from_text({config['tsig_name']: config['tsig_key']})
+    for hostname, ip in records.items():
+        print("Adding record for " + hostname + "(" + ip + ")")
+
+        rrtype = "A"
+        address = ipaddress.ip_address(ip)
+        if isinstance(address, ipaddress.IPv4Address):
+            rrtype = "A"
+        if isinstance(address, ipaddress.IPv6Address):
+            rrtype = "AAAA"
+
+        update = dns.update.Update(config['domain'], keyring=keyring)
+        update.add(hostname, 60, rrtype, ip)
+        dns.query.tcp(update, config['nameserver'], timeout=2)
+
+
+def delete_records(records):
+    keyring = dns.tsigkeyring.from_text({config['tsig_name']: config['tsig_key']})
+    for hostname, ip in records.items():
+        print("Deleting record for " + hostname + "(" + ip + ")")
+        update = dns.update.Update(config['domain'], keyring=keyring)
+        update.delete(hostname)
+        dns.query.tcp(update, config['nameserver'], timeout=2)
+
+
+def init(_config):
+    global config
+    config = _config