Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional
4import uuid
5 
6app = FastAPI()
7 
8users = {}
9sellers = {}
10tokens = {}
11next_user_id = 1
12next_seller_id = 1
13 
14class SignupRequest(BaseModel):
15    username: str
16    password: str
17 
18class LoginRequest(BaseModel):
19    username: str
20    password: str
21 
22class SellerCreate(BaseModel):
23    store_name: str
24    contact_email: str
25    contact_phone: str
26 
27class SellerUpdate(BaseModel):
28    store_name: Optional[str] = None
29    contact_email: Optional[str] = None
30    contact_phone: Optional[str] = None
31    shipping_settings: Optional[dict] = None
32    seller_tier: Optional[str] = None
33 
34def get_current_user(token: str = Header(...)):
35    if token not in tokens:
36        raise HTTPException(status_code=401, detail="Invalid token")
37    return tokens[token]
38 
39@app.post("/signup")
40def signup(req: SignupRequest):
41    global next_user_id
42    user_id = next_user_id
43    next_user_id += 1
44    users[user_id] = {"id": user_id, "username": req.username, "password": req.password}
45    return {"id": user_id, "username": req.username}
46 
47@app.post("/login")
48def login(req: LoginRequest):
49    for user in users.values():
50        if user["username"] == req.username and user["password"] == req.password:
51            token = str(uuid.uuid4())
52            tokens[token] = user["id"]
53            return {"token": token}
54    raise HTTPException(status_code=401, detail="Invalid credentials")
55 
56@app.post("/sellers")
57def create_seller(req: SellerCreate, token: str = Header(...)):
58    global next_seller_id
59    user_id = get_current_user(token)
60    seller_id = next_seller_id
61    next_seller_id += 1
62    sellers[seller_id] = {
63        "id": seller_id,
64        "owner_id": user_id,
65        "store_name": req.store_name,
66        "contact_email": req.contact_email,
67        "contact_phone": req.contact_phone,
68        "shipping_settings": {},
69        "seller_tier": "standard"
70    }
71    return sellers[seller_id]
72 
73@app.get("/sellers/{seller_id}")
74def get_seller(seller_id: int, token: str = Header(...)):
75    get_current_user(token)
76    if seller_id not in sellers:
77        raise HTTPException(status_code=404, detail="Seller not found")
78    return sellers[seller_id]
79 
80@app.patch("/sellers/{seller_id}")
81def update_seller(seller_id: int, req: SellerUpdate, token: str = Header(...)):
82    user_id = get_current_user(token)
83    if seller_id not in sellers:
84        raise HTTPException(status_code=404, detail="Seller not found")
85    seller = sellers[seller_id]
86    if seller["owner_id"] != user_id:
87        raise HTTPException(status_code=403, detail="Not your seller")
88    if req.store_name is not None:
89        seller["store_name"] = req.store_name
90    if req.contact_email is not None:
91        seller["contact_email"] = req.contact_email
92    if req.contact_phone is not None:
93        seller["contact_phone"] = req.contact_phone
94    if req.shipping_settings is not None:
95        seller["shipping_settings"] = req.shipping_settings
96    if req.seller_tier is not None:
97        seller["seller_tier"] = req.seller_tier
98    return seller

1fastapi
2uvicorn