Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from typing import Optional
3import secrets
4 
5app = FastAPI()
6 
7users = {}
8tokens = {}
9gigs = {}
10disputes = {}
11user_counter = 0
12gig_counter = 0
13dispute_counter = 0
14guild_admin_id = None
15moderators = set()
16 
17def get_current_user(authorization: Optional[str] = Header(None)):
18    if not authorization:
19        raise HTTPException(status_code=401, detail="Missing auth header")
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    global user_counter, guild_admin_id
28    user_counter += 1
29    user_id = user_counter
30    users[user_id] = {"id": user_id, "username": username, "password": password, "role": "user"}
31    if guild_admin_id is None:
32        guild_admin_id = user_id
33        users[user_id]["role"] = "admin"
34    return {"user_id": user_id, "username": username, "role": users[user_id]["role"]}
35 
36@app.post("/login")
37def login(username: str, password: str):
38    for uid, u in users.items():
39        if u["username"] == username and u["password"] == password:
40            token = secrets.token_hex(16)
41            tokens[token] = uid
42            return {"token": token, "user_id": uid}
43    raise HTTPException(status_code=401, detail="Invalid credentials")
44 
45@app.post("/promote/{user_id}")
46def promote_user(user_id: int, authorization: Optional[str] = Header(None)):
47    current_user = get_current_user(authorization)
48    if users[current_user]["role"] != "admin":
49        raise HTTPException(status_code=403, detail="Only guild admin can promote")
50    if user_id not in users:
51        raise HTTPException(status_code=404, detail="User not found")
52    users[user_id]["role"] = "moderator"
53    moderators.add(user_id)
54    return {"user_id": user_id, "role": "moderator"}
55 
56@app.get("/gig/{gig_id}")
57def get_gig(gig_id: int):
58    if gig_id not in gigs:
59        raise HTTPException(status_code=404, detail="Gig not found")
60    return gigs[gig_id]
61 
62@app.post("/gig")
63def create_gig(title: str, description: str, price: float, authorization: Optional[str] = Header(None)):
64    current_user = get_current_user(authorization)
65    global gig_counter
66    gig_counter += 1
67    gigs[gig_counter] = {
68        "id": gig_counter,
69        "title": title,
70        "description": description,
71        "price": price,
72        "created_by": current_user,
73        "approved": False
74    }
75    return gigs[gig_counter]
76 
77@app.post("/gig/{gig_id}/approve")
78def approve_gig(gig_id: int, authorization: Optional[str] = Header(None)):
79    current_user = get_current_user(authorization)
80    if current_user not in moderators and users[current_user]["role"] != "admin":
81        raise HTTPException(status_code=403, detail="Only moderators or admin can approve")
82    if gig_id not in gigs:
83        raise HTTPException(status_code=404, detail="Gig not found")
84    gigs[gig_id]["approved"] = True
85    return gigs[gig_id]
86 
87@app.get("/dispute/{dispute_id}")
88def get_dispute(dispute_id: int):
89    if dispute_id not in disputes:
90        raise HTTPException(status_code=404, detail="Dispute not found")
91    return disputes[dispute_id]
92 
93@app.post("/dispute")
94def create_dispute(gig_id: int, reason: str, authorization: Optional[str] = Header(None)):
95    current_user = get_current_user(authorization)
96    global dispute_counter
97    dispute_counter += 1
98    disputes[dispute_counter] = {
99        "id": dispute_counter,
100        "gig_id": gig_id,
101        "reason": reason,
102        "created_by": current_user,
103        "resolved": False
104    }
105    return disputes[dispute_counter]
106 
107@app.post("/dispute/{dispute_id}/resolve")
108def resolve_dispute(dispute_id: int, resolution: str, authorization: Optional[str] = Header(None)):
109    current_user = get_current_user(authorization)
110    if current_user not in moderators and users[current_user]["role"] != "admin":
111        raise HTTPException(status_code=403, detail="Only moderators or admin can resolve")
112    if dispute_id not in disputes:
113        raise HTTPException(status_code=404, detail="Dispute not found")
114    disputes[dispute_id]["resolved"] = True
115    disputes[dispute_id]["resolution"] = resolution
116    disputes[dispute_id]["resolved_by"] = current_user
117    return disputes[dispute_id]
118 
119@app.get("/user/{user_id}")
120def get_user(user_id: int):
121    if user_id not in users:
122        raise HTTPException(status_code=404, detail="User not found")
123    return {"id": users[user_id]["id"], "username": users[user_id]["username"], "role": users[user_id]["role"]}

1fastapi
2uvicorn