Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from typing import Optional
3import uuid
4import hashlib
5import secrets
6 
7app = FastAPI()
8 
9# In-memory stores
10users = {}
11mentorships = {}
12tokens = {}
13mentees = {}
14milestones = {}
15next_id = {"users": 1, "mentorships": 1, "mentees": 1, "milestones": 1}
16 
17def get_current_user(authorization: Optional[str] = Header(None)):
18    if not authorization:
19        raise HTTPException(status_code=401, detail="Missing auth token")
20    token = authorization.replace("Bearer ", "")
21    if token not in tokens:
22        raise HTTPException(status_code=401, detail="Invalid token")
23    return tokens[token]
24 
25@app.post("/signup")
26def signup(username: str, password: str):
27    if username in users:
28        raise HTTPException(status_code=400, detail="User already exists")
29    user_id = next_id["users"]
30    next_id["users"] += 1
31    users[username] = {"id": user_id, "username": username, "password": hashlib.sha256(password.encode()).hexdigest()}
32    return {"id": user_id, "username": username}
33 
34@app.post("/login")
35def login(username: str, password: str):
36    if username not in users:
37        raise HTTPException(status_code=401, detail="Invalid credentials")
38    if users[username]["password"] != hashlib.sha256(password.encode()).hexdigest():
39        raise HTTPException(status_code=401, detail="Invalid credentials")
40    token = secrets.token_hex(32)
41    tokens[token] = username
42    return {"token": token}
43 
44@app.get("/mentorships/{mentorship_id}")
45def get_mentorship(mentorship_id: int, authorization: Optional[str] = Header(None)):
46    current_user = get_current_user(authorization)
47    if mentorship_id not in mentorships:
48        raise HTTPException(status_code=404, detail="Mentorship not found")
49    return mentorships[mentorship_id]
50 
51@app.post("/mentorships")
52def create_mentorship(authorization: Optional[str] = Header(None)):
53    current_user = get_current_user(authorization)
54    mentorship_id = next_id["mentorships"]
55    next_id["mentorships"] += 1
56    mentorships[mentorship_id] = {
57        "id": mentorship_id,
58        "owner": current_user,
59        "goals": "",
60        "schedule": "",
61        "feedback": "",
62        "mentees": [],
63        "milestones": []
64    }
65    return mentorships[mentorship_id]
66 
67@app.put("/mentorships/{mentorship_id}")
68def update_mentorship(mentorship_id: int, goals: Optional[str] = None, schedule: Optional[str] = None, feedback: Optional[str] = None, authorization: Optional[str] = Header(None)):
69    current_user = get_current_user(authorization)
70    if mentorship_id not in mentorships:
71        raise HTTPException(status_code=404, detail="Mentorship not found")
72    if mentorships[mentorship_id]["owner"] != current_user:
73        raise HTTPException(status_code=403, detail="Not your mentorship")
74    if goals is not None:
75        mentorships[mentorship_id]["goals"] = goals
76    if schedule is not None:
77        mentorships[mentorship_id]["schedule"] = schedule
78    if feedback is not None:
79        mentorships[mentorship_id]["feedback"] = feedback
80    return mentorships[mentorship_id]
81 
82@app.post("/mentorships/{mentorship_id}/mentees")
83def add_mentee(mentorship_id: int, name: str, authorization: Optional[str] = Header(None)):
84    current_user = get_current_user(authorization)
85    if mentorship_id not in mentorships:
86        raise HTTPException(status_code=404, detail="Mentorship not found")
87    if mentorships[mentorship_id]["owner"] != current_user:
88        raise HTTPException(status_code=403, detail="Not your mentorship")
89    mentee_id = next_id["mentees"]
90    next_id["mentees"] += 1
91    mentee = {"id": mentee_id, "name": name, "mentorship_id": mentorship_id}
92    mentees[mentee_id] = mentee
93    mentorships[mentorship_id]["mentees"].append(mentee_id)
94    return mentee
95 
96@app.get("/mentees/{mentee_id}")
97def get_mentee(mentee_id: int, authorization: Optional[str] = Header(None)):
98    current_user = get_current_user(authorization)
99    if mentee_id not in mentees:
100        raise HTTPException(status_code=404, detail="Mentee not found")
101    return mentees[mentee_id]
102 
103@app.post("/mentorships/{mentorship_id}/milestones")
104def add_milestone(mentorship_id: int, title: str, description: Optional[str] = None, authorization: Optional[str] = Header(None)):
105    current_user = get_current_user(authorization)
106    if mentorship_id not in mentorships:
107        raise HTTPException(status_code=404, detail="Mentorship not found")
108    if mentorships[mentorship_id]["owner"] != current_user:
109        raise HTTPException(status_code=403, detail="Not your mentorship")
110    milestone_id = next_id["milestones"]
111    next_id["milestones"] += 1
112    milestone = {"id": milestone_id, "title": title, "description": description or "", "mentorship_id": mentorship_id, "completed": False}
113    milestones[milestone_id] = milestone
114    mentorships[mentorship_id]["milestones"].append(milestone_id)
115    return milestone
116 
117@app.get("/milestones/{milestone_id}")
118def get_milestone(milestone_id: int, authorization: Optional[str] = Header(None)):
119    current_user = get_current_user(authorization)
120    if milestone_id not in milestones:
121        raise HTTPException(status_code=404, detail="Milestone not found")
122    return milestones[milestone_id]
123 
124@app.patch("/milestones/{milestone_id}")
125def update_milestone(milestone_id: int, completed: Optional[bool] = None, title: Optional[str] = None, description: Optional[str] = None, authorization: Optional[str] = Header(None)):
126    current_user = get_current_user(authorization)
127    if milestone_id not in milestones:
128        raise HTTPException(status_code=404, detail="Milestone not found")
129    mentorship_id = milestones[milestone_id]["mentorship_id"]
130    if mentorships[mentorship_id]["owner"] != current_user:
131        raise HTTPException(status_code=403, detail="Not your mentorship")
132    if completed is not None:
133        milestones[milestone_id]["completed"] = completed
134    if title is not None:
135        milestones[milestone_id]["title"] = title
136    if description is not None:
137        milestones[milestone_id]["description"] = description
138    return milestones[milestone_id]

1fastapi
2uvicorn