Spaces:
Runtime error
Runtime error
File size: 1,446 Bytes
1b8ee86 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 |
"""Database connection and session management."""
import os
from pathlib import Path
from typing import Generator
from sqlalchemy import create_engine
from sqlalchemy.orm import sessionmaker
from sqlalchemy.ext.declarative import declarative_base
# Support both SQLite (dev) and PostgreSQL/Neon (prod)
_db_path = Path(__file__).parent.parent / "todos.db"
DATABASE_URL = os.getenv(
"DATABASE_URL",
f"sqlite:///{_db_path}"
)
# For SQLite development
if DATABASE_URL.startswith("sqlite"):
engine = create_engine(
DATABASE_URL,
connect_args={"check_same_thread": False},
echo=False
)
else:
# For PostgreSQL/Neon - use psycopg2 or asyncpg
engine = create_engine(
DATABASE_URL,
pool_pre_ping=True,
pool_size=5,
max_overflow=10,
echo=False
)
SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
# Base class for SQLAlchemy tables
Base = declarative_base()
def get_db() -> Generator:
"""Dependency that provides a database session."""
db = SessionLocal()
try:
yield db
finally:
db.close()
def init_db():
"""Initialize database tables."""
# Import models to ensure they're registered with Base
# Import Todo first since User has a relationship to it
from models.todo import Todo # noqa: F401
from models.user import User # noqa: F401
Base.metadata.create_all(bind=engine)
|