diff --git a/dakiny_bot/database/db.py b/dakiny_bot/database/db.py new file mode 100644 index 0000000..b97cf18 --- /dev/null +++ b/dakiny_bot/database/db.py @@ -0,0 +1,14 @@ +"""Simple file to set up the sqlmodel connection to database.""" + +from sqlmodel import SQLModel, create_engine + +from dakiny_bot.logger import dakiny_logger + +from . import models # noqa: F401 - Need this import for creating the models. + +url = "sqlite:///database/dakiny.db" +dakiny_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)