Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional, Dict
4import secrets
5import datetime
6 
7app = FastAPI()
8 
9# In-memory stores
10users = {}
11tokens = {}
12coupons = {}
13stores = {}
14next_user_id = 1
15next_coupon_id = 1
16next_store_id = 1
17 
18# Seed some data
19stores[next_store_id] = {"id": next_store_id, "name": "Amazon", "coupons": []}
20next_store_id += 1
21stores[next_store_id] = {"id": next_store_id, "name": "Walmart", "coupons": []}
22next_store_id += 1
23stores[next_store_id] = {"id": next_store_id, "name": "Target", "coupons": []}
24next_store_id += 1
25 
26class SignupRequest(BaseModel):
27    username: str
28    password: str
29 
30class LoginRequest(BaseModel):
31    username: str
32    password: str
33 
34class CouponCreate(BaseModel):
35    store_name: str
36    discount_percentage: float
37    expiration_date: str
38    code: str
39 
40class StoreCreate(BaseModel):
41    name: str
42 
43def get_current_user(authorization: Optional[str] = Header(None)):
44    if not authorization:
45        raise HTTPException(status_code=401, detail="Missing auth header")
46    token = authorization.replace("Bearer ", "")
47    user_id = tokens.get(token)
48    if not user_id:
49        raise HTTPException(status_code=401, detail="Invalid token")
50    return user_id
51 
52@app.post("/signup")
53def signup(req: SignupRequest):
54    global next_user_id
55    for u in users.values():
56        if u["username"] == req.username:
57            raise HTTPException(status_code=400, detail="Username already exists")
58    uid = next_user_id
59    next_user_id += 1
60    users[uid] = {"id": uid, "username": req.username, "password": req.password}
61    token = secrets.token_hex(16)
62    tokens[token] = uid
63    return {"user_id": uid, "token": token}
64 
65@app.post("/login")
66def login(req: LoginRequest):
67    for uid, u in users.items():
68        if u["username"] == req.username and u["password"] == req.password:
69            token = secrets.token_hex(16)
70            tokens[token] = uid
71            return {"user_id": uid, "token": token}
72    raise HTTPException(status_code=401, detail="Invalid credentials")
73 
74@app.get("/coupons")
75def get_coupons(authorization: Optional[str] = Header(None)):
76    get_current_user(authorization)
77    now = datetime.datetime.now()
78    active = []
79    for cid, c in coupons.items():
80        exp = datetime.datetime.strptime(c["expiration_date"], "%Y-%m-%d")
81        if exp > now:
82            active.append({
83                "id": cid,
84                "store_name": c["store_name"],
85                "discount_percentage": c["discount_percentage"],
86                "expiration_date": c["expiration_date"],
87                "code": c["code"]
88            })
89    return active
90 
91@app.get("/stores")
92def get_stores(authorization: Optional[str] = Header(None)):
93    get_current_user(authorization)
94    result = []
95    for sid, s in stores.items():
96        result.append({
97            "id": sid,
98            "name": s["name"],
99            "coupon_count": len(s["coupons"])
100        })
101    return result
102 
103@app.post("/coupons")
104def create_coupon(req: CouponCreate, authorization: Optional[str] = Header(None)):
105    get_current_user(authorization)
106    global next_coupon_id
107    store_found = None
108    for sid, s in stores.items():
109        if s["name"] == req.store_name:
110            store_found = s
111            break
112    if not store_found:
113        raise HTTPException(status_code=404, detail="Store not found")
114    cid = next_coupon_id
115    next_coupon_id += 1
116    coupon = {
117        "id": cid,
118        "store_name": req.store_name,
119        "discount_percentage": req.discount_percentage,
120        "expiration_date": req.expiration_date,
121        "code": req.code
122    }
123    coupons[cid] = coupon
124    store_found["coupons"].append(cid)
125    return coupon
126 
127@app.post("/stores")
128def create_store(req: StoreCreate, authorization: Optional[str] = Header(None)):
129    get_current_user(authorization)
130    global next_store_id
131    sid = next_store_id
132    next_store_id += 1
133    store = {"id": sid, "name": req.name, "coupons": []}
134    stores[sid] = store
135    return store
136 
137@app.get("/coupons/{coupon_id}")
138def get_coupon(coupon_id: int, authorization: Optional[str] = Header(None)):
139    get_current_user(authorization)
140    c = coupons.get(coupon_id)
141    if not c:
142        raise HTTPException(status_code=404, detail="Coupon not found")
143    return c
144 
145@app.get("/stores/{store_id}")
146def get_store(store_id: int, authorization: Optional[str] = Header(None)):
147    get_current_user(authorization)
148    s = stores.get(store_id)
149    if not s:
150        raise HTTPException(status_code=404, detail="Store not found")
151    return s

1fastapi
2uvicorn