diff --git a/app/database.py b/app/database.py new file mode 100644 index 00000000..bfff1a6b --- /dev/null +++ b/app/database.py @@ -0,0 +1,17 @@ +#!/usr/bin/env python3 + +from sqlalchemy import create_engine, Column, String, Integer +from sqlalchemy.ext.declarative import declarative_base +from sqlalchemy.orm import sessionmaker +from .config import settings + +Base = declarative_base() +engine = create_engine(settings.database_url, connect_args={"check_same_thread": False}) +SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine) + +def get_db(): + db = SessionLocal() + try: + yield db + finally: + db.close()