Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional, List
4import secrets
5import datetime
6 
7app = FastAPI()
8 
9users = {}
10tokens = {}
11templates = {}
12template_id_counter = 1
13 
14class SignupRequest(BaseModel):
15    username: str
16    password: str
17 
18class LoginRequest(BaseModel):
19    username: str
20    password: str
21 
22class ChecklistItem(BaseModel):
23    text: str
24    checked: bool = False
25 
26class TemplateCreate(BaseModel):
27    name: str
28    checklist_items: Optional[List[ChecklistItem]] = []
29    priority: Optional[int] = 0
30    assigned_role: Optional[str] = ""
31    recurrence: Optional[str] = ""
32 
33class TemplateUpdate(BaseModel):
34    name: Optional[str] = None
35    checklist_items: Optional[List[ChecklistItem]] = None
36    priority: Optional[int] = None
37    assigned_role: Optional[str] = None
38    recurrence: Optional[str] = None
39 
40def get_current_user(authorization: str = Header(...)):
41    if not authorization.startswith("Bearer "):
42        raise HTTPException(status_code=401, detail="Invalid auth header")
43    token = authorization.split(" ")[1]
44    user_id = tokens.get(token)
45    if user_id is None:
46        raise HTTPException(status_code=401, detail="Invalid token")
47    return user_id
48 
49@app.post("/signup")
50def signup(req: SignupRequest):
51    if req.username in users:
52        raise HTTPException(status_code=400, detail="User already exists")
53    users[req.username] = req.password
54    return {"message": "User created"}
55 
56@app.post("/login")
57def login(req: LoginRequest):
58    stored = users.get(req.username)
59    if stored is None or stored != req.password:
60        raise HTTPException(status_code=401, detail="Invalid credentials")
61    token = secrets.token_hex(32)
62    tokens[token] = req.username
63    return {"token": token}
64 
65@app.get("/templates/{template_id}")
66def get_template(template_id: int, authorization: str = Header(...)):
67    get_current_user(authorization)
68    template = templates.get(template_id)
69    if template is None:
70        raise HTTPException(status_code=404, detail="Template not found")
71    return template
72 
73@app.post("/templates")
74def create_template(req: TemplateCreate, authorization: str = Header(...)):
75    get_current_user(authorization)
76    global template_id_counter
77    new_template = {
78        "id": template_id_counter,
79        "name": req.name,
80        "checklist_items": [item.dict() for item in req.checklist_items],
81        "priority": req.priority,
82        "assigned_role": req.assigned_role,
83        "recurrence": req.recurrence
84    }
85    templates[template_id_counter] = new_template
86    template_id_counter += 1
87    return new_template
88 
89@app.patch("/templates/{template_id}")
90def update_template(template_id: int, req: TemplateUpdate, authorization: str = Header(...)):
91    get_current_user(authorization)
92    if template_id not in templates:
93        raise HTTPException(status_code=404, detail="Template not found")
94    template = templates[template_id]
95    if req.name is not None:
96        template["name"] = req.name
97    if req.checklist_items is not None:
98        template["checklist_items"] = [item.dict() for item in req.checklist_items]
99    if req.priority is not None:
100        template["priority"] = req.priority
101    if req.assigned_role is not None:
102        template["assigned_role"] = req.assigned_role
103    if req.recurrence is not None:
104        template["recurrence"] = req.recurrence
105    return template

1fastapi
2uvicorn
3pydantic