ALwrity Version 0.5.0 (Fastapi + React )

backend/services/database.py

@@ -0,0 +1,79 @@
+"""
+Database service for ALwrity backend.
+Handles database connections and sessions.
+"""
+
+import os
+from sqlalchemy import create_engine
+from sqlalchemy.orm import sessionmaker, Session
+from sqlalchemy.exc import SQLAlchemyError
+from loguru import logger
+from typing import Optional
+
+# Import models
+from models.onboarding import Base as OnboardingBase
+from models.seo_analysis import Base as SEOAnalysisBase
+from models.content_planning import Base as ContentPlanningBase
+
+# Database configuration
+DATABASE_URL = os.getenv('DATABASE_URL', 'sqlite:///./alwrity.db')
+
+# Create engine
+engine = create_engine(
+    DATABASE_URL,
+    echo=False,  # Set to True for SQL debugging
+    pool_pre_ping=True,
+    pool_recycle=300,
+)
+
+# Create session factory
+SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
+
+def get_db_session() -> Optional[Session]:
+    """
+    Get a database session.
+    
+    Returns:
+        Database session or None if connection fails
+    """
+    try:
+        db = SessionLocal()
+        return db
+    except SQLAlchemyError as e:
+        logger.error(f"Error creating database session: {str(e)}")
+        return None
+
+def init_database():
+    """
+    Initialize the database by creating all tables.
+    """
+    try:
+        # Create all tables for all models
+        OnboardingBase.metadata.create_all(bind=engine)
+        SEOAnalysisBase.metadata.create_all(bind=engine)
+        ContentPlanningBase.metadata.create_all(bind=engine)
+        logger.info("Database initialized successfully with all models")
+    except SQLAlchemyError as e:
+        logger.error(f"Error initializing database: {str(e)}")
+        raise
+
+def close_database():
+    """
+    Close database connections.
+    """
+    try:
+        engine.dispose()
+        logger.info("Database connections closed")
+    except Exception as e:
+        logger.error(f"Error closing database connections: {str(e)}")
+
+# Database dependency for FastAPI
+def get_db():
+    """
+    Database dependency for FastAPI endpoints.
+    """
+    db = SessionLocal()
+    try:
+        yield db
+    finally:
+        db.close()