src/database/__init__.py

import logging
from sqlmodel import SQLModel
from src.database.session import engine
from src.database.models import Animes, User

logging.basicConfig(level=logging.INFO)
logger = logging.getLogger(__name__)


def init_db():
    """Connects to PostgreSQL DB and creates all tables defined in SQLModel classes"""
    logger.info("Started table creation....")
    try:
        SQLModel.metadata.create_all(engine)
        logger.info("Database Tables created successfully")
    except Exception as e:
        logger.error(f"Failed to create tables: {e}")