Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional, List
4import secrets
5from datetime import datetime, timezone
6 
7app = FastAPI()
8 
9users = {}
10calendars = {}
11events = {}
12tokens = {}
13user_id_counter = 1
14calendar_id_counter = 1
15event_id_counter = 1
16 
17class SignupRequest(BaseModel):
18    email: str
19    password: str
20 
21class LoginRequest(BaseModel):
22    email: str
23    password: str
24 
25class CalendarCreate(BaseModel):
26    name: str
27    timezone: str = "UTC"
28    visibility: str = "private"
29    sharing_permissions: str = "none"
30 
31class CalendarUpdate(BaseModel):
32    name: Optional[str] = None
33    timezone: Optional[str] = None
34    visibility: Optional[str] = None
35    sharing_permissions: Optional[str] = None
36 
37class EventCreate(BaseModel):
38    title: str
39    start_time: str
40    end_time: str
41    description: str = ""
42    calendar_id: int
43 
44def get_user_id(authorization: str = Header(None)):
45    if not authorization:
46        raise HTTPException(status_code=401, detail="Missing authorization header")
47    token = authorization.replace("Bearer ", "")
48    user_id = tokens.get(token)
49    if not user_id:
50        raise HTTPException(status_code=401, detail="Invalid token")
51    return user_id
52 
53@app.post("/signup")
54def signup(req: SignupRequest):
55    global user_id_counter
56    for u in users.values():
57        if u["email"] == req.email:
58            raise HTTPException(status_code=400, detail="Email already exists")
59    user_id = user_id_counter
60    user_id_counter += 1
61    users[user_id] = {"id": user_id, "email": req.email, "password": req.password}
62    token = secrets.token_hex(32)
63    tokens[token] = user_id
64    return {"user_id": user_id, "token": token}
65 
66@app.post("/login")
67def login(req: LoginRequest):
68    for u in users.values():
69        if u["email"] == req.email and u["password"] == req.password:
70            token = secrets.token_hex(32)
71            tokens[token] = u["id"]
72            return {"user_id": u["id"], "token": token}
73    raise HTTPException(status_code=401, detail="Invalid credentials")
74 
75@app.post("/calendars")
76def create_calendar(req: CalendarCreate, authorization: str = Header(None)):
77    global calendar_id_counter
78    user_id = get_user_id(authorization)
79    cal_id = calendar_id_counter
80    calendar_id_counter += 1
81    calendars[cal_id] = {
82        "id": cal_id,
83        "name": req.name,
84        "timezone": req.timezone,
85        "visibility": req.visibility,
86        "sharing_permissions": req.sharing_permissions,
87        "owner_id": user_id
88    }
89    return calendars[cal_id]
90 
91@app.get("/calendars/{cal_id}")
92def get_calendar(cal_id: int, authorization: str = Header(None)):
93    user_id = get_user_id(authorization)
94    cal = calendars.get(cal_id)
95    if not cal:
96        raise HTTPException(status_code=404, detail="Calendar not found")
97    return cal
98 
99@app.put("/calendars/{cal_id}")
100def update_calendar(cal_id: int, req: CalendarUpdate, authorization: str = Header(None)):
101    user_id = get_user_id(authorization)
102    cal = calendars.get(cal_id)
103    if not cal:
104        raise HTTPException(status_code=404, detail="Calendar not found")
105    if cal["owner_id"] != user_id:
106        raise HTTPException(status_code=403, detail="Not authorized to update this calendar")
107    if req.name is not None:
108        cal["name"] = req.name
109    if req.timezone is not None:
110        cal["timezone"] = req.timezone
111    if req.visibility is not None:
112        cal["visibility"] = req.visibility
113    if req.sharing_permissions is not None:
114        cal["sharing_permissions"] = req.sharing_permissions
115    return cal
116 
117@app.post("/events")
118def create_event(req: EventCreate, authorization: str = Header(None)):
119    global event_id_counter
120    user_id = get_user_id(authorization)
121    cal = calendars.get(req.calendar_id)
122    if not cal:
123        raise HTTPException(status_code=404, detail="Calendar not found")
124    if cal["owner_id"] != user_id:
125        raise HTTPException(status_code=403, detail="Not authorized to add events to this calendar")
126    evt_id = event_id_counter
127    event_id_counter += 1
128    events[evt_id] = {
129        "id": evt_id,
130        "title": req.title,
131        "start_time": req.start_time,
132        "end_time": req.end_time,
133        "description": req.description,
134        "calendar_id": req.calendar_id
135    }
136    return events[evt_id]
137 
138@app.get("/events/{evt_id}")
139def get_event(evt_id: int, authorization: str = Header(None)):
140    user_id = get_user_id(authorization)
141    evt = events.get(evt_id)
142    if not evt:
143        raise HTTPException(status_code=404, detail="Event not found")
144    return evt

1fastapi
2uvicorn