21 lines
336 B
Python
21 lines
336 B
Python
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())
|
|
|