Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from datetime import datetime, date
3from typing import Optional
4import secrets
5import hashlib
6 
7app = FastAPI()
8 
9users = {}
10tokens = {}
11stations = {}
12readings = {}
13reading_id_counter = 1
14station_id_counter = 1
15user_id_counter = 1
16 
17def get_current_user(authorization: Optional[str] = Header(None)):
18    if not authorization or not authorization.startswith("Bearer "):
19        raise HTTPException(status_code=401, detail="Invalid auth")
20    token = authorization.split(" ")[1]
21    for uid, t in tokens.items():
22        if t == token:
23            return uid
24    raise HTTPException(status_code=401, detail="Invalid token")
25 
26@app.post("/signup")
27def signup(username: str, password: str):
28    global user_id_counter
29    if any(u["username"] == username for u in users.values()):
30        raise HTTPException(status_code=400, detail="User exists")
31    uid = user_id_counter
32    users[uid] = {"id": uid, "username": username, "password": hashlib.sha256(password.encode()).hexdigest()}
33    user_id_counter += 1
34    return {"id": uid, "username": username}
35 
36@app.post("/login")
37def login(username: str, password: str):
38    for uid, u in users.items():
39        if u["username"] == username and u["password"] == hashlib.sha256(password.encode()).hexdigest():
40            token = secrets.token_hex(16)
41            tokens[uid] = token
42            return {"token": token}
43    raise HTTPException(status_code=401, detail="Invalid credentials")
44 
45@app.post("/stations")
46def create_station(name: str, location: str, elevation: float, authorization: str = Header(None)):
47    user_id = get_current_user(authorization)
48    global station_id_counter
49    sid = station_id_counter
50    stations[sid] = {"id": sid, "name": name, "location": location, "elevation": elevation, "user_id": user_id}
51    station_id_counter += 1
52    return stations[sid]
53 
54@app.get("/stations/{station_id}")
55def get_station(station_id: int, authorization: str = Header(None)):
56    get_current_user(authorization)
57    if station_id not in stations:
58        raise HTTPException(status_code=404, detail="Station not found")
59    return stations[station_id]
60 
61@app.post("/readings")
62def create_reading(station_id: int, temperature: float, humidity: float, wind_speed: float, recorded_at: str, authorization: str = Header(None)):
63    get_current_user(authorization)
64    if station_id not in stations:
65        raise HTTPException(status_code=404, detail="Station not found")
66    global reading_id_counter
67    rid = reading_id_counter
68    readings[rid] = {
69        "id": rid,
70        "station_id": station_id,
71        "temperature": temperature,
72        "humidity": humidity,
73        "wind_speed": wind_speed,
74        "recorded_at": recorded_at
75    }
76    reading_id_counter += 1
77    return readings[rid]
78 
79@app.get("/readings/{reading_id}")
80def get_reading(reading_id: int, authorization: str = Header(None)):
81    get_current_user(authorization)
82    if reading_id not in readings:
83        raise HTTPException(status_code=404, detail="Reading not found")
84    return readings[reading_id]
85 
86@app.get("/stations/{station_id}/daily")
87def get_daily_aggregation(station_id: int, date_str: str, authorization: str = Header(None)):
88    get_current_user(authorization)
89    if station_id not in stations:
90        raise HTTPException(status_code=404, detail="Station not found")
91    target_date = date.fromisoformat(date_str)
92    temps = []
93    for r in readings.values():
94        if r["station_id"] == station_id:
95            r_date = datetime.fromisoformat(r["recorded_at"]).date()
96            if r_date == target_date:
97                temps.append(r["temperature"])
98    if not temps:
99        return {"station_id": station_id, "date": date_str, "min_temp": None, "max_temp": None}
100    return {"station_id": station_id, "date": date_str, "min_temp": min(temps), "max_temp": max(temps)}

1fastapi
2uvicorn