17 lines
371 B
Python
17 lines
371 B
Python
from sqlmodel import SQLModel, create_engine, Session
|
|
from dotenv import load_dotenv
|
|
import os
|
|
|
|
load_dotenv()
|
|
|
|
DATABASE_URL = os.getenv("DATABASE_URL")
|
|
|
|
if not DATABASE_URL:
|
|
raise ValueError("DATABASE_URL environment variable is not set")
|
|
|
|
engine = create_engine(DATABASE_URL, echo=True)
|
|
|
|
def get_session():
|
|
with Session(engine) as session:
|
|
yield session
|