Browse the data

1import os
2import hashlib
3import time
4import random
5from fastapi import FastAPI, HTTPException, Header
6from fastapi.responses import FileResponse
7from pydantic import BaseModel
8 
9app = FastAPI()
10 
11users = {}
12tokens = {}
13orders = {}
14weather_data = {}
15next_user_id = 1
16next_order_id = 1
17next_weather_id = 1
18 
19TILE_CACHE_DIR = "tile_cache"
20 
21class SignupRequest(BaseModel):
22    username: str
23    password: str
24 
25class LoginRequest(BaseModel):
26    username: str
27    password: str
28 
29class OrderCreate(BaseModel):
30    item: str
31    quantity: int
32 
33class WeatherCreate(BaseModel):
34    location: str
35    temperature: float
36 
37def generate_token():
38    return hashlib.sha256(str(time.time()).encode() + os.urandom(16)).hexdigest()
39 
40def verify_token(authorization: str = Header(None)):
41    if not authorization:
42        raise HTTPException(status_code=401, detail="Missing Authorization header")
43    token = authorization.replace("Bearer ", "")
44    if token not in tokens:
45        raise HTTPException(status_code=401, detail="Invalid token")
46    return tokens[token]
47 
48@app.post("/signup")
49def signup(req: SignupRequest):
50    global next_user_id
51    user_id = next_user_id
52    next_user_id += 1
53    users[user_id] = {"username": req.username, "password": req.password, "id": user_id}
54    return {"id": user_id, "username": req.username}
55 
56@app.post("/login")
57def login(req: LoginRequest):
58    for uid, user in users.items():
59        if user["username"] == req.username and user["password"] == req.password:
60            token = generate_token()
61            tokens[token] = uid
62            return {"token": token}
63    raise HTTPException(status_code=401, detail="Invalid credentials")
64 
65@app.get("/orders/{order_id}")
66def get_order(order_id: int, authorization: str = Header(None)):
67    verify_token(authorization)
68    if order_id not in orders:
69        raise HTTPException(status_code=404, detail="Order not found")
70    return orders[order_id]
71 
72@app.post("/orders")
73def create_order(order: OrderCreate, authorization: str = Header(None)):
74    global next_order_id
75    verify_token(authorization)
76    order_id = next_order_id
77    next_order_id += 1
78    orders[order_id] = {"id": order_id, "item": order.item, "quantity": order.quantity}
79    return orders[order_id]
80 
81@app.get("/weather/{weather_id}")
82def get_weather(weather_id: int, authorization: str = Header(None)):
83    verify_token(authorization)
84    if weather_id not in weather_data:
85        raise HTTPException(status_code=404, detail="Weather not found")
86    return weather_data[weather_id]
87 
88@app.post("/weather")
89def create_weather(weather: WeatherCreate, authorization: str = Header(None)):
90    global next_weather_id
91    verify_token(authorization)
92    weather_id = next_weather_id
93    next_weather_id += 1
94    weather_data[weather_id] = {"id": weather_id, "location": weather.location, "temperature": weather.temperature}
95    return weather_data[weather_id]
96 
97@app.get("/weather_tiles/{z}/{x}/{y}.png")
98def get_weather_tile(z: int, x: int, y: int):
99    tile_path = os.path.join(TILE_CACHE_DIR, str(z), str(x), f"{y}.png")
100    if not os.path.exists(tile_path):
101        raise HTTPException(status_code=404, detail="Tile not found")
102    return FileResponse(tile_path, media_type="image/png")

1fastapi
2uvicorn