Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional
4import secrets
5from datetime import datetime, timedelta
6import uvicorn
7 
8app = FastAPI()
9 
10users = {}
11tokens = {}
12orders = {}
13subscriptions = {}
14receipts = {}
15next_user_id = 1
16next_order_id = 1
17next_subscription_id = 1
18next_receipt_id = 1
19 
20def get_current_user(authorization: Optional[str] = Header(None)):
21    if not authorization:
22        raise HTTPException(status_code=401, detail="Missing auth header")
23    token = authorization.replace("Bearer ", "")
24    user_id = tokens.get(token)
25    if not user_id:
26        raise HTTPException(status_code=401, detail="Invalid token")
27    return user_id
28 
29class SignupRequest(BaseModel):
30    email: str
31    password: str
32 
33class LoginRequest(BaseModel):
34    email: str
35    password: str
36 
37class CreateOrder(BaseModel):
38    description: str
39    amount: float
40 
41class CreateSubscription(BaseModel):
42    plan: str
43    price: float
44 
45@app.post("/signup")
46def signup(req: SignupRequest):
47    global next_user_id
48    for u in users.values():
49        if u["email"] == req.email:
50            raise HTTPException(status_code=400, detail="Email already exists")
51    user_id = next_user_id
52    next_user_id += 1
53    users[user_id] = {"email": req.email, "password": req.password}
54    return {"user_id": user_id, "email": req.email}
55 
56@app.post("/login")
57def login(req: LoginRequest):
58    for uid, u in users.items():
59        if u["email"] == req.email and u["password"] == req.password:
60            token = secrets.token_hex(16)
61            tokens[token] = uid
62            return {"token": token, "user_id": uid}
63    raise HTTPException(status_code=401, detail="Invalid credentials")
64 
65@app.post("/orders")
66def create_order(order: CreateOrder, authorization: Optional[str] = Header(None)):
67    current_user = get_current_user(authorization)
68    global next_order_id
69    oid = next_order_id
70    next_order_id += 1
71    orders[oid] = {"id": oid, "user_id": current_user, **order.dict()}
72    return orders[oid]
73 
74@app.get("/orders/{order_id}")
75def get_order(order_id: int, authorization: Optional[str] = Header(None)):
76    current_user = get_current_user(authorization)
77    order = orders.get(order_id)
78    if not order:
79        raise HTTPException(status_code=404, detail="Order not found")
80    return order
81 
82@app.post("/subscriptions")
83def create_subscription(sub: CreateSubscription, authorization: Optional[str] = Header(None)):
84    current_user = get_current_user(authorization)
85    global next_subscription_id
86    sid = next_subscription_id
87    next_subscription_id += 1
88    subscriptions[sid] = {"id": sid, "user_id": current_user, **sub.dict()}
89    return subscriptions[sid]
90 
91@app.get("/subscriptions/{subscription_id}")
92def get_subscription(subscription_id: int, authorization: Optional[str] = Header(None)):
93    current_user = get_current_user(authorization)
94    sub = subscriptions.get(subscription_id)
95    if not sub:
96        raise HTTPException(status_code=404, detail="Subscription not found")
97    return sub
98 
99from fastapi.responses import FileResponse
100import os
101 
102@app.get("/receipts/{subscription_id}/{month}")
103def get_receipt(subscription_id: int, month: str, authorization: Optional[str] = Header(None)):
104    current_user = get_current_user(authorization)
105    sub = subscriptions.get(subscription_id)
106    if not sub:
107        raise HTTPException(status_code=404, detail="Subscription not found")
108    if sub["user_id"] != current_user:
109        raise HTTPException(status_code=403, detail="Not your subscription")
110    pdf_path = f"/billing/receipts/{subscription_id}/{subscription_id}_{month}.pdf"
111    if not os.path.exists(pdf_path):
112        raise HTTPException(status_code=404, detail="Receipt not found")
113    return FileResponse(pdf_path, media_type="application/pdf", filename=f"receipt_{subscription_id}_{month}.pdf")

1fastapi
2uvicorn