diff --git a/app/api/prompt/__init__.py b/app/api/prompt/__init__.py index 7ba875d..76637df 100644 --- a/app/api/prompt/__init__.py +++ b/app/api/prompt/__init__.py @@ -1,4 +1,6 @@ """Prompt Routes""" + from .prompt import router as prompt_router from .linkedin import router as linkedin_router +from .drop import router as drop_router diff --git a/app/api/prompt/drop.py b/app/api/prompt/drop.py new file mode 100644 index 0000000..f016e33 --- /dev/null +++ b/app/api/prompt/drop.py @@ -0,0 +1,24 @@ +import os + +from fastapi import APIRouter, Depends, HTTPException + +from app.utils.api_key_auth import get_api_key +from app.utils.db import get_db_connection_direct +from app.utils.make_meta import make_meta + +router = APIRouter() + +# PATCH /prompt/drop: empties the prompt table +@router.patch("/prompt/drop") +def drop_prompt_table(api_key: str = Depends(get_api_key)) -> dict: + """PATCH /prompt/drop: empties the prompt table.""" + try: + conn = get_db_connection_direct() + cur = conn.cursor() + cur.execute("TRUNCATE TABLE prompt RESTART IDENTITY CASCADE;") + conn.commit() + cur.close() + conn.close() + return {"meta": make_meta("success", "Prompt table emptied"), "data": {}} + except Exception as e: + return {"meta": make_meta("error", f"Failed to empty prompt table: {str(e)}"), "data": {}} diff --git a/app/api/routes.py b/app/api/routes.py index 80757f1..c43596a 100644 --- a/app/api/routes.py +++ b/app/api/routes.py @@ -10,6 +10,7 @@ from app.utils.notify.resend import router as resend_router from app.api.prompt.prompt import router as prompt_router from app.api.prompt.linkedin import router as linkedin_router +from app.api.prompt.drop import router as drop_router from app.api.prospects.prospects import router as prospects_router from app.api.orders.orders import router as orders_router @@ -18,5 +19,6 @@ router.include_router(health_router) router.include_router(prompt_router) router.include_router(linkedin_router) +router.include_router(drop_router) router.include_router(prospects_router) router.include_router(orders_router)