Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from datetime import datetime, timedelta
3import hashlib
4import secrets
5 
6app = FastAPI()
7 
8users = {}
9tokens = {}
10plants = {}
11plant_id_counter = 1
12user_id_counter = 1
13water_logs = {}
14 
15def get_user_id_from_token(authorization: str = Header(...)):
16    if not authorization.startswith("Bearer "):
17        raise HTTPException(status_code=401, detail="Invalid auth header")
18    token = authorization[7:]
19    if token not in tokens:
20        raise HTTPException(status_code=401, detail="Invalid token")
21    return tokens[token]
22 
23@app.post("/signup")
24def signup(username: str, password: str):
25    global user_id_counter
26    if any(u["username"] == username for u in users.values()):
27        raise HTTPException(status_code=400, detail="User exists")
28    user_id = user_id_counter
29    user_id_counter += 1
30    users[user_id] = {"id": user_id, "username": username, "password": hashlib.sha256(password.encode()).hexdigest()}
31    return {"user_id": user_id, "username": username}
32 
33@app.post("/login")
34def login(username: str, password: str):
35    for uid, u in users.items():
36        if u["username"] == username and u["password"] == hashlib.sha256(password.encode()).hexdigest():
37            token = secrets.token_hex(16)
38            tokens[token] = uid
39            return {"token": token}
40    raise HTTPException(status_code=401, detail="Invalid credentials")
41 
42@app.get("/plants")
43def get_plants(authorization: str = Header(...)):
44    get_user_id_from_token(authorization)
45    result = []
46    now = datetime.now()
47    for pid, p in plants.items():
48        next_water = p["last_watered"] + timedelta(hours=p["watering_frequency_hours"])
49        result.append({
50            "id": pid,
51            "species": p["species"],
52            "watering_frequency_hours": p["watering_frequency_hours"],
53            "next_water_date": next_water.isoformat()
54        })
55    return result
56 
57@app.post("/plants")
58def create_plant(species: str, watering_frequency_hours: int, authorization: str = Header(...)):
59    global plant_id_counter
60    user_id = get_user_id_from_token(authorization)
61    plant_id = plant_id_counter
62    plant_id_counter += 1
63    plants[plant_id] = {
64        "id": plant_id,
65        "species": species,
66        "watering_frequency_hours": watering_frequency_hours,
67        "last_watered": datetime.now(),
68        "user_id": user_id
69    }
70    return {"id": plant_id, "species": species}
71 
72@app.get("/plants/{plant_id}")
73def get_plant(plant_id: int, authorization: str = Header(...)):
74    get_user_id_from_token(authorization)
75    if plant_id not in plants:
76        raise HTTPException(status_code=404, detail="Plant not found")
77    p = plants[plant_id]
78    next_water = p["last_watered"] + timedelta(hours=p["watering_frequency_hours"])
79    return {
80        "id": plant_id,
81        "species": p["species"],
82        "watering_frequency_hours": p["watering_frequency_hours"],
83        "next_water_date": next_water.isoformat()
84    }
85 
86@app.post("/plants/{plant_id}/water")
87def water_plant(plant_id: int, authorization: str = Header(...)):
88    user_id = get_user_id_from_token(authorization)
89    if plant_id not in plants:
90        raise HTTPException(status_code=404, detail="Plant not found")
91    p = plants[plant_id]
92    p["last_watered"] = datetime.now()
93    if plant_id not in water_logs:
94        water_logs[plant_id] = []
95    water_logs[plant_id].append({"date": datetime.now().isoformat(), "user_id": user_id})
96    return {"message": "Watered today", "plant_id": plant_id}
97 
98@app.get("/reminders")
99def get_reminders(authorization: str = Header(...)):
100    get_user_id_from_token(authorization)
101    now = datetime.now()
102    reminders = []
103    for pid, p in plants.items():
104        next_water = p["last_watered"] + timedelta(hours=p["watering_frequency_hours"])
105        if now <= next_water <= now + timedelta(hours=48):
106            reminders.append({
107                "id": pid,
108                "species": p["species"],
109                "next_water_date": next_water.isoformat()
110            })
111    return reminders

1fastapi
2uvicorn