Aryan Jain commited on
Commit
10c4301
·
1 Parent(s): 84f69d1

add api to delete all data from databse

Browse files
Files changed (1) hide show
  1. 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")