Added inventory_id column for schedule

alembic/versions/4fc231816655_added_inventory_id_column_for_schedule.py

@@ -0,0 +1,28 @@
+"""Added inventory_id column for schedule
+
+Revision ID: 4fc231816655
+Revises: d945da89599c
+Create Date: 2024-01-07 11:27:52.029590
+
+"""
+from alembic import op
+import sqlalchemy as sa
+
+
+# revision identifiers, used by Alembic.
+revision = '4fc231816655'
+down_revision = 'd945da89599c'
+branch_labels = None
+depends_on = None
+
+
+def upgrade() -> None:
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.add_column('schedule', sa.Column('inventory_id', sa.String(), nullable=True))
+    # ### end Alembic commands ###
+
+
+def downgrade() -> None:
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.drop_column('schedule', 'inventory_id')
+    # ### end Alembic commands ###

app/api/schedule/controller.py

@@ -59,6 +59,7 @@ def upsert_user_schedules(db: Session, user_id: str, schedules: list[Schedule]):
         set_={
             "updated_at": datetime.utcnow(),
             "time_to_take": stmt.excluded.time_to_take,
+            "inventory_id": stmt.excluded.inventory_id,
             "schedule_state": stmt.excluded.schedule_state,
             "notes": stmt.excluded.notes,
             "quantity": stmt.excluded.quantity,

app/api/schedule/schemas.py

@@ -7,6 +7,7 @@
 
 class Schedule(BaseModel):
     schedule_id: Optional[str]
+    inventory_id: Optional[str]
     created_at: Optional[datetime]
     updated_at: Optional[datetime]
     deleted_at: Optional[datetime]

app/models.py

@@ -61,6 +61,7 @@ class Schedule(Base):
     __tablename__ = "schedule"
 
     schedule_id = Column(String, unique=True, nullable=False)
+    inventory_id = Column(String, nullable=True)
     user_id = Column(UUID, primary_key=True, nullable=False)
     created_at = Column(DateTime, nullable=True)
     updated_at = Column(DateTime, nullable=True)