This commit is contained in:
Waylon Walker 2024-03-01 07:10:57 -06:00
parent 85554e2169
commit a21dbb08d4
No known key found for this signature in database
GPG key ID: 66E2BF2B4190EFE4
8 changed files with 285 additions and 43 deletions

21
sqlmodel_base/database.py Normal file
View file

@ -0,0 +1,21 @@
from functools import lru_cache
from sqlmodel import Field, Session, SQLModel, create_engine, select
sqlite_file_name = "database.db"
sqlite_url = f"sqlite:///{sqlite_file_name}"
@lru_cache
def get_engine():
from sqlmodel_base.hero.models import Hero
from sqlmodel_base.team.models import Team
engine = create_engine(sqlite_url)
SQLModel.metadata.create_all(engine)
return engine
def get_session():
with Session(get_engine()) as session:
yield session