19 lines
660 B
Python
19 lines
660 B
Python
"""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)
|