Add more demo objects to the demo suite

MinBZK · Jun 28, 2024 · 1cf3d25 · 1cf3d25
1 parent d81cb83
commit 1cf3d25
Showing 1 changed file with 16 additions and 17 deletions.
diff --git a/tad/core/db.py b/tad/core/db.py
@@ -45,8 +45,10 @@ def init_db():
         if get_settings().ENVIRONMENT == "demo":
             logger.info("Creating demo data")
             _add_demo_users(session)
-            todo_status = _add_demo_statuses(session)
-            _add_demo_tasks(session, todo_status)
+            _add_demo_statuses(session)
+            todo_status = session.exec(select(Status).where(Status.name == "todo")).first()
+            if todo_status is not None:
+                _add_demo_tasks(session, todo_status)
             session.commit()
     logger.info("Finished initializing database")
 
@@ -58,26 +60,23 @@ def _add_demo_users(session: Session) -> None:
         session.add(user)
 
 
-def _add_demo_tasks(session: Session, status: Status | None) -> None:
-    if status is not None:
-        for index in range(1, 3):
-            task = session.exec(select(Task).where(Task.title == "Example task " + str(index))).first()
-            if not task:
-                task = Task(
-                    title="Example task " + str(index),
-                    description="Example description " + str(index),
-                    sort_order=index,
-                    status_id=todo_status.id,
-                )
-                session.add(task)
+def _add_demo_tasks(session: Session, status: Status) -> None:
+    for index in range(1, 3):
+        task = session.exec(select(Task).where(Task.title == "Example task " + str(index))).first()
+        if not task:
+            task = Task(
+                title="Example task " + str(index),
+                description="Example description " + str(index),
+                sort_order=index,
+                status_id=status.id,
+            )
+            session.add(task)
 
 
-def _add_demo_statuses(session: Session) -> Status | None:
+def _add_demo_statuses(session: Session) -> None:
     for index, status_name in enumerate(["todo", "review", "in_progress", "done"]):
         status = session.exec(select(Status).where(Status.name == status_name)).first()
         if not status:
             status = Status(name=status_name, sort_order=index + 1)
             session.add(status)
             session.commit()
-    todo_status = session.exec(select(Status).where(Status.name == "todo")).first()
-    return todo_status