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

from sqlmodel import SQLModel, create_engine

from utilities import config

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}"
else:
    url = "sqlite:///database.db"
engine = create_engine(url)

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