refactor how we deal with TTLs

[zonemaker.git] / zone-maker

diff --git a/zone-maker b/zone-maker
index 336a7101123001075b4706c13db32d02a9533318..4a02e17d85ead5a6bd0c7328d5c3b33b124db570 100755 (executable)
--- a/zone-maker
+++ b/zone-maker
@@ -1,14 +1,20 @@
 #!/usr/bin/python3
 import sys, os
 from zonemaker.zone import Zone
 #!/usr/bin/python3
 import sys, os
 from zonemaker.zone import Zone

-import typing
+#import typing

 
 def load_module(name, path, write_bytecode = False):
 
 def load_module(name, path, write_bytecode = False):

-    import importlib.machinery

     old_val = sys.dont_write_bytecode
     sys.dont_write_bytecode = not write_bytecode
     old_val = sys.dont_write_bytecode
     sys.dont_write_bytecode = not write_bytecode

-    module = importlib.machinery.SourceFileLoader(name, path).load_module()
-    sys.dont_write_bytecode = old_val
+    module = None
+    try:
+        from importlib.machinery import SourceFileLoader
+        module = SourceFileLoader(name, path).load_module()
+    except ImportError:
+        import imp
+        module = imp.load_source(name, path)
+    finally:
+        sys.dont_write_bytecode = old_val

     return module
 
 def make_zone(filename: str) -> None:
     return module
 
 def make_zone(filename: str) -> None: