Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from datetime import datetime, timedelta
3import secrets
4import hashlib
5 
6app = FastAPI()
7 
8users = {}
9stations = {}
10observations = {}
11tokens = {}
12next_user_id = 1
13next_station_id = 1
14next_observation_id = 1
15 
16def get_user_from_token(authorization: str = Header(None)):
17    if not authorization:
18        raise HTTPException(status_code=401, detail="Missing auth header")
19    token = authorization.replace("Bearer ", "")
20    if token not in tokens:
21        raise HTTPException(status_code=401, detail="Invalid token")
22    return tokens[token]
23 
24@app.post("/signup")
25def signup(username: str, password: str):
26    global next_user_id
27    for u in users.values():
28        if u["username"] == username:
29            raise HTTPException(status_code=400, detail="Username taken")
30    user_id = next_user_id
31    next_user_id += 1
32    users[user_id] = {
33        "id": user_id,
34        "username": username,
35        "password_hash": hashlib.sha256(password.encode()).hexdigest()
36    }
37    return {"id": user_id, "username": username}
38 
39@app.post("/login")
40def login(username: str, password: str):
41    for u in users.values():
42        if u["username"] == username and u["password_hash"] == hashlib.sha256(password.encode()).hexdigest():
43            token = secrets.token_hex(32)
44            tokens[token] = u["id"]
45            return {"token": token}
46    raise HTTPException(status_code=401, detail="Invalid credentials")
47 
48@app.get("/station/{station_id}")
49def get_station(station_id: int, authorization: str = Header(None)):
50    user_id = get_user_from_token(authorization)
51    if station_id not in stations:
52        raise HTTPException(status_code=404, detail="Station not found")
53    return stations[station_id]
54 
55@app.post("/station")
56def create_station(name: str, latitude: float, longitude: float, authorization: str = Header(None)):
57    global next_station_id
58    user_id = get_user_from_token(authorization)
59    station_id = next_station_id
60    next_station_id += 1
61    stations[station_id] = {
62        "id": station_id,
63        "name": name,
64        "latitude": latitude,
65        "longitude": longitude,
66        "operator_id": user_id,
67        "co_operators": []
68    }
69    return stations[station_id]
70 
71@app.post("/station/{station_id}/promote")
72def promote_observer(station_id: int, observer_id: int, authorization: str = Header(None)):
73    user_id = get_user_from_token(authorization)
74    if station_id not in stations:
75        raise HTTPException(status_code=404, detail="Station not found")
76    station = stations[station_id]
77    if station["operator_id"] != user_id:
78        raise HTTPException(status_code=403, detail="Only operator can promote")
79    if observer_id not in users:
80        raise HTTPException(status_code=404, detail="Observer not found")
81    if observer_id not in station["co_operators"]:
82        station["co_operators"].append(observer_id)
83    return {"status": "promoted"}
84 
85@app.get("/observation/{observation_id}")
86def get_observation(observation_id: int, authorization: str = Header(None)):
87    user_id = get_user_from_token(authorization)
88    if observation_id not in observations:
89        raise HTTPException(status_code=404, detail="Observation not found")
90    return observations[observation_id]
91 
92@app.post("/observation")
93def create_observation(station_id: int, temperature: float, humidity: float, wind_speed: float, authorization: str = Header(None)):
94    global next_observation_id
95    user_id = get_user_from_token(authorization)
96    if station_id not in stations:
97        raise HTTPException(status_code=404, detail="Station not found")
98    station = stations[station_id]
99    if user_id != station["operator_id"] and user_id not in station["co_operators"]:
100        raise HTTPException(status_code=403, detail="Not authorized for this station")
101    observation_id = next_observation_id
102    next_observation_id += 1
103    observations[observation_id] = {
104        "id": observation_id,
105        "station_id": station_id,
106        "temperature": temperature,
107        "humidity": humidity,
108        "wind_speed": wind_speed,
109        "timestamp": datetime.utcnow().isoformat()
110    }
111    return observations[observation_id]

1fastapi
2uvicorn