backup worked

backend/src/core/database.py

@@ -12,6 +12,8 @@
 from sqlalchemy import create_engine
 from sqlalchemy.orm import sessionmaker, Session
 from backend.src.models.mapping import Base
+# Import TaskRecord to ensure it's registered with Base
+from backend.src.models.task import TaskRecord
 import os
 # [/SECTION]
 
@@ -19,18 +21,31 @@ import os
 DATABASE_URL = os.getenv("DATABASE_URL", "sqlite:///./mappings.db")
 # [/DEF:DATABASE_URL]
 
+# [DEF:TASKS_DATABASE_URL:Constant]
+TASKS_DATABASE_URL = os.getenv("TASKS_DATABASE_URL", "sqlite:///./tasks.db")
+# [/DEF:TASKS_DATABASE_URL]
+
 # [DEF:engine:Variable]
 engine = create_engine(DATABASE_URL, connect_args={"check_same_thread": False})
 # [/DEF:engine]
 
+# [DEF:tasks_engine:Variable]
+tasks_engine = create_engine(TASKS_DATABASE_URL, connect_args={"check_same_thread": False})
+# [/DEF:tasks_engine]
+
 # [DEF:SessionLocal:Class]
 SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
 # [/DEF:SessionLocal]
 
+# [DEF:TasksSessionLocal:Class]
+TasksSessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=tasks_engine)
+# [/DEF:TasksSessionLocal]
+
 # [DEF:init_db:Function]
 # @PURPOSE: Initializes the database by creating all tables.
 def init_db():
     Base.metadata.create_all(bind=engine)
+    Base.metadata.create_all(bind=tasks_engine)
 # [/DEF:init_db]
 
 # [DEF:get_db:Function]
@@ -45,4 +60,16 @@ def get_db():
         db.close()
 # [/DEF:get_db]
 
+# [DEF:get_tasks_db:Function]
+# @PURPOSE: Dependency for getting a tasks database session.
+# @POST:    Session is closed after use.
+# @RETURN:  Generator[Session, None, None]
+def get_tasks_db():
+    db = TasksSessionLocal()
+    try:
+        yield db
+    finally:
+        db.close()
+# [/DEF:get_tasks_db]
+
 # [/DEF:backend.src.core.database]