diff --git a/main.py b/main.py index 64cede8..ce37a59 100644 --- a/main.py +++ b/main.py @@ -1,11 +1,11 @@ import connectors +import argparse import logging import config logging.basicConfig( filename="log", filemode="w", - level=logging.INFO, format="%(levelname)s - %(message)s", ) @@ -198,10 +198,21 @@ if __name__ == "__main__": logging.getLogger("urllib3").setLevel(logging.WARNING) logging.getLogger("httpcore").setLevel(logging.WARNING) + parser = argparse.ArgumentParser() + parser.add_argument("-d", "--dev", action="store_true") + parser.add_argument("-v", "--verbose", action="store_true") + + args = parser.parse_args() + + if args.dev: + abs = connectors.ABSConnectorMock(config.ABS_API_URL, config.ABS_API_TOKEN) + audible = connectors.AudibleConnectorMock(config.AUDIBLE_AUTH_FILE) + audnexus = connectors.AudNexusConnectorMock() + else: + abs = connectors.ABSConnector(config.ABS_API_URL, config.ABS_API_TOKEN) + audible = connectors.AudibleConnector(config.AUDIBLE_AUTH_FILE) + audnexus = connectors.AudNexusConnector() + logger = logging.getLogger(__name__) - - abs = connectors.ABSConnector(config.ABS_API_URL, config.ABS_API_TOKEN) - audible = connectors.AudibleConnector(config.AUDIBLE_AUTH_FILE) - audnexus = connectors.AudNexusConnector() - + logger.setLevel(logging.DEBUG if args.verbose else logging.INFO) main() diff --git a/requirements.txt b/requirements.txt index d6f640e..ae6ebb6 100644 Binary files a/requirements.txt and b/requirements.txt differ