27 lines
403 B
Python
27 lines
403 B
Python
from sqlmodel import Session, SQLModel, create_engine
|
|
|
|
from config import Settings
|
|
from model.model import Department, Employee
|
|
|
|
PGSQL = Settings().PGSQL
|
|
|
|
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())
|
|
|
|
|
|
__all__ = [
|
|
"Department",
|
|
"Employee",
|
|
]
|