Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from datetime import datetime, timedelta
3import secrets
4import hashlib
5 
6app = FastAPI()
7 
8users = {}
9tokens = {}
10projects = {}
11tasks = {}
12project_id_counter = 1
13task_id_counter = 1
14user_id_counter = 1
15 
16def get_current_user(authorization: str = Header(None)):
17    if not authorization:
18        raise HTTPException(status_code=401, detail="Missing auth header")
19    token = authorization.replace("Bearer ", "")
20    if token not in tokens:
21        raise HTTPException(status_code=401, detail="Invalid token")
22    return tokens[token]
23 
24@app.post("/signup")
25def signup(username: str, password: str):
26    global user_id_counter
27    if any(u["username"] == username for u in users.values()):
28        raise HTTPException(status_code=400, detail="User exists")
29    user_id = user_id_counter
30    user_id_counter += 1
31    users[user_id] = {
32        "id": user_id,
33        "username": username,
34        "password": hashlib.sha256(password.encode()).hexdigest()
35    }
36    return {"user_id": user_id, "username": username}
37 
38@app.post("/login")
39def login(username: str, password: str):
40    user = next((u for u in users.values() if u["username"] == username), None)
41    if not user or user["password"] != hashlib.sha256(password.encode()).hexdigest():
42        raise HTTPException(status_code=401, detail="Invalid credentials")
43    token = secrets.token_hex(32)
44    tokens[token] = user["id"]
45    return {"token": token}
46 
47@app.get("/projects")
48def get_projects(authorization: str = Header(None)):
49    user_id = get_current_user(authorization)
50    result = []
51    for pid, p in projects.items():
52        member_list = p["members"]
53        task_counts = {"todo": 0, "in_progress": 0, "done": 0}
54        for t in tasks.values():
55            if t["project_id"] == pid:
56                status = t["status"]
57                if status in task_counts:
58                    task_counts[status] += 1
59        result.append({
60            "id": pid,
61            "name": p["name"],
62            "members": member_list,
63            "task_counts": task_counts
64        })
65    return result
66 
67@app.post("/projects")
68def create_project(name: str, members: list = [], authorization: str = Header(None)):
69    global project_id_counter
70    user_id = get_current_user(authorization)
71    pid = project_id_counter
72    project_id_counter += 1
73    projects[pid] = {
74        "id": pid,
75        "name": name,
76        "members": members,
77        "created_by": user_id
78    }
79    return {"id": pid, "name": name}
80 
81@app.get("/projects/{project_id}")
82def get_project(project_id: int, authorization: str = Header(None)):
83    user_id = get_current_user(authorization)
84    if project_id not in projects:
85        raise HTTPException(status_code=404, detail="Project not found")
86    p = projects[project_id]
87    return {
88        "id": project_id,
89        "name": p["name"],
90        "members": p["members"]
91    }
92 
93@app.post("/tasks")
94def create_task(project_id: int, title: str, assignee: str = None, due_date: str = None, priority: str = "medium", authorization: str = Header(None)):
95    global task_id_counter
96    user_id = get_current_user(authorization)
97    if project_id not in projects:
98        raise HTTPException(status_code=404, detail="Project not found")
99    tid = task_id_counter
100    task_id_counter += 1
101    tasks[tid] = {
102        "id": tid,
103        "project_id": project_id,
104        "title": title,
105        "status": "todo",
106        "assignee": assignee,
107        "due_date": due_date,
108        "priority": priority,
109        "created_by": user_id
110    }
111    return {"id": tid, "title": title}
112 
113@app.get("/tasks/{task_id}")
114def get_task(task_id: int, authorization: str = Header(None)):
115    user_id = get_current_user(authorization)
116    if task_id not in tasks:
117        raise HTTPException(status_code=404, detail="Task not found")
118    return tasks[task_id]

1fastapi
2uvicorn