Browse the data

1import secrets
2from fastapi import FastAPI, HTTPException, Header
3from pydantic import BaseModel
4 
5app = FastAPI()
6 
7users = {}
8tokens = {}
9workflows = {}
10workflow_id_counter = 1
11execution_logs = []
12execution_log_id_counter = 1
13 
14def authenticate(authorization: str = Header(None)):
15    if not authorization:
16        raise HTTPException(status_code=401, detail="Missing auth header")
17    token = authorization.replace("Bearer ", "")
18    user_id = tokens.get(token)
19    if not user_id:
20        raise HTTPException(status_code=401, detail="Invalid token")
21    return user_id
22 
23class SignupRequest(BaseModel):
24    username: str
25    password: str
26 
27class LoginRequest(BaseModel):
28    username: str
29    password: str
30 
31class WorkflowCreate(BaseModel):
32    name: str
33    trigger: dict
34    actions: list
35 
36class WorkflowTrigger(BaseModel):
37    workflow_id: int
38 
39@app.post("/signup")
40def signup(req: SignupRequest):
41    if req.username in users:
42        raise HTTPException(status_code=400, detail="User already exists")
43    users[req.username] = req.password
44    return {"message": "User created"}
45 
46@app.post("/login")
47def login(req: LoginRequest):
48    if users.get(req.username) != req.password:
49        raise HTTPException(status_code=401, detail="Invalid credentials")
50    token = secrets.token_hex(16)
51    tokens[token] = req.username
52    return {"token": token}
53 
54@app.post("/workflows")
55def create_workflow(req: WorkflowCreate, authorization: str = Header(None)):
56    user = authenticate(authorization)
57    global workflow_id_counter
58    wid = workflow_id_counter
59    workflow_id_counter += 1
60    workflows[wid] = {
61        "id": wid,
62        "name": req.name,
63        "trigger": req.trigger,
64        "actions": req.actions,
65        "owner": user
66    }
67    return {"id": wid}
68 
69@app.get("/workflows/{workflow_id}")
70def get_workflow(workflow_id: int, authorization: str = Header(None)):
71    authenticate(authorization)
72    wf = workflows.get(workflow_id)
73    if not wf:
74        raise HTTPException(status_code=404, detail="Workflow not found")
75    return wf
76 
77@app.post("/workflows/{workflow_id}/trigger")
78def trigger_workflow(workflow_id: int, authorization: str = Header(None)):
79    authenticate(authorization)
80    wf = workflows.get(workflow_id)
81    if not wf:
82        raise HTTPException(status_code=404, detail="Workflow not found")
83    global execution_log_id_counter
84    log_id = execution_log_id_counter
85    execution_log_id_counter += 1
86    log_entry = {
87        "id": log_id,
88        "workflow_id": workflow_id,
89        "trigger": wf["trigger"],
90        "actions": wf["actions"],
91        "status": "executed"
92    }
93    execution_logs.append(log_entry)
94    return {"message": "Workflow triggered", "execution_id": log_id}
95 
96@app.get("/execution-logs")
97def get_execution_logs(authorization: str = Header(None)):
98    authenticate(authorization)
99    return execution_logs
100 
101@app.get("/execution-logs/{log_id}")
102def get_execution_log(log_id: int, authorization: str = Header(None)):
103    authenticate(authorization)
104    for log in execution_logs:
105        if log["id"] == log_id:
106            return log
107    raise HTTPException(status_code=404, detail="Execution log not found")

1fastapi
2uvicorn