DB updated for HF
Browse files- api/database.py +4 -1
api/database.py
CHANGED
|
@@ -8,7 +8,10 @@ from dotenv import load_dotenv
|
|
| 8 |
|
| 9 |
load_dotenv()
|
| 10 |
|
| 11 |
-
DATABASE_URL = os.getenv(
|
|
|
|
|
|
|
|
|
|
| 12 |
|
| 13 |
engine = create_engine(
|
| 14 |
DATABASE_URL, connect_args={"check_same_thread": False} if DATABASE_URL.startswith("sqlite") else {}
|
|
|
|
| 8 |
|
| 9 |
load_dotenv()
|
| 10 |
|
| 11 |
+
DATABASE_URL = os.getenv(
|
| 12 |
+
"DATABASE_URL",
|
| 13 |
+
"sqlite:////data/test.db" if os.getenv("SPACE_ID") else "sqlite:///./test.db"
|
| 14 |
+
) if os.getenv("SPACE_ID") else "sqlite:///./test.db"
|
| 15 |
|
| 16 |
engine = create_engine(
|
| 17 |
DATABASE_URL, connect_args={"check_same_thread": False} if DATABASE_URL.startswith("sqlite") else {}
|