X-Git-Url: https://us-delaporte.pagekite.me/gitweb/matrix-goal-bot.git/blobdiff_plain/b1558c1522430ae420563c31cb67f9f65f7a375b..e653a07c23fa77ce2d47e73d3bdd3a72fa1ea57e:/matrix_goal_bot/db.py diff --git a/matrix_goal_bot/db.py b/matrix_goal_bot/db.py index e232cd9..8301733 100644 --- a/matrix_goal_bot/db.py +++ b/matrix_goal_bot/db.py @@ -4,9 +4,6 @@ from datetime import datetime as dt from sqlmodel import Field, Session, SQLModel, create_engine, select, text, Relationship -## Todo List -# - [ ] goal / vision / task - DATE_FORMAT = '%Y-%m-%d' @@ -15,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): @@ -27,6 +24,10 @@ class Vision(SQLModel, table=True): return f'{self.name} (Vision {self.id})' +class PreferGoal(SQLModel, table=True): + goal_id: int | None = Field(default=None, foreign_key="goal.id", primary_key=True) + + class Goal(SQLModel, table=True): id: int | None = Field(default=None, primary_key=True) name: str = Field(default=None, unique=True) @@ -36,7 +37,11 @@ class Goal(SQLModel, table=True): tasks: list["Task"] = Relationship(back_populates="goal") vision: Vision = Relationship(back_populates="goals") def __str__(self): - return f'{self.name} (Goal {self.id})' + if not self.complete_date: + return f'{self.name} (Goal {self.id})' + else: + completed = self.complete_date.strftime(DATE_FORMAT) + return f'(Done {completed}) {self.name} (Goal {self.id})' class Task(SQLModel, table=True): id: int | None = Field(default=None, primary_key=True) @@ -65,7 +70,7 @@ def pretty_print(goal, tasks): class Data(): def __init__(self, db_file=None): if not db_file: - db_file = os.path.join(os.environ.get('MATRIX_DATA_DIR'), 'goals.db') + db_file = os.path.join(os.environ.get('MATRIX_DATA_DIR', ''), 'goals.db') self.file = db_file self.engine = self.get_engine() @@ -79,14 +84,28 @@ class Data(): return 'Success' + def _set_pref_goal_id(self, goal_id): + with Session(self.engine) as session: + prefs = session.exec(select(PreferGoal)).all() + for pref in prefs: + session.delete(pref) + pref = PreferGoal() + pref.goal_id = goal_id + session.add(pref) + session.commit() + + def _get_pref_goal_id(self): + with Session(self.engine) as session: + pref = session.exec(select(PreferGoal)).one() + return pref.goal_id + + + def get_all_visions(self): with Session(self.engine) as session: cmd = select(Vision).where(Vision.complete_date == None) return session.exec(cmd).all() - def print_ophan_tasks(self): - pass - def get_all_goals(self): '''List all goals''' with Session(self.engine) as session: @@ -95,15 +114,16 @@ class Data(): def get_done_goals(self): - '''List all goals''' + '''List completed goals''' with Session(self.engine) as session: cmd = select(Goal).where(Goal.complete_date != None) return session.exec(cmd).all() - def get_goal(self, id): + def get_goal(self, goal_id): + self._set_pref_goal_id(goal_id) with Session(self.engine) as session: - cmd = select(Goal).where(Goal.id == id) + cmd = select(Goal).where(Goal.id == goal_id) goal = session.exec(cmd).first() return (goal, goal.tasks) @@ -125,16 +145,29 @@ class Data(): session.commit() - def complete_goal(self, id): + def complete_goal(self, goal_id): + self._set_pref_goal_id(goal_id) vision_id = -1 with Session(self.engine) as session: - cmd = select(Goal).where(Goal.id == id) + cmd = select(Goal).where(Goal.id == goal_id) goal = session.exec(cmd).first() goal.complete_date = dt.today() vision_id = goal.vision_id session.commit() return vision_id + def rename_goal(self, goal_id, name): + self._set_pref_goal_id(goal_id) + vision_id = -1 + with Session(self.engine) as session: + cmd = select(Goal).where(Goal.id == goal_id) + goal = session.exec(cmd).first() + goal.name = name + vision_id = goal.vision_id + session.commit() + return vision_id + + def complete_task(self, id): with Session(self.engine) as session: cmd = select(Task).where(Task.id == id) @@ -142,6 +175,7 @@ class Data(): task.complete_date = dt.today() goal_id = task.goal_id session.commit() + self._set_pref_goal_id(goal_id) return goal_id def add_with_name(self, cls, name): @@ -157,7 +191,13 @@ class Data(): return self.add_with_name(Vision, name) def add_goal(self, name): - return self.add_with_name(Goal, name) + goal_id = self.add_with_name(Goal, name) + self._set_pref_goal_id(goal_id) + return goal_id + + def quick_add_task(self, name): + goal_id = self._get_pref_goal_id() + return self.add_task(goal_id, name) def add_task(self, goal_id, name): result = self.add_with_name(Task, name) @@ -170,27 +210,23 @@ class Data(): def assign_vision(self, goal_id, vision_id): '''Goals may move between visions''' + self._set_pref_goal_id(goal_id) with Session(self.engine) as session: cmd = select(Goal).where(Goal.id == goal_id) goal = session.exec(cmd).first() goal.vision_id = vision_id session.commit() + def visionless(self): + '''Get goals with no vision''' + with Session(self.engine) as session: + cmd = select(Goal).where(Goal.vision_id == None) + return session.exec(cmd).all() + + def add_note(self, note, goal): '''Notes print with the goal.''' pass # 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 -