Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3import secrets
4import uvicorn
5 
6app = FastAPI()
7 
8users = {}
9tokens = {}
10snippets = {}
11snippet_versions = {}
12orders = {}
13products = {}
14 
15class UserCreate(BaseModel):
16    username: str
17    password: str
18 
19class UserLogin(BaseModel):
20    username: str
21    password: str
22 
23class SnippetCreate(BaseModel):
24    content: str
25 
26class SnippetVersionCreate(BaseModel):
27    content: str
28 
29class OrderCreate(BaseModel):
30    product_id: int
31    quantity: int
32 
33class ProductCreate(BaseModel):
34    name: str
35    price: float
36 
37def get_current_user(authorization: str = Header(None)):
38    if not authorization:
39        raise HTTPException(status_code=401, detail="Missing auth header")
40    token = authorization.replace("Bearer ", "")
41    if token not in tokens:
42        raise HTTPException(status_code=401, detail="Invalid token")
43    return tokens[token]
44 
45@app.post("/signup")
46def signup(user: UserCreate):
47    if user.username in users:
48        raise HTTPException(status_code=400, detail="User already exists")
49    users[user.username] = {"username": user.username, "password": user.password}
50    return {"message": "User created"}
51 
52@app.post("/login")
53def login(user: UserLogin):
54    if user.username not in users or users[user.username]["password"] != user.password:
55        raise HTTPException(status_code=401, detail="Invalid credentials")
56    token = secrets.token_hex(16)
57    tokens[token] = user.username
58    return {"token": token}
59 
60@app.post("/snippets")
61def create_snippet(snippet: SnippetCreate, authorization: str = Header(None)):
62    get_current_user(authorization)
63    new_id = len(snippets) + 1
64    snippets[new_id] = {"id": new_id, "content": snippet.content}
65    return snippets[new_id]
66 
67@app.get("/snippets/{snippet_id}")
68def get_snippet(snippet_id: int, authorization: str = Header(None)):
69    get_current_user(authorization)
70    if snippet_id not in snippets:
71        raise HTTPException(status_code=404, detail="Snippet not found")
72    return snippets[snippet_id]
73 
74@app.post("/snippets/{snippet_id}/versions")
75def create_snippet_version(snippet_id: int, version: SnippetVersionCreate, authorization: str = Header(None)):
76    get_current_user(authorization)
77    if snippet_id not in snippets:
78        raise HTTPException(status_code=404, detail="Snippet not found")
79    if snippet_id not in snippet_versions:
80        snippet_versions[snippet_id] = {}
81    new_version_id = len(snippet_versions[snippet_id]) + 1
82    snippet_versions[snippet_id][new_version_id] = {"id": new_version_id, "content": version.content}
83    return snippet_versions[snippet_id][new_version_id]
84 
85@app.get("/snippets/{snippet_id}/versions/{version_id}/raw")
86def get_snippet_version_raw(snippet_id: int, version_id: int, authorization: str = Header(None)):
87    get_current_user(authorization)
88    if snippet_id not in snippet_versions or version_id not in snippet_versions[snippet_id]:
89        raise HTTPException(status_code=404, detail="Version not found")
90    return {"content": snippet_versions[snippet_id][version_id]["content"]}
91 
92@app.post("/orders")
93def create_order(order: OrderCreate, authorization: str = Header(None)):
94    get_current_user(authorization)
95    new_id = len(orders) + 1
96    orders[new_id] = {"id": new_id, "product_id": order.product_id, "quantity": order.quantity}
97    return orders[new_id]
98 
99@app.get("/orders/{order_id}")
100def get_order(order_id: int, authorization: str = Header(None)):
101    get_current_user(authorization)
102    if order_id not in orders:
103        raise HTTPException(status_code=404, detail="Order not found")
104    return orders[order_id]
105 
106@app.post("/products")
107def create_product(product: ProductCreate, authorization: str = Header(None)):
108    get_current_user(authorization)
109    new_id = len(products) + 1
110    products[new_id] = {"id": new_id, "name": product.name, "price": product.price}
111    return products[new_id]
112 
113@app.get("/products/{product_id}")
114def get_product(product_id: int, authorization: str = Header(None)):
115    get_current_user(authorization)
116    if product_id not in products:
117        raise HTTPException(status_code=404, detail="Product not found")
118    return products[product_id]
119 
120if __name__ == "__main__":
121    uvicorn.run(app, host="127.0.0.1", port=8000)

1fastapi
2uvicorn