"""Simple file to set up the sqlmodel connection to database."""

from sqlmodel import SQLModel, create_engine
from utilities import config
from utilities.logger import agon_logger

from . import models  # noqa: F401 - Need this import for creating the models.

if config.ENVIRONMENT == "prod":
    url = f"mysql+pymysql://{config.SQL_USER}:{config.SQL_PASS}@{config.SQL_HOST}/{config.SQL_DB}"
    agon_logger.info("Connected to prod database.")
else:
    url = "sqlite:///database.db"
    agon_logger.info("Connected to local dev database.")
engine = create_engine(url)

# Don't need to create all when connected to prod.
SQLModel.metadata.create_all(engine)