Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from typing import Optional
3import secrets
4import hashlib
5 
6app = FastAPI()
7 
8users = {}
9tokens = {}
10agents = {}
11orders = {}
12bookings = {}
13next_user_id = 1
14next_agent_id = 1
15next_order_id = 1
16next_booking_id = 1
17 
18def get_current_user(authorization: Optional[str] = Header(None)):
19    if not authorization:
20        raise HTTPException(status_code=401, detail="Missing auth token")
21    token = authorization.replace("Bearer ", "")
22    for uid, t in tokens.items():
23        if t == token:
24            return uid
25    raise HTTPException(status_code=401, detail="Invalid token")
26 
27@app.post("/signup")
28def signup(username: str, password: str):
29    global next_user_id
30    for u in users.values():
31        if u["username"] == username:
32            raise HTTPException(status_code=400, detail="Username taken")
33    uid = next_user_id
34    next_user_id += 1
35    users[uid] = {"id": uid, "username": username, "password": hashlib.sha256(password.encode()).hexdigest(), "role": "user"}
36    return {"id": uid, "username": username}
37 
38@app.post("/login")
39def login(username: str, password: str):
40    for uid, u in users.items():
41        if u["username"] == username and u["password"] == hashlib.sha256(password.encode()).hexdigest():
42            token = secrets.token_hex(16)
43            tokens[uid] = token
44            return {"token": token, "user_id": uid}
45    raise HTTPException(status_code=401, detail="Invalid credentials")
46 
47@app.post("/promote/{user_id}")
48def promote(user_id: int, authorization: Optional[str] = Header(None)):
49    admin_id = get_current_user(authorization)
50    if users.get(admin_id, {}).get("role") != "admin":
51        raise HTTPException(status_code=403, detail="Only admins can promote")
52    if user_id not in users:
53        raise HTTPException(status_code=404, detail="User not found")
54    users[user_id]["role"] = "premium_agent"
55    return {"message": f"User {user_id} promoted to premium agent"}
56 
57@app.post("/agents")
58def create_agent(name: str, authorization: Optional[str] = Header(None)):
59    user_id = get_current_user(authorization)
60    global next_agent_id
61    aid = next_agent_id
62    next_agent_id += 1
63    agents[aid] = {"id": aid, "name": name, "owner_id": user_id}
64    return agents[aid]
65 
66@app.get("/agents/{agent_id}")
67def get_agent(agent_id: int):
68    if agent_id not in agents:
69        raise HTTPException(status_code=404, detail="Agent not found")
70    return agents[agent_id]
71 
72@app.post("/orders")
73def create_order(destination: str, authorization: Optional[str] = Header(None)):
74    user_id = get_current_user(authorization)
75    global next_order_id
76    oid = next_order_id
77    next_order_id += 1
78    orders[oid] = {"id": oid, "destination": destination, "user_id": user_id}
79    return orders[oid]
80 
81@app.get("/orders/{order_id}")
82def get_order(order_id: int):
83    if order_id not in orders:
84        raise HTTPException(status_code=404, detail="Order not found")
85    return orders[order_id]
86 
87@app.post("/bookings")
88def create_booking(details: str, authorization: Optional[str] = Header(None)):
89    user_id = get_current_user(authorization)
90    global next_booking_id
91    bid = next_booking_id
92    next_booking_id += 1
93    bookings[bid] = {"id": bid, "details": details, "user_id": user_id}
94    return bookings[bid]
95 
96@app.get("/bookings/{booking_id}")
97def get_booking(booking_id: int):
98    if booking_id not in bookings:
99        raise HTTPException(status_code=404, detail="Booking not found")
100    return bookings[booking_id]

1fastapi
2uvicorn