Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional, List
4import secrets
5 
6app = FastAPI()
7 
8users = {}
9tokens = {}
10next_user_id = 1
11 
12notebooks = {}
13next_notebook_id = 1
14 
15notes = {}
16next_note_id = 1
17 
18class SignupRequest(BaseModel):
19    username: str
20    password: str
21 
22class LoginRequest(BaseModel):
23    username: str
24    password: str
25 
26class CreateNotebookRequest(BaseModel):
27    name: str
28 
29class CreateNoteRequest(BaseModel):
30    notebook_id: int
31    title: str
32    content: str = ""
33 
34class UpdateProfileRequest(BaseModel):
35    name: Optional[str] = None
36    theme: Optional[str] = None
37    default_notebook_id: Optional[int] = None
38    sharing_settings: Optional[dict] = None
39    role: Optional[str] = None
40 
41def get_current_user(authorization: str = Header(None)):
42    if not authorization:
43        raise HTTPException(status_code=401, detail="Missing auth header")
44    token = authorization.replace("Bearer ", "")
45    if token not in tokens:
46        raise HTTPException(status_code=401, detail="Invalid token")
47    user_id = tokens[token]
48    return user_id
49 
50@app.post("/signup")
51def signup(req: SignupRequest):
52    global next_user_id
53    user_id = next_user_id
54    next_user_id += 1
55    users[user_id] = {
56        "id": user_id,
57        "username": req.username,
58        "password": req.password,
59        "name": req.username,
60        "theme": "light",
61        "default_notebook_id": None,
62        "sharing_settings": {},
63        "role": "editor"
64    }
65    token = secrets.token_hex(16)
66    tokens[token] = user_id
67    return {"user_id": user_id, "token": token}
68 
69@app.post("/login")
70def login(req: LoginRequest):
71    for uid, u in users.items():
72        if u["username"] == req.username and u["password"] == req.password:
73            token = secrets.token_hex(16)
74            tokens[token] = uid
75            return {"user_id": uid, "token": token}
76    raise HTTPException(status_code=401, detail="Invalid credentials")
77 
78@app.get("/users/{user_id}")
79def get_user(user_id: int):
80    if user_id not in users:
81        raise HTTPException(status_code=404, detail="User not found")
82    return users[user_id]
83 
84@app.post("/notebooks")
85def create_notebook(req: CreateNotebookRequest, authorization: str = Header(None)):
86    user_id = get_current_user(authorization)
87    global next_notebook_id
88    nid = next_notebook_id
89    next_notebook_id += 1
90    notebooks[nid] = {
91        "id": nid,
92        "name": req.name,
93        "owner_id": user_id
94    }
95    return notebooks[nid]
96 
97@app.get("/notebooks/{notebook_id}")
98def get_notebook(notebook_id: int):
99    if notebook_id not in notebooks:
100        raise HTTPException(status_code=404, detail="Notebook not found")
101    return notebooks[notebook_id]
102 
103@app.post("/notes")
104def create_note(req: CreateNoteRequest, authorization: str = Header(None)):
105    user_id = get_current_user(authorization)
106    if req.notebook_id not in notebooks:
107        raise HTTPException(status_code=404, detail="Notebook not found")
108    global next_note_id
109    nid = next_note_id
110    next_note_id += 1
111    notes[nid] = {
112        "id": nid,
113        "notebook_id": req.notebook_id,
114        "title": req.title,
115        "content": req.content,
116        "owner_id": user_id
117    }
118    return notes[nid]
119 
120@app.get("/notes/{note_id}")
121def get_note(note_id: int):
122    if note_id not in notes:
123        raise HTTPException(status_code=404, detail="Note not found")
124    return notes[note_id]
125 
126@app.patch("/users/{user_id}/profile")
127def update_profile(user_id: int, req: UpdateProfileRequest, authorization: str = Header(None)):
128    current_user_id = get_current_user(authorization)
129    if user_id != current_user_id:
130        raise HTTPException(status_code=403, detail="Cannot update another user's profile")
131    if user_id not in users:
132        raise HTTPException(status_code=404, detail="User not found")
133    user = users[user_id]
134    if req.name is not None:
135        user["name"] = req.name
136    if req.theme is not None:
137        user["theme"] = req.theme
138    if req.default_notebook_id is not None:
139        if req.default_notebook_id not in notebooks:
140            raise HTTPException(status_code=404, detail="Notebook not found")
141        user["default_notebook_id"] = req.default_notebook_id
142    if req.sharing_settings is not None:
143        user["sharing_settings"] = req.sharing_settings
144    if req.role is not None:
145        if req.role not in ["editor", "admin"]:
146            raise HTTPException(status_code=400, detail="Invalid role")
147        user["role"] = req.role
148    return user

1fastapi
2uvicorn