Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from typing import Optional
3import secrets
4 
5app = FastAPI()
6 
7users = {}
8mentors = {}
9mentees = {}
10sessions = {}
11bookings = {}
12tokens = {}
13user_id_counter = 1
14mentor_id_counter = 1
15mentee_id_counter = 1
16session_id_counter = 1
17booking_id_counter = 1
18 
19def get_current_user(authorization: Optional[str] = Header(None)):
20    if not authorization:
21        raise HTTPException(status_code=401, detail="Missing auth header")
22    token = authorization.replace("Bearer ", "")
23    if token not in tokens:
24        raise HTTPException(status_code=401, detail="Invalid token")
25    return tokens[token]
26 
27@app.post("/signup")
28def signup(username: str, password: str):
29    global user_id_counter
30    for u in users.values():
31        if u["username"] == username:
32            raise HTTPException(status_code=400, detail="Username taken")
33    user_id = user_id_counter
34    user_id_counter += 1
35    users[user_id] = {"id": user_id, "username": username, "password": password}
36    return {"id": user_id, "username": username}
37 
38@app.post("/login")
39def login(username: str, password: str):
40    for u in users.values():
41        if u["username"] == username and u["password"] == password:
42            token = secrets.token_hex(16)
43            tokens[token] = u["id"]
44            return {"token": token}
45    raise HTTPException(status_code=401, detail="Invalid credentials")
46 
47@app.get("/mentors/{mentor_id}")
48def get_mentor(mentor_id: int):
49    if mentor_id not in mentors:
50        raise HTTPException(status_code=404, detail="Mentor not found")
51    return mentors[mentor_id]
52 
53@app.get("/mentees/{mentee_id}")
54def get_mentee(mentee_id: int):
55    if mentee_id not in mentees:
56        raise HTTPException(status_code=404, detail="Mentee not found")
57    return mentees[mentee_id]
58 
59@app.get("/sessions/{session_id}")
60def get_session(session_id: int):
61    if session_id not in sessions:
62        raise HTTPException(status_code=404, detail="Session not found")
63    return sessions[session_id]
64 
65@app.get("/bookings/{booking_id}")
66def get_booking(booking_id: int):
67    if booking_id not in bookings:
68        raise HTTPException(status_code=404, detail="Booking not found")
69    return bookings[booking_id]
70 
71@app.post("/mentors")
72def create_mentor(name: str, bio: str, authorization: Optional[str] = Header(None)):
73    get_current_user(authorization)
74    global mentor_id_counter
75    mentor_id = mentor_id_counter
76    mentor_id_counter += 1
77    mentors[mentor_id] = {"id": mentor_id, "name": name, "bio": bio}
78    return mentors[mentor_id]
79 
80@app.post("/mentees")
81def create_mentee(name: str, interests: str, authorization: Optional[str] = Header(None)):
82    get_current_user(authorization)
83    global mentee_id_counter
84    mentee_id = mentee_id_counter
85    mentee_id_counter += 1
86    mentees[mentee_id] = {"id": mentee_id, "name": name, "interests": interests}
87    return mentees[mentee_id]
88 
89@app.post("/sessions")
90def create_session(mentor_id: int, title: str, description: str, authorization: Optional[str] = Header(None)):
91    user_id = get_current_user(authorization)
92    if mentor_id not in mentors:
93        raise HTTPException(status_code=404, detail="Mentor not found")
94    global session_id_counter
95    session_id = session_id_counter
96    session_id_counter += 1
97    sessions[session_id] = {
98        "id": session_id,
99        "mentor_id": mentor_id,
100        "title": title,
101        "description": description,
102        "booked": False
103    }
104    return sessions[session_id]
105 
106@app.post("/bookings")
107def create_booking(session_id: int, authorization: Optional[str] = Header(None)):
108    user_id = get_current_user(authorization)
109    if session_id not in sessions:
110        raise HTTPException(status_code=404, detail="Session not found")
111    if sessions[session_id]["booked"]:
112        raise HTTPException(status_code=400, detail="Session already booked")
113    sessions[session_id]["booked"] = True
114    global booking_id_counter
115    booking_id = booking_id_counter
116    booking_id_counter += 1
117    bookings[booking_id] = {
118        "id": booking_id,
119        "session_id": session_id,
120        "mentee_user_id": user_id
121    }
122    return bookings[booking_id]

1fastapi
2uvicorn