Update storage/media_routers.py

storage/media_routers.py

@@ -13,6 +13,7 @@ router = APIRouter(prefix="/media", tags=["Media Manager"])
 MEDIA_ROOT = Path("/data/media")
 file_manager = FileManager(MEDIA_ROOT)
 HF_TOKEN = os.getenv("HF_TOKEN")
+DB_PATH = Path("/data/db")
 
 def validate_token(token: str):
     """
@@ -25,6 +26,42 @@ def validate_token(token: str):
     if token != HF_TOKEN:
         raise HTTPException(status_code=401, detail="Invalid token")
 
+@router.post("/upload_db_file", tags=["Database Manager"])
+async def upload_db_file(
+    file: UploadFile = File(...),
+    token: str = Query(..., description="Token required for authorization")
+):
+    """
+    Upload a file to the /data/db folder.
+
+    Steps:
+    - Validate the token.
+    - Ensure /data/db exists (create it if missing).
+    - Save the uploaded file inside /data/db using its original filename.
+    - Return a JSON response confirming the upload.
+    """
+    validate_token(token)
+
+    # Crear carpeta /data/db si no existe
+    DB_PATH.mkdir(parents=True, exist_ok=True)
+
+    final_path = DB_PATH / file.filename
+
+    try:
+        # Leer contenido en memoria y guardar
+        file_bytes = await file.read()
+        with open(final_path, "wb") as f:
+            f.write(file_bytes)
+    except Exception as exc:
+        raise HTTPException(status_code=500, detail=f"Failed to save file: {exc}")
+
+    return JSONResponse(
+        status_code=200,
+        content={
+            "status": "ok",
+            "saved_to": str(final_path)
+        }
+    )
 
 @router.delete("/clear_media", tags=["Media Manager"])
 def clear_media(token: str = Query(..., description="Token required for authorization")):