Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional, Dict, List
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[str, str] = {}
27 
28class CalendarUpdate(BaseModel):
29    name: Optional[str] = None
30    timezone: Optional[str] = None
31    visibility: Optional[str] = None
32    sharing_permissions: Optional[Dict[str, str]] = None
33 
34def get_current_user(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] = {"id": user_id, "username": req.username, "password": req.password}
46    token = secrets.token_hex(16)
47    tokens[token] = user_id
48    return {"user_id": user_id, "token": token}
49 
50@app.post("/login")
51def login(req: LoginRequest):
52    for uid, u in users.items():
53        if u["username"] == req.username and u["password"] == req.password:
54            token = secrets.token_hex(16)
55            tokens[token] = uid
56            return {"token": token}
57    raise HTTPException(status_code=401, detail="Invalid credentials")
58 
59@app.get("/calendars/{cal_id}")
60def get_calendar(cal_id: int, authorization: str = Header(...)):
61    user_id = get_current_user(authorization)
62    if cal_id not in calendars:
63        raise HTTPException(status_code=404, detail="Calendar not found")
64    return calendars[cal_id]
65 
66@app.post("/calendars")
67def create_calendar(cal: CalendarCreate, authorization: str = Header(...)):
68    global next_calendar_id
69    user_id = get_current_user(authorization)
70    cal_id = next_calendar_id
71    next_calendar_id += 1
72    calendars[cal_id] = {
73        "id": cal_id,
74        "owner_id": user_id,
75        "name": cal.name,
76        "timezone": cal.timezone,
77        "visibility": cal.visibility,
78        "sharing_permissions": cal.sharing_permissions
79    }
80    return calendars[cal_id]
81 
82@app.put("/calendars/{cal_id}")
83def update_calendar(cal_id: int, cal: CalendarUpdate, authorization: str = Header(...)):
84    user_id = get_current_user(authorization)
85    if cal_id not in calendars:
86        raise HTTPException(status_code=404, detail="Calendar not found")
87    if calendars[cal_id]["owner_id"] != user_id:
88        raise HTTPException(status_code=403, detail="Not authorized to update this calendar")
89    existing = calendars[cal_id]
90    if cal.name is not None:
91        existing["name"] = cal.name
92    if cal.timezone is not None:
93        existing["timezone"] = cal.timezone
94    if cal.visibility is not None:
95        existing["visibility"] = cal.visibility
96    if cal.sharing_permissions is not None:
97        existing["sharing_permissions"] = cal.sharing_permissions
98    return existing

1fastapi
2uvicorn