Make config path an argument

endreszabo · Aug 3, 2017 · 28e0a50 · 28e0a50
1 parent c1ae10f
commit 28e0a50
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/pdns-dynamic-reverse-backend.py b/pdns-dynamic-reverse-backend.py
@@ -1,4 +1,4 @@
-#!/usr/bin/python
+#!/usr/bin/env python
 #
 """
 PowerDNS pipe backend for generating reverse DNS entries and their
@@ -248,5 +248,10 @@ def parse_config(config_path):
 if __name__ == '__main__':
     import sys
 
-    prefixes, rtree = parse_config(CONFIG)
+    if len(sys.argv) > 1:
+        config_path = sys.argv[1]
+    else:
+        config_path = CONFIG
+
+    prefixes, rtree = parse_config(config_path)
     sys.exit(parse(prefixes, rtree, sys.stdin, sys.stdout))