dbname

app.py

@@ -11,7 +11,7 @@ from fastapi import FastAPI, Request
 from fastapi.middleware.cors import CORSMiddleware
 from fastapi.responses import JSONResponse
 
-from core.database import init_db
+from core.database import init_db, DB_FILENAME
 from routers import auth, blink, contact, credits, general, gemini, payments
 from services.drive_service import DriveService
 
@@ -35,14 +35,17 @@ async def lifespan(app: FastAPI):
     
     # Check for RESET_DB environment variable
     if os.getenv("RESET_DB", "").lower() == "true":
-        logger.warning("RESET_DB is set to true. Skipping download and clearing local database.")
-        if os.path.exists("blink_data.db"):
-            os.remove("blink_data.db")
+        logger.warning(f"RESET_DB is set to true. Skipping download and clearing local database ({DB_FILENAME}).")
+        if os.path.exists(DB_FILENAME):
+            os.remove(DB_FILENAME)
             logger.info("Local database deleted.")
     else:
-        # Startup: Download DB from Drive
-        logger.info("Startup: Attempting to download database from Google Drive...")
-        drive_service.download_db()
+        # Startup: Download DB from Drive ONLY if local file doesn't exist
+        if not os.path.exists(DB_FILENAME):
+            logger.info("Startup: Local DB missing. Attempting to download from Google Drive...")
+            drive_service.download_db()
+        else:
+            logger.info("Startup: Local DB found. Skipping download to preserve data.")
     
     await init_db()
     logger.info("Database initialized successfully")

core/database.py

@@ -5,8 +5,13 @@ from sqlalchemy.ext.asyncio import create_async_engine, AsyncSession, async_sess
 from sqlalchemy.orm import DeclarativeBase
 import os
 
+# Database configuration
+DB_NAME = os.getenv("DB_NAME", "blink_data")
+ENV = os.getenv("ENV", "prod")
+DB_FILENAME = f"{DB_NAME}_{ENV}.db"
+
 # Database URL - SQLite file in the same directory
-DATABASE_URL = os.getenv("DATABASE_URL", "sqlite+aiosqlite:///./blink_data.db")
+DATABASE_URL = os.getenv("DATABASE_URL", f"sqlite+aiosqlite:///./{DB_FILENAME}")
 
 # Create async engine
 engine = create_async_engine(

services/drive_service.py

@@ -18,8 +18,9 @@ class DriveService:
         'https://www.googleapis.com/auth/gmail.send',
         'https://www.googleapis.com/auth/drive.file'
     ]
+from core.database import DB_FILENAME
     FOLDER_NAME = "apigateway"
-    DB_FILENAME = "blink_data.db"
+    # DB_FILENAME is now imported from core.database
 
     def __init__(self):
         self.creds = None