diff --git a/hamstertools/__init__.py b/hamstertools/__init__.py index bcaa4b0..3d5672f 100755 --- a/hamstertools/__init__.py +++ b/hamstertools/__init__.py @@ -1,5 +1,6 @@ #!/usr/bin/env python3.7 import csv +import logging from datetime import datetime from itertools import chain from pathlib import Path @@ -8,6 +9,7 @@ import sys import click import requests from peewee import fn, JOIN +from textual.logging import TextualHandler from .db import ( db, @@ -29,8 +31,15 @@ db.init(HAMSTER_FILE) @click.group() -def cli(): - pass +@click.option("-d", "--debug", is_flag=True) +def cli(debug): + if debug: + peewee_logger = logging.getLogger("peewee") + peewee_logger.addHandler(TextualHandler()) + peewee_logger.setLevel(logging.DEBUG) + requests_log = logging.getLogger("requests.packages.urllib3") + requests_log.setLevel(logging.DEBUG) + requests_log.propagate = True @cli.group() diff --git a/hamstertools/db.py b/hamstertools/db.py index c22c594..f54c696 100644 --- a/hamstertools/db.py +++ b/hamstertools/db.py @@ -1,5 +1,4 @@ from datetime import datetime -import logging from peewee import ( SqliteDatabase, Model, @@ -10,11 +9,6 @@ from peewee import ( BooleanField, ) -from textual.logging import TextualHandler - -# logger = logging.getLogger("peewee") -# logger.addHandler(TextualHandler()) -# logger.setLevel(logging.DEBUG) db = SqliteDatabase(None) diff --git a/scripts/apitest.py b/scripts/apitest.py index 98bc63f..99624e4 100644 --- a/scripts/apitest.py +++ b/scripts/apitest.py @@ -4,16 +4,9 @@ import sys sys.path.append(os.path.join(os.path.dirname(__file__), "..")) from datetime import datetime, timedelta -import logging from hamstertools.kimai import KimaiAPI, Timesheet, Project, Activity -logging.basicConfig() -logging.getLogger().setLevel(logging.DEBUG) -requests_log = logging.getLogger("requests.packages.urllib3") -requests_log.setLevel(logging.DEBUG) -requests_log.propagate = True - api = KimaiAPI() # print(Timesheet.list(api))