protect main
This commit is contained in:
parent
a0916ed6bd
commit
22c06a48e1
1 changed files with 17 additions and 13 deletions
|
@ -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()
|
||||
|
|
Loading…
Add table
Reference in a new issue