Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional, List
4import secrets
5import hashlib
6 
7app = FastAPI()
8 
9users = {}
10volunteers = {}
11tokens = {}
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: List[str] = []
27    project_role: str = ""
28 
29class VolunteerUpdate(BaseModel):
30    name: Optional[str] = None
31    skills: Optional[List[str]] = None
32    availability: Optional[List[str]] = None
33    project_role: Optional[str] = None
34 
35def get_current_user(authorization: str = Header(None)):
36    if not authorization:
37        raise HTTPException(status_code=401, detail="Missing Authorization header")
38    token = authorization.replace("Bearer ", "")
39    if token not in tokens:
40        raise HTTPException(status_code=401, detail="Invalid token")
41    return tokens[token]
42 
43@app.post("/signup")
44def signup(req: SignupRequest):
45    global user_id_counter
46    if req.username in users:
47        raise HTTPException(status_code=400, detail="Username already exists")
48    user_id = user_id_counter
49    user_id_counter += 1
50    users[req.username] = {"id": user_id, "password": hashlib.sha256(req.password.encode()).hexdigest()}
51    return {"id": user_id, "username": req.username}
52 
53@app.post("/login")
54def login(req: LoginRequest):
55    if req.username not in users:
56        raise HTTPException(status_code=401, detail="Invalid credentials")
57    if users[req.username]["password"] != hashlib.sha256(req.password.encode()).hexdigest():
58        raise HTTPException(status_code=401, detail="Invalid credentials")
59    token = secrets.token_hex(16)
60    tokens[token] = req.username
61    return {"token": token}
62 
63@app.get("/volunteers/{volunteer_id}")
64def get_volunteer(volunteer_id: int, authorization: str = Header(None)):
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(volunteer: VolunteerCreate, authorization: str = Header(None)):
72    global volunteer_id_counter
73    get_current_user(authorization)
74    vid = volunteer_id_counter
75    volunteer_id_counter += 1
76    volunteers[vid] = {
77        "id": vid,
78        "name": volunteer.name,
79        "skills": volunteer.skills,
80        "availability": volunteer.availability,
81        "project_role": volunteer.project_role
82    }
83    return volunteers[vid]
84 
85@app.patch("/volunteers/{volunteer_id}")
86def update_volunteer(volunteer_id: int, update: VolunteerUpdate, authorization: str = Header(None)):
87    get_current_user(authorization)
88    if volunteer_id not in volunteers:
89        raise HTTPException(status_code=404, detail="Volunteer not found")
90    v = volunteers[volunteer_id]
91    if update.name is not None:
92        v["name"] = update.name
93    if update.skills is not None:
94        v["skills"] = update.skills
95    if update.availability is not None:
96        v["availability"] = update.availability
97    if update.project_role is not None:
98        v["project_role"] = update.project_role
99    return v

1fastapi
2uvicorn