初始化数据库连接

This commit is contained in:
2026-01-14 18:04:09 +08:00
parent 4620e349d9
commit 655f4348c4
4 changed files with 29 additions and 0 deletions

20
model/__init__.py Normal file
View File

@@ -0,0 +1,20 @@
from sqlmodel import Session, SQLModel, create_engine
from config import setting
PGSQL = setting.env.PGSQL or "sqlite:///database.db"
engine = create_engine(str(PGSQL))
def create_db_and_tables():
SQLModel.metadata.create_all(engine)
def get_engine():
return engine
def get_session():
return Session(get_engine())

0
model/model.py Normal file
View File