import logging
import os
from datetime import datetime as dt
+from datetime import timedelta
from sqlmodel import Field, Session, SQLModel, create_engine, select, text, Relationship
+logger = logging.getLogger(__name__)
DATE_FORMAT = '%Y-%m-%d'
return dt.now().strftime(DATE_FORMAT)
-logger = logging.getLogger(__name__)
-logging.basicConfig(filename='goals.log', encoding='utf-8', level=logging.DEBUG)
-
-
class Vision(SQLModel, table=True):
id: int | None = Field(default=None, primary_key=True)
name: str = Field(default=None, unique=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)
+ start_date: dt | None = Field(default=dt.today())
complete_date: dt | None = Field(default=None)
vision_id: int | None = Field(default=None, foreign_key="vision.id")
notes: str | None = Field(default=None)
return results
+def _get_with_id(cls, item_id):
+ return select(cls).where(cls.id == item_id)
+
+def _get_with_name(cls, name):
+ return select(cls).where(func.lower(cls.name) == func.lower(name))
+
+
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', ''), 'goal.db')
self.file = db_file
self.engine = self.get_engine()
SQLModel.metadata.create_all(self.engine)
return 'Success'
+ def migrate(self):
+ with Session(self.engine) as session:
+ session.exec(text(''' alter table goal add start_date DATETIME; '''))
+ session.commit()
+ 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)
+ session.commit()
+ 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 clear(self, goal_id, clear_all=False):
+ with Session(self.engine) as session:
+ goal = session.exec(_get_with_id(Goal, goal_id)).first()
+ for item in goal.tasks:
+ if clear_all or item.complete_date:
+ session.delete(item)
+ session.commit()
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:
cmd = select(Goal).where(Goal.complete_date == None)
return session.exec(cmd).all()
+ def get_current_goals(self):
+ '''List current goals'''
+ with Session(self.engine) as session:
+ cmd = select(Goal).where(Goal.complete_date == None).where(Goal.start_date <= dt.today())
+ results = session.exec(cmd).all()
+ logger.debug(f'Current goals: {results}')
+ return results
def get_done_goals(self):
'''List completed goals'''
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)
session.commit()
- def complete_goal(self, id):
+ def complete_goal(self, 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 put_off_goal(self, 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.start_date = dt.today() + timedelta(days=14)
+ vision_id = goal.vision_id
+ session.commit()
+ return vision_id
+
+ def start_goal(self, goal_id):
+ vision = None
+ with Session(self.engine) as session:
+ cmd = select(Goal).where(Goal.id == goal_id)
+ goal = session.exec(cmd).first()
+ goal.start_date = dt.today()
+ vision_id = goal.vision_id
+ session.commit()
+ if vision:
+ return (vision, vision.goals)
+ else:
+ return ('No Vision Assigned', [])
+
+
+ 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)
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):
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 add_task(self, goal_id, name):
- result = self.add_with_name(Task, name)
+ def quick_add_task(self, name):
+ goal_id = self._get_pref_goal_id()
+ task_id = self.add_with_name(Task, name)
with Session(self.engine) as session:
- cmd = select(Task).where(Task.name == name)
+ cmd = select(Task).where(Task.id == task_id)
new = session.exec(cmd).first()
new.goal_id = goal_id
session.commit()
- return result
+ return goal_id
+
+ def remove_task(self, task_id):
+ task_name = ''
+ logger.debug(f'Removing task with id {task_id}')
+ with Session(self.engine) as session:
+ cmd = select(Task).where(Task.id == int(task_id))
+ to_remove = session.exec(cmd).first()
+ if not to_remove:
+ return f'Unable to find task with id {task_id}'
+
+ logger.debug(f'Removing task with name {to_remove.name}')
+ task_name = to_remove.name
+ session.delete(to_remove)
+ session.commit()
+ return f'Removed task "{task_name}"'
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()
projects / matrix-goal-bot.git / blobdiff