Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional
4import secrets
5 
6app = FastAPI()
7 
8users = {}
9tokens = {}
10calendars = {}
11next_user_id = 1
12next_calendar_id = 1
13 
14class SignupRequest(BaseModel):
15    username: str
16    password: str
17 
18class LoginRequest(BaseModel):
19    username: str
20    password: str
21 
22class CalendarCreate(BaseModel):
23    name: str
24    timezone: str = "UTC"
25    visibility: str = "private"
26    sharing_permissions: dict = {}
27 
28class CalendarUpdate(BaseModel):
29    name: Optional[str] = None
30    timezone: Optional[str] = None
31    visibility: Optional[str] = None
32    sharing_permissions: Optional[dict] = None
33 
34def get_user_id_from_token(authorization: str = Header(...)):
35    token = authorization.replace("Bearer ", "")
36    if token not in tokens:
37        raise HTTPException(status_code=401, detail="Invalid token")
38    return tokens[token]
39 
40@app.post("/signup")
41def signup(req: SignupRequest):
42    global next_user_id
43    user_id = next_user_id
44    next_user_id += 1
45    users[user_id] = {"username": req.username, "password": req.password}
46    return {"user_id": user_id}
47 
48@app.post("/login")
49def login(req: LoginRequest):
50    for uid, u in users.items():
51        if u["username"] == req.username and u["password"] == req.password:
52            token = secrets.token_hex(16)
53            tokens[token] = uid
54            return {"token": token}
55    raise HTTPException(status_code=401, detail="Invalid credentials")
56 
57@app.get("/calendars/{calendar_id}")
58def get_calendar(calendar_id: int, authorization: str = Header(...)):
59    user_id = get_user_id_from_token(authorization)
60    if calendar_id not in calendars:
61        raise HTTPException(status_code=404, detail="Calendar not found")
62    return calendars[calendar_id]
63 
64@app.post("/calendars")
65def create_calendar(req: CalendarCreate, authorization: str = Header(...)):
66    global next_calendar_id
67    user_id = get_user_id_from_token(authorization)
68    calendar_id = next_calendar_id
69    next_calendar_id += 1
70    calendars[calendar_id] = {
71        "id": calendar_id,
72        "name": req.name,
73        "timezone": req.timezone,
74        "visibility": req.visibility,
75        "sharing_permissions": req.sharing_permissions,
76        "owner_id": user_id
77    }
78    return calendars[calendar_id]
79 
80@app.put("/calendars/{calendar_id}")
81def update_calendar(calendar_id: int, req: CalendarUpdate, authorization: str = Header(...)):
82    user_id = get_user_id_from_token(authorization)
83    if calendar_id not in calendars:
84        raise HTTPException(status_code=404, detail="Calendar not found")
85    cal = calendars[calendar_id]
86    if cal["owner_id"] != user_id:
87        raise HTTPException(status_code=403, detail="Not authorized")
88    if req.name is not None:
89        cal["name"] = req.name
90    if req.timezone is not None:
91        cal["timezone"] = req.timezone
92    if req.visibility is not None:
93        cal["visibility"] = req.visibility
94    if req.sharing_permissions is not None:
95        cal["sharing_permissions"] = req.sharing_permissions
96    return cal

1fastapi
2uvicorn