protect main

This commit is contained in:
Dima Gerasimov 2019-04-30 12:53:59 +02:00
parent a0916ed6bd
commit 22c06a48e1

View file

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