Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from typing import Optional
3import random
4import string
5 
6app = FastAPI()
7 
8users = {}
9tokens = {}
10offers = {}
11coupons = {}
12coupon_batches = {}
13next_user_id = 1
14next_offer_id = 1
15next_coupon_id = 1
16next_batch_id = 1
17 
18def generate_token():
19    return ''.join(random.choices(string.ascii_letters + string.digits, k=32))
20 
21def auth_required(authorization: Optional[str] = Header(None)):
22    if not authorization:
23        raise HTTPException(status_code=401, detail="Missing auth header")
24    token = authorization.replace("Bearer ", "")
25    if token not in tokens:
26        raise HTTPException(status_code=401, detail="Invalid token")
27    return tokens[token]
28 
29@app.post("/signup")
30def signup(username: str, password: str):
31    global next_user_id
32    user_id = next_user_id
33    next_user_id += 1
34    users[user_id] = {"id": user_id, "username": username, "password": password}
35    return {"user_id": user_id, "username": username}
36 
37@app.post("/login")
38def login(username: str, password: str):
39    for uid, u in users.items():
40        if u["username"] == username and u["password"] == password:
41            token = generate_token()
42            tokens[token] = uid
43            return {"token": token}
44    raise HTTPException(status_code=401, detail="Invalid credentials")
45 
46@app.get("/offers/{offer_id}")
47def get_offer(offer_id: int, authorization: Optional[str] = Header(None)):
48    auth_required(authorization)
49    if offer_id not in offers:
50        raise HTTPException(status_code=404, detail="Offer not found")
51    return offers[offer_id]
52 
53@app.post("/offers")
54def create_offer(name: str, discount_value: float, terms: str, authorization: Optional[str] = Header(None)):
55    auth_required(authorization)
56    global next_offer_id
57    offer_id = next_offer_id
58    next_offer_id += 1
59    offers[offer_id] = {"id": offer_id, "name": name, "discount_value": discount_value, "terms": terms}
60    return {"offer_id": offer_id}
61 
62@app.get("/offers/{offer_id}/redeem")
63def redeem_offer(offer_id: int, authorization: Optional[str] = Header(None)):
64    auth_required(authorization)
65    if offer_id not in offers:
66        raise HTTPException(status_code=404, detail="Offer not found")
67    offer = offers[offer_id]
68    return {"discount_value": offer["discount_value"], "terms": offer["terms"]}
69 
70@app.post("/coupons/generate")
71def generate_coupons(offer_id: int, count: int, max_uses_per_code: int = 1, authorization: Optional[str] = Header(None)):
72    auth_required(authorization)
73    if offer_id not in offers:
74        raise HTTPException(status_code=404, detail="Offer not found")
75    global next_batch_id, next_coupon_id
76    batch_id = next_batch_id
77    next_batch_id += 1
78    codes = []
79    for _ in range(count):
80        code = ''.join(random.choices(string.ascii_uppercase + string.digits, k=10))
81        coupon_id = next_coupon_id
82        next_coupon_id += 1
83        coupons[coupon_id] = {
84            "id": coupon_id,
85            "code": code,
86            "offer_id": offer_id,
87            "max_uses": max_uses_per_code,
88            "uses_remaining": max_uses_per_code
89        }
90        codes.append(code)
91    coupon_batches[batch_id] = {
92        "batch_id": batch_id,
93        "offer_id": offer_id,
94        "codes": codes,
95        "max_uses_per_code": max_uses_per_code
96    }
97    return {"batch_id": batch_id, "codes": codes}
98 
99@app.get("/coupons/{coupon_id}")
100def get_coupon(coupon_id: int, authorization: Optional[str] = Header(None)):
101    auth_required(authorization)
102    if coupon_id not in coupons:
103        raise HTTPException(status_code=404, detail="Coupon not found")
104    return coupons[coupon_id]
105 
106@app.post("/coupons")
107def create_coupon(code: str, offer_id: int, max_uses: int = 1, authorization: Optional[str] = Header(None)):
108    auth_required(authorization)
109    if offer_id not in offers:
110        raise HTTPException(status_code=404, detail="Offer not found")
111    global next_coupon_id
112    coupon_id = next_coupon_id
113    next_coupon_id += 1
114    coupons[coupon_id] = {
115        "id": coupon_id,
116        "code": code,
117        "offer_id": offer_id,
118        "max_uses": max_uses,
119        "uses_remaining": max_uses
120    }
121    return {"coupon_id": coupon_id}

1fastapi
2uvicorn