Spaces:
Runtime error
Runtime error
Aryan Jain commited on
Commit ·
10c4301
1
Parent(s): 84f69d1
add api to delete all data from databse
Browse files- src/app.py +7 -0
src/app.py
CHANGED
|
@@ -12,6 +12,7 @@ from fastapi.responses import RedirectResponse
|
|
| 12 |
from src.config import DatabaseConfig, logger
|
| 13 |
from src.controllers import api_router
|
| 14 |
|
|
|
|
| 15 |
|
| 16 |
|
| 17 |
def run_upgrade(connection, alembic_config: Config):
|
|
@@ -63,5 +64,11 @@ app.add_middleware(
|
|
| 63 |
async def check_health():
|
| 64 |
return RedirectResponse(url="/docs")
|
| 65 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 66 |
|
| 67 |
app.include_router(api_router, prefix="/api/v1")
|
|
|
|
| 12 |
from src.config import DatabaseConfig, logger
|
| 13 |
from src.controllers import api_router
|
| 14 |
|
| 15 |
+
from src.models import Base
|
| 16 |
|
| 17 |
|
| 18 |
def run_upgrade(connection, alembic_config: Config):
|
|
|
|
| 64 |
async def check_health():
|
| 65 |
return RedirectResponse(url="/docs")
|
| 66 |
|
| 67 |
+
@app.delete("/api/v1/databse")
|
| 68 |
+
async def delete_all_data():
|
| 69 |
+
async with DatabaseConfig.get_engine().begin() as session:
|
| 70 |
+
await session.run_sync(Base.metadata.drop_all)
|
| 71 |
+
await session.run_sync(Base.metadata.create_all)
|
| 72 |
+
return {"response": "All data deleted successfully!"}
|
| 73 |
|
| 74 |
app.include_router(api_router, prefix="/api/v1")
|