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=today())
+ start_date: dt | None = Field(default=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)
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:
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 <= today())
- cmd = select(Goal).where(Goal.complete_date == None)
+ cmd = select(Goal).where(Goal.complete_date == None).where(Goal.start_date <= today())
+ # cmd = select(Goal).where(Goal.complete_date == None)
return session.exec(cmd).all()
def get_done_goals(self):
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)