X-Git-Url: https://apps.delaporte.us/gitweb/matrix-goal-bot.git/blobdiff_plain/bb2ab70af3a361241b656903691e4c9828ac376c..a85dc265c4ca559adc5a773b56cdb76ce3eb5f16:/matrix_goal_bot/db.py diff --git a/matrix_goal_bot/db.py b/matrix_goal_bot/db.py index 0041222..405893a 100644 --- a/matrix_goal_bot/db.py +++ b/matrix_goal_bot/db.py @@ -4,10 +4,6 @@ from datetime import datetime as dt from sqlmodel import Field, Session, SQLModel, create_engine, select, text, Relationship -## Todo List -# - [ ] goal / vision / task - -DB_FILE = os.path.join(os.environ.get('MATRIX_DATA_DIR'), 'goals.db') DATE_FORMAT = '%Y-%m-%d' @@ -16,7 +12,7 @@ def today(): logger = logging.getLogger(__name__) -logging.basicConfig(filename='chore.log', encoding='utf-8', level=logging.DEBUG) +logging.basicConfig(filename='goals.log', encoding='utf-8', level=logging.DEBUG) class Vision(SQLModel, table=True): @@ -64,7 +60,10 @@ def pretty_print(goal, tasks): class Data(): - def __init__(self, db_file = DB_FILE): + def __init__(self, db_file=None): + if not db_file: + db_file = os.path.join(os.environ.get('MATRIX_DATA_DIR', ''), 'goals.db') + self.file = db_file self.engine = self.get_engine() @@ -180,15 +179,3 @@ class Data(): # Someday: Notes on each - def list(self): - results = [] - with Session(self.engine) as session: - cmd = select(Chore) - chores = session.exec(cmd).all() - if(not chores): - retuls.append('None Found') - - for chore in chores: - results.append(Data.check_chore(session, chore.id)) - return results -