X-Git-Url: https://git.ralfj.de/zonemaker.git/blobdiff_plain/8afd5eb407397e88da8f7d2151088d14abbe51e4..aa7f7564c6e419cff1c414dc79ec30e11bb87163:/zone-maker?ds=inline

diff --git a/zone-maker b/zone-maker
index 100b17a..676bf9c 100755
--- a/zone-maker
+++ b/zone-maker
@@ -1,5 +1,7 @@
 #!/usr/bin/python3
 import sys, os
+from zonemaker.zone import Zone
+#import typing
 
 def load_module(name, path, write_bytecode = False):
     import importlib.machinery
@@ -9,12 +11,10 @@ def load_module(name, path, write_bytecode = False):
     sys.dont_write_bytecode = old_val
     return module
 
-def make_zone(filename):
+def make_zone(filename: str) -> None:
     zonefile = load_module(os.path.basename(filename), filename)
-    zones = zonefile.__zones__
-    # TODO do something more clever with the zones
-    for zone in zones:
-        zone.write(sys.stdout)
+    zone = zonefile.__zone__ # type: Zone
+    zone.write()
 
 if __name__ == "__main__":
     for name in sys.argv[1:]: