Browse the data

1import os
2import tarfile
3import io
4import hashlib
5import secrets
6from fastapi import FastAPI, HTTPException, Header, UploadFile, File
7from fastapi.responses import Response, StreamingResponse
8 
9app = FastAPI()
10 
11users = {}
12tokens = {}
13recipes = {}
14recipe_photos = {}
15next_user_id = 1
16next_recipe_id = 1
17AUTH_TOKEN_PREFIX = "token_"
18 
19def generate_token():
20    return AUTH_TOKEN_PREFIX + secrets.token_hex(16)
21 
22def get_user_from_token(authorization: str = Header(None)):
23    if not authorization:
24        raise HTTPException(status_code=401, detail="Missing Authorization header")
25    token = authorization.replace("Bearer ", "")
26    if token not in tokens:
27        raise HTTPException(status_code=401, detail="Invalid token")
28    return tokens[token]
29 
30@app.post("/signup")
31def signup(username: str, password: str):
32    global next_user_id
33    for u in users.values():
34        if u["username"] == username:
35            raise HTTPException(status_code=400, detail="Username already exists")
36    user_id = next_user_id
37    next_user_id += 1
38    password_hash = hashlib.sha256(password.encode()).hexdigest()
39    users[user_id] = {"id": user_id, "username": username, "password_hash": password_hash}
40    return {"id": user_id, "username": username}
41 
42@app.post("/login")
43def login(username: str, password: str):
44    for u in users.values():
45        if u["username"] == username and u["password_hash"] == hashlib.sha256(password.encode()).hexdigest():
46            token = generate_token()
47            tokens[token] = u["id"]
48            return {"token": token}
49    raise HTTPException(status_code=401, detail="Invalid credentials")
50 
51@app.post("/recipe")
52def create_recipe(name: str, authorization: str = Header(None)):
53    user_id = get_user_from_token(authorization)
54    global next_recipe_id
55    recipe_id = next_recipe_id
56    next_recipe_id += 1
57    recipes[recipe_id] = {"id": recipe_id, "name": name, "user_id": user_id}
58    os.makedirs(f"/var/recipes/{recipe_id}", exist_ok=True)
59    recipe_photos[recipe_id] = {}
60    return recipes[recipe_id]
61 
62@app.get("/recipe/{recipe_id}")
63def get_recipe(recipe_id: int, authorization: str = Header(None)):
64    get_user_from_token(authorization)
65    if recipe_id not in recipes:
66        raise HTTPException(status_code=404, detail="Recipe not found")
67    return recipes[recipe_id]
68 
69@app.post("/recipe/{recipe_id}/photos")
70async def upload_photo(recipe_id: int, file: UploadFile = File(...), authorization: str = Header(None)):
71    get_user_from_token(authorization)
72    if recipe_id not in recipes:
73        raise HTTPException(status_code=404, detail="Recipe not found")
74    content = await file.read()
75    filepath = f"/var/recipes/{recipe_id}/{file.filename}"
76    with open(filepath, "wb") as f:
77        f.write(content)
78    if recipe_id not in recipe_photos:
79        recipe_photos[recipe_id] = {}
80    recipe_photos[recipe_id][file.filename] = filepath
81    return {"filename": file.filename, "path": filepath}
82 
83@app.get("/recipe/{recipe_id}/photos/{file_name}")
84def get_photo(recipe_id: int, file_name: str, authorization: str = Header(None)):
85    get_user_from_token(authorization)
86    if recipe_id not in recipes:
87        raise HTTPException(status_code=404, detail="Recipe not found")
88    filepath = f"/var/recipes/{recipe_id}/{file_name}"
89    if not os.path.exists(filepath):
90        raise HTTPException(status_code=404, detail="File not found")
91    with open(filepath, "rb") as f:
92        content = f.read()
93    return Response(content=content, media_type="application/octet-stream")
94 
95@app.get("/recipe/{recipe_id}/photos")
96def get_all_photos(recipe_id: int, authorization: str = Header(None)):
97    get_user_from_token(authorization)
98    if recipe_id not in recipes:
99        raise HTTPException(status_code=404, detail="Recipe not found")
100    photos_dir = f"/var/recipes/{recipe_id}"
101    if not os.path.exists(photos_dir):
102        raise HTTPException(status_code=404, detail="No photos directory")
103    
104    tar_buffer = io.BytesIO()
105    with tarfile.open(fileobj=tar_buffer, mode="w:gz") as tar:
106        for filename in os.listdir(photos_dir):
107            filepath = os.path.join(photos_dir, filename)
108            if os.path.isfile(filepath):
109                tar.add(filepath, arcname=filename)
110    tar_buffer.seek(0)
111    return StreamingResponse(tar_buffer, media_type="application/gzip", headers={"Content-Disposition": f"attachment; filename=recipe_{recipe_id}_photos.tar.gz"})

1fastapi
2uvicorn
3python-multipart