diff --git a/app/main.py b/app/main.py index 22ea763..f422a23 100644 --- a/app/main.py +++ b/app/main.py @@ -1,19 +1,24 @@ from fastapi import FastAPI, Depends +from pydantic import BaseModel from app.auth import verify_token from app.scripts import script1, script2 app = FastAPI() +class Payload(BaseModel): + name: str + action: str + @app.get("/") def root(): return {"status": "API running"} @app.post("/script1") -def run_script1(auth=Depends(verify_token)): - result = script1.run() +def run_script1(payload: Payload,auth=Depends(verify_token)): + result = script1.run(payload) return {"result": result} @app.post("/script2") -def run_script2(auth=Depends(verify_token)): - result = script2.run() +def run_script2(payload: Payload, auth=Depends(verify_token)): + result = script2.run(payload) return {"result": result} \ No newline at end of file diff --git a/app/scripts/script1.py b/app/scripts/script1.py index 2d4b19c..78863bc 100644 --- a/app/scripts/script1.py +++ b/app/scripts/script1.py @@ -1,2 +1,5 @@ -def run(): - return "Script1 executed" \ No newline at end of file +def run(data): + name = data.get("name") + action = data.get("action") + + return f"{name} requested {action}" \ No newline at end of file