diff --git a/src/overhaul_schedule/model.py b/src/overhaul_schedule/model.py index afee5bb..707b009 100644 --- a/src/overhaul_schedule/model.py +++ b/src/overhaul_schedule/model.py @@ -13,6 +13,6 @@ class OverhaulSchedule(Base, DefaultMixin): plan_duration = Column(Integer, nullable=True) planned_outage = Column(Integer, nullable=True) actual_shutdown = Column(Integer, nullable=True) - start = Column(DateTime, nullable=True) - finish = Column(DateTime, nullable=True) + start = Column(DateTime(timezone=True)) # This will be TIMESTAMP WITH TIME ZONE + finish = Column(DateTime(timezone=True)) remark = Column(String, nullable=True) diff --git a/src/overhaul_schedule/service.py b/src/overhaul_schedule/service.py index 74ca67b..d4d2a3d 100644 --- a/src/overhaul_schedule/service.py +++ b/src/overhaul_schedule/service.py @@ -25,6 +25,8 @@ async def get_all(*, common): async def create( *, db_session: DbSession, overhaul_job_in: OverhaulScheduleCreate ): + + schedule = OverhaulSchedule(**overhaul_job_in.model_dump()) db_session.add(schedule) await db_session.commit()