From 29e70967f9833c1dc5c3035dbfa091b418d68237 Mon Sep 17 00:00:00 2001 From: Dima Gerasimov Date: Wed, 24 Apr 2019 18:55:52 +0100 Subject: [PATCH] add main --- photos/__main__.py | 37 ++++++++++++++++++++++--------------- 1 file changed, 22 insertions(+), 15 deletions(-) diff --git a/photos/__main__.py b/photos/__main__.py index 09480ba..3bbced2 100644 --- a/photos/__main__.py +++ b/photos/__main__.py @@ -1,25 +1,32 @@ import logging +# TODO eh? logging.basicConfig(level=logging.INFO) from kython.klogging import setup_logzero from photos import get_photos, iter_photos, get_logger -setup_logzero(get_logger(), level=logging.DEBUG) - import sys -if len(sys.argv) > 1: - cmd = sys.argv[1] - if cmd == "update_cache": - from photos import update_cache, get_photos - update_cache() - get_photos(cached=True) + +def main(): + setup_logzero(get_logger(), level=logging.DEBUG) + + if len(sys.argv) > 1: + cmd = sys.argv[1] + if cmd == "update_cache": + from photos import update_cache, get_photos + update_cache() + get_photos(cached=True) + else: + raise RuntimeError(f"Unknown command {cmd}") else: - raise RuntimeError(f"Unknown command {cmd}") -else: - for p in iter_photos(): - print(f"{p.dt} {p.path} {p.tags}") - pass - # TODO need datetime! - # print(p) + for p in iter_photos(): + print(f"{p.dt} {p.path} {p.tags}") + pass + # TODO need datetime! + # print(p) + + +if __name__ == '__main__': + main()