Browse the data

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

1fastapi
2uvicorn