Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional, Dict, List
4import uuid
5 
6app = FastAPI()
7 
8users = {}
9tokens = {}
10projects = {}
11project_settings = {}
12next_user_id = 1
13next_project_id = 1
14next_setting_id = 1
15 
16class SignupRequest(BaseModel):
17    username: str
18    password: str
19 
20class LoginRequest(BaseModel):
21    username: str
22    password: str
23 
24class CreateProjectRequest(BaseModel):
25    name: str
26    description: Optional[str] = ""
27 
28class UpdateProjectSettingsRequest(BaseModel):
29    name: Optional[str] = None
30    description: Optional[str] = None
31    deadline: Optional[str] = None
32    members: Optional[Dict[str, str]] = None
33 
34def get_user_id_from_token(authorization: str = Header(...)):
35    token = authorization.replace("Bearer ", "")
36    user_id = tokens.get(token)
37    if user_id is None:
38        raise HTTPException(status_code=401, detail="Invalid token")
39    return user_id
40 
41@app.post("/signup")
42def signup(req: SignupRequest):
43    global next_user_id
44    user_id = next_user_id
45    next_user_id += 1
46    users[user_id] = {"username": req.username, "password": req.password}
47    return {"user_id": user_id, "message": "User created"}
48 
49@app.post("/login")
50def login(req: LoginRequest):
51    for uid, u in users.items():
52        if u["username"] == req.username and u["password"] == req.password:
53            token = str(uuid.uuid4())
54            tokens[token] = uid
55            return {"token": token}
56    raise HTTPException(status_code=401, detail="Invalid credentials")
57 
58@app.get("/projects/{project_id}")
59def get_project(project_id: int, authorization: str = Header(...)):
60    user_id = get_user_id_from_token(authorization)
61    if project_id not in projects:
62        raise HTTPException(status_code=404, detail="Project not found")
63    return projects[project_id]
64 
65@app.post("/projects")
66def create_project(req: CreateProjectRequest, authorization: str = Header(...)):
67    global next_project_id
68    user_id = get_user_id_from_token(authorization)
69    project_id = next_project_id
70    next_project_id += 1
71    projects[project_id] = {"id": project_id, "name": req.name, "description": req.description, "owner_id": user_id}
72    return projects[project_id]
73 
74@app.put("/projects/{project_id}/settings")
75def update_project_settings(project_id: int, req: UpdateProjectSettingsRequest, authorization: str = Header(...)):
76    user_id = get_user_id_from_token(authorization)
77    if project_id not in projects:
78        raise HTTPException(status_code=404, detail="Project not found")
79    if project_id not in project_settings:
80        project_settings[project_id] = {"members": {}}
81    settings = project_settings[project_id]
82    if req.name is not None:
83        projects[project_id]["name"] = req.name
84    if req.description is not None:
85        projects[project_id]["description"] = req.description
86    if req.deadline is not None:
87        settings["deadline"] = req.deadline
88    if req.members is not None:
89        for member_id, role in req.members.items():
90            if role not in ["manager", "admin", "member"]:
91                raise HTTPException(status_code=400, detail=f"Invalid role: {role}")
92            settings["members"][member_id] = role
93    project_settings[project_id] = settings
94    return {
95        "project": projects[project_id],
96        "settings": project_settings[project_id]
97    }

1fastapi
2uvicorn