Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional, Dict, List
4import secrets
5import uvicorn
6 
7app = FastAPI()
8 
9users = {}
10tokens = {}
11workspaces = {}
12workspace_id_counter = 1
13 
14class SignupRequest(BaseModel):
15    username: str
16    password: str
17 
18class LoginRequest(BaseModel):
19    username: str
20    password: str
21 
22class WorkspaceCreate(BaseModel):
23    name: str
24    description: Optional[str] = ""
25 
26class WorkspaceUpdate(BaseModel):
27    name: Optional[str] = None
28    description: Optional[str] = None
29    member_roles: Optional[Dict[str, str]] = None
30    admin_settings: Optional[Dict] = None
31 
32def get_current_user(authorization: str = Header(None)):
33    if not authorization:
34        raise HTTPException(status_code=401, detail="Missing auth token")
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    if req.username in users:
43        raise HTTPException(status_code=400, detail="User already exists")
44    users[req.username] = {"username": req.username, "password": req.password}
45    token = secrets.token_hex(16)
46    tokens[token] = req.username
47    return {"token": token, "username": req.username}
48 
49@app.post("/login")
50def login(req: LoginRequest):
51    if req.username not in users or users[req.username]["password"] != req.password:
52        raise HTTPException(status_code=401, detail="Invalid credentials")
53    token = secrets.token_hex(16)
54    tokens[token] = req.username
55    return {"token": token, "username": req.username}
56 
57@app.get("/workspaces/{workspace_id}")
58def get_workspace(workspace_id: int, authorization: str = Header(None)):
59    user = get_current_user(authorization)
60    if workspace_id not in workspaces:
61        raise HTTPException(status_code=404, detail="Workspace not found")
62    return workspaces[workspace_id]
63 
64@app.post("/workspaces")
65def create_workspace(req: WorkspaceCreate, authorization: str = Header(None)):
66    global workspace_id_counter
67    user = get_current_user(authorization)
68    ws = {
69        "id": workspace_id_counter,
70        "name": req.name,
71        "description": req.description,
72        "owner": user,
73        "members": {user: "admin"},
74        "admin_settings": {}
75    }
76    workspaces[workspace_id_counter] = ws
77    workspace_id_counter += 1
78    return ws
79 
80@app.patch("/workspaces/{workspace_id}")
81def update_workspace(workspace_id: int, req: WorkspaceUpdate, authorization: str = Header(None)):
82    user = get_current_user(authorization)
83    if workspace_id not in workspaces:
84        raise HTTPException(status_code=404, detail="Workspace not found")
85    ws = workspaces[workspace_id]
86    if ws["owner"] != user:
87        raise HTTPException(status_code=403, detail="Only owner can update workspace")
88    if req.name is not None:
89        ws["name"] = req.name
90    if req.description is not None:
91        ws["description"] = req.description
92    if req.member_roles is not None:
93        ws["members"] = req.member_roles
94    if req.admin_settings is not None:
95        ws["admin_settings"] = req.admin_settings
96    return ws

1fastapi
2uvicorn