We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 20774d7 commit 4c13d5eCopy full SHA for 4c13d5e
backend/app/database.py
@@ -1,14 +1,15 @@
1
from sqlalchemy import create_engine
2
from sqlalchemy.orm import sessionmaker, declarative_base
3
-import os
4
-from dotenv import load_dotenv
5
6
-load_dotenv()
+SQLALCHEMY_DATABASE_URL = "sqlite:///./test.db"
7
8
-DATABASE_URL = os.getenv("DATABASE_URL", "sqlite:///./app.db")
9
-
10
-engine = create_engine(DATABASE_URL, pool_pre_ping=True)
+engine = create_engine(SQLALCHEMY_DATABASE_URL)
11
SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
12
13
-# Global Base for ALL models
14
Base = declarative_base()
+
+def get_db():
+ db = SessionLocal()
+ try:
+ yield db
+ finally:
15
+ db.close()
0 commit comments