Browse the data

1import os
2import hashlib
3import secrets
4from fastapi import FastAPI, HTTPException, Header, Query
5from fastapi.responses import FileResponse, Response
6from PIL import Image
7from io import BytesIO
8import uvicorn
9 
10app = FastAPI()
11 
12users = {}
13tokens = {}
14orders = {}
15articles = {}
16next_user_id = 1
17next_order_id = 1
18next_article_id = 1
19 
20def get_current_user(authorization: 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 user_id is None:
26        raise HTTPException(status_code=401, detail="Invalid token")
27    return user_id
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    password_hash = hashlib.sha256(password.encode()).hexdigest()
35    users[user_id] = {"username": username, "password_hash": password_hash}
36    return {"id": user_id, "username": username}
37 
38@app.post("/login")
39def login(username: str, password: str):
40    for user_id, user in users.items():
41        if user["username"] == username and user["password_hash"] == hashlib.sha256(password.encode()).hexdigest():
42            token = secrets.token_hex(16)
43            tokens[token] = user_id
44            return {"token": token}
45    raise HTTPException(status_code=401, detail="Invalid credentials")
46 
47@app.post("/orders")
48def create_order(item: str, quantity: int = 1, authorization: str = Header(None)):
49    user_id = get_current_user(authorization)
50    global next_order_id
51    order_id = next_order_id
52    next_order_id += 1
53    orders[order_id] = {"id": order_id, "item": item, "quantity": quantity, "user_id": user_id}
54    return orders[order_id]
55 
56@app.get("/orders/{order_id}")
57def get_order(order_id: int, authorization: str = Header(None)):
58    user_id = get_current_user(authorization)
59    order = orders.get(order_id)
60    if not order:
61        raise HTTPException(status_code=404, detail="Order not found")
62    return order
63 
64@app.post("/articles")
65def create_article(title: str, content: str, authorization: str = Header(None)):
66    user_id = get_current_user(authorization)
67    global next_article_id
68    article_id = next_article_id
69    next_article_id += 1
70    articles[article_id] = {"id": article_id, "title": title, "content": content, "user_id": user_id}
71    return articles[article_id]
72 
73@app.get("/articles/{article_id}")
74def get_article(article_id: int, authorization: str = Header(None)):
75    user_id = get_current_user(authorization)
76    article = articles.get(article_id)
77    if not article:
78        raise HTTPException(status_code=404, detail="Article not found")
79    return article
80 
81@app.get("/images/{article_id}/{image_name}")
82def get_image(article_id: int, image_name: str, width: int = Query(None)):
83    image_path = os.path.join("/news_media", image_name)
84    if not os.path.exists(image_path):
85        raise HTTPException(status_code=404, detail="Image not found")
86    
87    if width:
88        cache_dir = "/news_media/cache"
89        os.makedirs(cache_dir, exist_ok=True)
90        cache_path = os.path.join(cache_dir, f"{article_id}_{image_name}_{width}.jpg")
91        
92        if os.path.exists(cache_path):
93            return FileResponse(cache_path, media_type="image/jpeg")
94        
95        img = Image.open(image_path)
96        aspect = img.height / img.width
97        new_height = int(width * aspect)
98        img_resized = img.resize((width, new_height), Image.LANCZOS)
99        
100        img_resized.save(cache_path, "JPEG")
101        return FileResponse(cache_path, media_type="image/jpeg")
102    
103    return FileResponse(image_path, media_type="image/jpeg")

1fastapi
2uvicorn
3Pillow