Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3import hashlib
4import os
5 
6app = FastAPI()
7 
8users = {}
9tokens = {}
10next_user_id = 1
11 
12orders = {}
13next_order_id = 1
14 
15chat_attachments = {}
16 
17class SignupRequest(BaseModel):
18    username: str
19    password: str
20 
21class LoginRequest(BaseModel):
22    username: str
23    password: str
24 
25def get_current_user(authorization: str = Header(None)):
26    if not authorization:
27        raise HTTPException(status_code=401, detail="Missing auth header")
28    token = authorization.replace("Bearer ", "")
29    if token not in tokens:
30        raise HTTPException(status_code=401, detail="Invalid token")
31    return tokens[token]
32 
33@app.post("/signup")
34def signup(req: SignupRequest):
35    global next_user_id
36    if req.username in users:
37        raise HTTPException(status_code=400, detail="User exists")
38    user_id = next_user_id
39    next_user_id += 1
40    users[req.username] = {"id": user_id, "password": req.password}
41    return {"id": user_id, "username": req.username}
42 
43@app.post("/login")
44def login(req: LoginRequest):
45    if req.username not in users or users[req.username]["password"] != req.password:
46        raise HTTPException(status_code=401, detail="Invalid credentials")
47    token = hashlib.sha256(os.urandom(32)).hexdigest()
48    tokens[token] = req.username
49    return {"token": token}
50 
51@app.post("/orders")
52def create_order(authorization: str = Header(None)):
53    get_current_user(authorization)
54    global next_order_id
55    order_id = next_order_id
56    next_order_id += 1
57    orders[order_id] = {"id": order_id, "items": []}
58    return orders[order_id]
59 
60@app.get("/orders/{order_id}")
61def get_order(order_id: int, authorization: str = Header(None)):
62    get_current_user(authorization)
63    if order_id not in orders:
64        raise HTTPException(status_code=404, detail="Order not found")
65    return orders[order_id]
66 
67@app.get("/chats/{chat_id}/attachments/{filename}")
68def get_chat_attachment(chat_id: int, filename: str, authorization: str = Header(None)):
69    get_current_user(authorization)
70    path = f"/chat_attachments/{chat_id}/{filename}"
71    if not os.path.exists(path):
72        raise HTTPException(status_code=404, detail="File not found")
73    with open(path, "rb") as f:
74        content = f.read()
75    return content

1fastapi
2uvicorn