Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional, List
4import secrets
5import datetime
6 
7app = FastAPI()
8 
9users = {}
10tasks = {}
11teams = {}
12task_id_counter = 1
13user_id_counter = 1
14team_id_counter = 1
15tokens = {}
16 
17class SignupRequest(BaseModel):
18    username: str
19    password: str
20 
21class LoginRequest(BaseModel):
22    username: str
23    password: str
24 
25class TaskCreate(BaseModel):
26    title: str
27    description: str = ""
28    assignee: Optional[str] = None
29    due_date: Optional[str] = None
30    priority: str = "medium"
31    status: str = "todo"
32 
33class TaskUpdate(BaseModel):
34    title: Optional[str] = None
35    description: Optional[str] = None
36    assignee: Optional[str] = None
37    due_date: Optional[str] = None
38    priority: Optional[str] = None
39 
40class BulkStatusUpdate(BaseModel):
41    task_ids: List[int]
42    status: str
43 
44class TeamCreate(BaseModel):
45    name: str
46    member_usernames: List[str] = []
47 
48def get_current_user(authorization: str = Header(None)):
49    if not authorization:
50        raise HTTPException(status_code=401, detail="No auth header")
51    token = authorization.replace("Bearer ", "")
52    if token not in tokens:
53        raise HTTPException(status_code=401, detail="Invalid token")
54    return tokens[token]
55 
56@app.post("/signup")
57def signup(req: SignupRequest):
58    global user_id_counter
59    if req.username in users:
60        raise HTTPException(status_code=400, detail="Username taken")
61    user_id = user_id_counter
62    user_id_counter += 1
63    users[req.username] = {"id": user_id, "username": req.username, "password": req.password}
64    return {"id": user_id, "username": req.username}
65 
66@app.post("/login")
67def login(req: LoginRequest):
68    if req.username not in users or users[req.username]["password"] != req.password:
69        raise HTTPException(status_code=401, detail="Invalid credentials")
70    token = secrets.token_hex(16)
71    tokens[token] = req.username
72    return {"token": token}
73 
74@app.get("/tasks/{task_id}")
75def get_task(task_id: int, authorization: str = Header(None)):
76    get_current_user(authorization)
77    if task_id not in tasks:
78        raise HTTPException(status_code=404, detail="Task not found")
79    return tasks[task_id]
80 
81@app.post("/tasks")
82def create_task(task: TaskCreate, authorization: str = Header(None)):
83    global task_id_counter
84    get_current_user(authorization)
85    task_id = task_id_counter
86    task_id_counter += 1
87    tasks[task_id] = {
88        "id": task_id,
89        "title": task.title,
90        "description": task.description,
91        "assignee": task.assignee,
92        "due_date": task.due_date,
93        "priority": task.priority,
94        "status": task.status,
95        "created_at": datetime.datetime.utcnow().isoformat()
96    }
97    return tasks[task_id]
98 
99@app.put("/tasks/{task_id}")
100def update_task(task_id: int, update: TaskUpdate, authorization: str = Header(None)):
101    get_current_user(authorization)
102    if task_id not in tasks:
103        raise HTTPException(status_code=404, detail="Task not found")
104    task = tasks[task_id]
105    if update.title is not None:
106        task["title"] = update.title
107    if update.description is not None:
108        task["description"] = update.description
109    if update.assignee is not None:
110        task["assignee"] = update.assignee
111    if update.due_date is not None:
112        task["due_date"] = update.due_date
113    if update.priority is not None:
114        task["priority"] = update.priority
115    return task
116 
117@app.patch("/tasks/bulk-status")
118def bulk_update_status(bulk: BulkStatusUpdate, authorization: str = Header(None)):
119    get_current_user(authorization)
120    updated = []
121    for tid in bulk.task_ids:
122        if tid in tasks:
123            tasks[tid]["status"] = bulk.status
124            updated.append(tasks[tid])
125    return {"updated": updated}
126 
127@app.get("/teams/{team_id}")
128def get_team(team_id: int, authorization: str = Header(None)):
129    get_current_user(authorization)
130    if team_id not in teams:
131        raise HTTPException(status_code=404, detail="Team not found")
132    return teams[team_id]
133 
134@app.post("/teams")
135def create_team(team: TeamCreate, authorization: str = Header(None)):
136    global team_id_counter
137    get_current_user(authorization)
138    team_id = team_id_counter
139    team_id_counter += 1
140    teams[team_id] = {
141        "id": team_id,
142        "name": team.name,
143        "members": team.member_usernames,
144        "created_at": datetime.datetime.utcnow().isoformat()
145    }
146    return teams[team_id]

1fastapi
2uvicorn