Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional, List
4import secrets
5import uvicorn
6 
7app = FastAPI()
8 
9users = {}
10volunteers = {}
11user_tokens = {}
12user_id_counter = 1
13volunteer_id_counter = 1
14 
15class SignupRequest(BaseModel):
16    username: str
17    password: str
18 
19class LoginRequest(BaseModel):
20    username: str
21    password: str
22 
23class VolunteerCreate(BaseModel):
24    name: str
25    skills: List[str] = []
26    availability: str = ""
27    project_role: str = ""
28 
29class VolunteerUpdate(BaseModel):
30    name: Optional[str] = None
31    skills: Optional[List[str]] = None
32    availability: Optional[str] = None
33    project_role: Optional[str] = None
34 
35def get_current_user(authorization: str = Header(...)):
36    if not authorization.startswith("Bearer "):
37        raise HTTPException(status_code=401, detail="Invalid auth header")
38    token = authorization.split(" ")[1]
39    for uid, t in user_tokens.items():
40        if t == token:
41            return uid
42    raise HTTPException(status_code=401, detail="Invalid token")
43 
44@app.post("/signup")
45def signup(req: SignupRequest):
46    global user_id_counter
47    uid = user_id_counter
48    user_id_counter += 1
49    users[uid] = {"username": req.username, "password": req.password}
50    token = secrets.token_hex(16)
51    user_tokens[uid] = token
52    return {"user_id": uid, "token": token}
53 
54@app.post("/login")
55def login(req: LoginRequest):
56    for uid, u in users.items():
57        if u["username"] == req.username and u["password"] == req.password:
58            token = secrets.token_hex(16)
59            user_tokens[uid] = token
60            return {"user_id": uid, "token": token}
61    raise HTTPException(status_code=401, detail="Invalid credentials")
62 
63@app.get("/volunteers/{volunteer_id}")
64def get_volunteer(volunteer_id: int, authorization: str = Header(...)):
65    get_current_user(authorization)
66    if volunteer_id not in volunteers:
67        raise HTTPException(status_code=404, detail="Volunteer not found")
68    return volunteers[volunteer_id]
69 
70@app.post("/volunteers")
71def create_volunteer(v: VolunteerCreate, authorization: str = Header(...)):
72    get_current_user(authorization)
73    global volunteer_id_counter
74    vid = volunteer_id_counter
75    volunteer_id_counter += 1
76    volunteers[vid] = {
77        "id": vid,
78        "name": v.name,
79        "skills": v.skills,
80        "availability": v.availability,
81        "project_role": v.project_role
82    }
83    return volunteers[vid]
84 
85@app.patch("/volunteers/{volunteer_id}")
86def update_volunteer(volunteer_id: int, v: VolunteerUpdate, authorization: str = Header(...)):
87    get_current_user(authorization)
88    if volunteer_id not in volunteers:
89        raise HTTPException(status_code=404, detail="Volunteer not found")
90    existing = volunteers[volunteer_id]
91    if v.name is not None:
92        existing["name"] = v.name
93    if v.skills is not None:
94        existing["skills"] = v.skills
95    if v.availability is not None:
96        existing["availability"] = v.availability
97    if v.project_role is not None:
98        existing["project_role"] = v.project_role
99    return existing
100 
101if __name__ == "__main__":
102    uvicorn.run(app, host="127.0.0.1", port=8000)

1fastapi
2uvicorn