modified config file; create and remove wildcard

config.example.json

@@ -6,9 +6,16 @@
     "username": "powerdns",
     "password": "password"
   },
-  "domain": "local.example.com",
-  "reverseDomain": "1.168.192.in-addr.arpa",
-  "dhcpdFilePath": "/etc/dhcp/dhcpd.conf",
-  "dhcpdReloadCommand": "systemctl restart isc-dhcp-server",
-  "recordTTL": 600
+
+  "dns": {
+    "domain": "local.example.com",
+    "reverseDomain": "1.168.192.in-addr.arpa",
+    "recordTTL": 600,
+    "wildcard": true
+  },
+
+  "dhcpd": {
+    "filePath": "/etc/dhcp/dhcpd.conf",
+    "reloadCommand": "systemctl restart isc-dhcp-server"
+  }
 }

localhostgen.py

@@ -2,6 +2,7 @@
 
 import argparse
 import json
+import os
 from time import time
 import psycopg2
 import sys
@@ -82,19 +83,24 @@ class LocalHostGen:
     def delete_host(self, record):
         cur = self.db_instance.cursor()
         cur.execute("DELETE FROM records WHERE name='%s' RETURNING id" % record)
-        record_id = cur.fetchone()[0]
+        record_id = cur.fetchone()
         cur.close()
         self.db_instance.commit()
-        return record_id
+        if record_id is not None:
+            return record_id[0]
+        return None
 
     def remove_host_dhcp(self, host, dhcpd_file_content):
         return re.sub(" *host +%s +\{[^}]+} *\n" % host, "", dhcpd_file_content)
 
-    def create_host(self, domain, reverse_domain, host, ip, mac, ttl, dhcpd_file_path):
+    def create_host(self, domain, reverse_domain, host, ip, mac, ttl, dhcpd_file_path, wildcard):
         record_full = "%s.%s" % (host, domain)
+        host_wildcard = "*.%s" % host
         reverse_domain_record = self.get_reverse_domain_record(reverse_domain, ip)
 
         self.insert_host(domain, host, ip, ttl, 'A')
+        if wildcard:
+            self.insert_host(domain, host_wildcard, record_full, ttl, 'CNAME')
         self.insert_host(reverse_domain, reverse_domain_record, record_full, ttl, 'PTR')
 
         with open(dhcpd_file_path, "r") as f:
@@ -114,10 +120,12 @@ class LocalHostGen:
 
     def remove_host(self, domain, reverse_domain, host, dhcpd_file_path):
         record_full = "%s.%s" % (host, domain)
+        record_full_wildcard = "*.%s" % record_full
         ip = self.get_record_content(record_full)
         if ip is not None:
             reverse_domain_record = self.get_reverse_domain_record(reverse_domain, ip)
             reverse_domain_record_full = "%s.%s" %(reverse_domain_record, reverse_domain)
+            self.delete_host(record_full_wildcard)
             self.delete_host(record_full)
             self.delete_host(reverse_domain_record_full)
 
@@ -154,7 +162,6 @@ def main():
     host = args.host
     ip = args.ip
     mac = args.mac
-    ttl = config['recordTTL']
 
     local_host_gen = LocalHostGen(config['database'])
     local_host_gen.connect()
@@ -163,11 +170,15 @@ def main():
         if ip is None or mac is None:
             eprint("IP and MAC are required to create a host")
             exit(1)
-        local_host_gen.create_host(config['domain'], config['reverseDomain'], host, ip, mac, ttl, config['dhcpdFilePath'])
+        local_host_gen.create_host(config['dns']['domain'], config['dns']['reverseDomain'], host, ip, mac,
+                                   config['dns']['recordTTL'], config['dhcpd']['filePath'], config['dns']['wildcard'])
     elif args.remove:
-        local_host_gen.remove_host(config['domain'], config['reverseDomain'], host, config['dhcpdFilePath'])
+        local_host_gen.remove_host(config['dns']['domain'], config['dns']['reverseDomain'], host,
+                                   config['dhcpd']['filePath'])
 
     local_host_gen.disconnect()
 
+    os.system(config['dhcpd']['reloadCommand'])
+
 
 main()