protect main

location/__main__.py

@@ -5,20 +5,24 @@ from location import get_logger, get_locations, iter_locations, get_groups
 
 from kython.klogging import setup_logzero
 
-logger = get_logger()
+def main():
-setup_logzero(logger, level=logging.INFO)
+    logger = get_logger()
+    setup_logzero(logger, level=logging.INFO)
 
 
 
-if len(sys.argv) > 1:
+    if len(sys.argv) > 1:
-    cmd = sys.argv[1]
+        cmd = sys.argv[1]
-    if cmd == "update_cache":
+        if cmd == "update_cache":
-        from location import update_cache, get_locations
+            from location import update_cache, get_locations
-        update_cache()
+            update_cache()
-        get_locations(cached=True)
+            get_locations(cached=True)
+        else:
+            raise RuntimeError(f"Unknown command {cmd}")
     else:
-        raise RuntimeError(f"Unknown command {cmd}")
+        for p in get_groups(cached=True):
-else:
+            print(p)
-    for p in get_groups(cached=True):
+            # TODO need datetime!
-        print(p)
+
-        # TODO need datetime!
+if __name__ == '__main__':
+    main()