from sqlmodel import SQLModel, create_engine, Session
from app.config import get_settings
import os

settings = get_settings()

# Create engine
engine = create_engine(
    settings.database_url,
    echo=True,
    pool_pre_ping=True
)

# Test engine
test_engine = None


def get_test_engine():
    global test_engine
    if test_engine is None:
        test_engine = create_engine(
            settings.test_database_url,
            echo=True,
            pool_pre_ping=True
        )
    return test_engine


def create_db_and_tables():
    """Create all database tables"""
    SQLModel.metadata.create_all(engine)


def get_session():
    """Dependency for getting database session"""
    with Session(engine) as session:
        yield session


def get_test_session():
    """Dependency for getting test database session"""
    test_eng = get_test_engine()
    with Session(test_eng) as session:
        yield session