diff options
Diffstat (limited to 'database.py')
| -rw-r--r-- | database.py | 20 |
1 files changed, 20 insertions, 0 deletions
diff --git a/database.py b/database.py new file mode 100644 index 0000000..aab1d4f --- /dev/null +++ b/database.py @@ -0,0 +1,20 @@ +from sqlalchemy import create_engine +from sqlalchemy.orm import sessionmaker, declarative_base +from config import settings + +engine = create_engine( + settings.database_url, + connect_args={"check_same_thread": False} +) + +SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine) + +Base = declarative_base() + + +def get_db(): + db = SessionLocal() + try: + yield db + finally: + db.close() |
