"""
Database module for the Document Translation API
Provides PostgreSQL/SQLite support with async SQLAlchemy 2.0
"""

from database.connection import (
    get_db,
    get_db_session,
    get_async_session,
    engine,
    AsyncSessionLocal,
    init_db,
    get_engine,
)
from database.models import (
    Base,
    User,
    Translation,
    ApiKey,
    UsageLog,
    PaymentHistory,
    PlanType,
    SubscriptionStatus,
    Glossary,
    GlossaryTerm,
)

__all__ = [
    "get_db",
    "get_db_session",
    "get_async_session",
    "engine",
    "AsyncSessionLocal",
    "init_db",
    "get_engine",
    "Base",
    "User",
    "Translation",
    "ApiKey",
    "UsageLog",
    "PaymentHistory",
    "PlanType",
    "SubscriptionStatus",
    "Glossary",
    "GlossaryTerm",
]
