14 lines
275 B
Python
14 lines
275 B
Python
from sqlmodel import SQLModel, create_engine, Session
|
|
from .config import DATABASE_URL
|
|
|
|
engine = create_engine(DATABASE_URL, echo=False)
|
|
|
|
|
|
def init_db():
|
|
SQLModel.metadata.create_all(engine)
|
|
|
|
|
|
def get_session():
|
|
with Session(engine) as session:
|
|
yield session
|