From 0451515d071ec80e7355b23baf061f8fa766c5da Mon Sep 17 00:00:00 2001
From: Benjamin Collet <benjamin@collet.eu>
Date: Sat, 10 Apr 2021 16:30:02 +0200
Subject: [PATCH] Add an option to generate all DNS zone files specified in
 configuration

---
 config.yml.dist | 10 ++++++++++
 exporter.py     | 28 ++++++++++++++++++++++++++--
 2 files changed, 36 insertions(+), 2 deletions(-)

diff --git a/config.yml.dist b/config.yml.dist
index 7d31b45..103e5cb 100644
--- a/config.yml.dist
+++ b/config.yml.dist
@@ -1,3 +1,13 @@
 netbox:
   url:    'https://netbox.local'
   token:  ''
+zones:
+  - type: 'reverse'
+    name: '192.0.2.0/24'
+    file: '/path/to/file/db.192.0.2'
+  - type: 'reverse'
+    name: '2001:db8::/32'
+    file: '/path/to/file/db'2001:db8'
+  - type: 'forward'
+    name: 'example.com'
+    file: '/path/to/file/example.com'
diff --git a/exporter.py b/exporter.py
index 33e3e16..0040e4d 100755
--- a/exporter.py
+++ b/exporter.py
@@ -83,7 +83,13 @@ def ptr(nb, args):
     env = Environment(loader=file_loader)
     template = env.get_template("zonefile.jj2")
     output = template.render(serial=serial,records=records)
-    print(output)
+
+    if not hasattr(args, 'output'):
+        print(output)
+    else:
+        f = open(args.output,'w')
+        f.write(output)
+        f.close()
 
 def dns(nb, args):
     devices_id = []
@@ -162,7 +168,23 @@ def dns(nb, args):
     env = Environment(loader=file_loader)
     template = env.get_template("zonefile.jj2")
     output = template.render(serial=serial,records=records)
-    print(output)
+
+    if not hasattr(args, 'output'):
+        print(output)
+    else:
+        f = open(args.output,'w')
+        f.write(output)
+        f.close()
+
+def autodns(nb, args):
+    if not 'zones' in cfg:
+        return
+
+    for zone in cfg['zones']:
+        if zone['type'] == 'reverse':
+            ptr(nb, argparse.Namespace(prefix=zone['name'],output=zone['file']))
+        elif zone['type'] == 'forward':
+            dns(nb, argparse.Namespace(domain=zone['name'],output=zone['file']))
 
 
 if __name__ == '__main__':
@@ -174,6 +196,8 @@ if __name__ == '__main__':
     parser = argparse.ArgumentParser(description='Netbox API exporter')
     subparsers = parser.add_subparsers(help='Action to perform',dest='action',required=True)
 
+    subparser = subparsers.add_parser('autodns', help='Generate all DNS zone files using configuration')
+
     subparser = subparsers.add_parser('ptr', help='Generate reverse DNS zone file for prefix')
     subparser.add_argument('prefix', type=str, help='Prefix')