Add project selection option

alembic/versions/813a2379f36d_add_project_column_to_time.py

@@ -0,0 +1,29 @@
+"""Add project column to time.
+
+Revision ID: 813a2379f36d
+Revises: 348acf3ce3c3
+Create Date: 2025-10-25 07:49:49.562787
+
+"""
+
+from collections.abc import Sequence
+
+import sqlalchemy as sa
+
+from alembic import op
+
+# revision identifiers, used by Alembic.
+revision: str = "813a2379f36d"
+down_revision: str | Sequence[str] | None = "348acf3ce3c3"
+branch_labels: str | Sequence[str] | None = None
+depends_on: str | Sequence[str] | None = None
+
+
+def upgrade() -> None:
+    """Upgrade schema."""
+    op.add_column("Events", sa.Column("Project", sa.String(), nullable=True))
+
+
+def downgrade() -> None:
+    """Downgrade schema."""
+    op.drop_column("Events", "Project")

scripts/insert_working_days.py

@@ -41,17 +41,17 @@ def main() -> None:
             end_minute = random.randint(0, 59)
             end_dt = datetime.datetime.combine(current, datetime.time(end_hour, end_minute))
             # Insert start event
-            DB_CONTROLLER.add_event("start", start_dt)
+            DB_CONTROLLER.add_event("start", start_dt, "default")
             # 40% chance for a break from 12:00 to 13:00
             pause_chance = 0.8
             if random.random() < pause_chance:
                 pause_start = datetime.datetime.combine(current, datetime.time(12, 0))
                 pause_end = datetime.datetime.combine(current, datetime.time(13, 0))
-                DB_CONTROLLER.add_event("stop", pause_start)
-                DB_CONTROLLER.add_event("start", pause_end)
+                DB_CONTROLLER.add_event("stop", pause_start, "default")
+                DB_CONTROLLER.add_event("start", pause_end, "default")
                 # Add pause (60 minutes)
             # Insert end event
-            DB_CONTROLLER.add_event("stop", end_dt)
+            DB_CONTROLLER.add_event("stop", end_dt, "default")
         current += datetime.timedelta(days=1)