working auth + users systems

backend/core/database.py

@@ -0,0 +1,36 @@
+# core/database.py
+from sqlalchemy import create_engine
+from sqlalchemy.orm import sessionmaker, Session, declarative_base
+from typing import Generator
+
+from core.config import settings
+
+Base = declarative_base()  # Used for models
+
+_engine = None
+_SessionLocal = None
+
+def get_engine():
+    global _engine
+    if _engine is None:
+        if not settings.DB_URL:
+            raise ValueError("DB_URL is not set in Settings.")
+        print(f"Connecting to database at {settings.DB_URL}")
+        _engine = create_engine(settings.DB_URL)
+        Base.metadata.create_all(_engine)  # Create tables here
+    return _engine
+
+def get_sessionmaker():
+    global _SessionLocal
+    if _SessionLocal is None:
+        engine = get_engine()
+        _SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
+    return _SessionLocal
+
+def get_db() -> Generator[Session, None, None]:
+    SessionLocal = get_sessionmaker()
+    db = SessionLocal()
+    try:
+        yield db
+    finally:
+        db.close()