Browse the data

1from fastapi import FastAPI, HTTPException, Header
2import uuid
3from typing import Optional
4 
5app = FastAPI()
6 
7users = {}
8tokens = {}
9plots = {}
10watering_logs = {}
11harvest_history = {}
12plot_id_counter = 1
13watering_id_counter = 1
14harvest_id_counter = 1
15user_id_counter = 1
16 
17def get_current_user(authorization: Optional[str] = Header(None)):
18    if not authorization:
19        raise HTTPException(status_code=401, detail="No auth header")
20    if not authorization.startswith("Bearer "):
21        raise HTTPException(status_code=401, detail="Invalid auth header format")
22    token = authorization.split(" ")[1]
23    if token not in tokens:
24        raise HTTPException(status_code=401, detail="Invalid token")
25    return tokens[token]
26 
27@app.post("/signup")
28def signup(username: str, password: str):
29    global user_id_counter
30    for u in users.values():
31        if u["username"] == username:
32            raise HTTPException(status_code=400, detail="Username already exists")
33    user_id = user_id_counter
34    user_id_counter += 1
35    users[user_id] = {"id": user_id, "username": username, "password": password}
36    return {"id": user_id, "username": username}
37 
38@app.post("/login")
39def login(username: str, password: str):
40    for u in users.values():
41        if u["username"] == username and u["password"] == password:
42            token = str(uuid.uuid4())
43            tokens[token] = u["id"]
44            return {"token": token}
45    raise HTTPException(status_code=401, detail="Invalid credentials")
46 
47@app.post("/plots")
48def create_plot(description: str = "", location: str = "", authorization: Optional[str] = Header(None)):
49    user_id = get_current_user(authorization)
50    global plot_id_counter
51    plot_id = plot_id_counter
52    plot_id_counter += 1
53    plots[plot_id] = {"id": plot_id, "description": description, "location": location, "claimed_by": None, "user_id": user_id}
54    return plots[plot_id]
55 
56@app.get("/plots/{plot_id}")
57def get_plot(plot_id: int, authorization: Optional[str] = Header(None)):
58    get_current_user(authorization)
59    if plot_id not in plots:
60        raise HTTPException(status_code=404, detail="Plot not found")
61    return plots[plot_id]
62 
63@app.post("/plots/{plot_id}/claim")
64def claim_plot(plot_id: int, authorization: Optional[str] = Header(None)):
65    user_id = get_current_user(authorization)
66    if plot_id not in plots:
67        raise HTTPException(status_code=404, detail="Plot not found")
68    if plots[plot_id]["claimed_by"] is not None:
69        raise HTTPException(status_code=400, detail="Plot already claimed")
70    plots[plot_id]["claimed_by"] = user_id
71    return plots[plot_id]
72 
73@app.post("/watering")
74def log_watering(plot_id: int, plant_type: str, amount: float = 1.0, authorization: Optional[str] = Header(None)):
75    user_id = get_current_user(authorization)
76    if plot_id not in plots:
77        raise HTTPException(status_code=404, detail="Plot not found")
78    if plots[plot_id]["claimed_by"] != user_id:
79        raise HTTPException(status_code=403, detail="Not your plot")
80    global watering_id_counter
81    log_id = watering_id_counter
82    watering_id_counter += 1
83    watering_logs[log_id] = {"id": log_id, "plot_id": plot_id, "plant_type": plant_type, "amount": amount, "user_id": user_id}
84    return watering_logs[log_id]
85 
86@app.get("/watering/{log_id}")
87def get_watering_log(log_id: int, authorization: Optional[str] = Header(None)):
88    get_current_user(authorization)
89    if log_id not in watering_logs:
90        raise HTTPException(status_code=404, detail="Watering log not found")
91    return watering_logs[log_id]
92 
93@app.post("/harvest")
94def log_harvest(plot_id: int, plant_type: str, quantity: float = 1.0, authorization: Optional[str] = Header(None)):
95    user_id = get_current_user(authorization)
96    if plot_id not in plots:
97        raise HTTPException(status_code=404, detail="Plot not found")
98    if plots[plot_id]["claimed_by"] != user_id:
99        raise HTTPException(status_code=403, detail="Not your plot")
100    global harvest_id_counter
101    harvest_id = harvest_id_counter
102    harvest_id_counter += 1
103    harvest_history[harvest_id] = {"id": harvest_id, "plot_id": plot_id, "plant_type": plant_type, "quantity": quantity, "user_id": user_id}
104    return harvest_history[harvest_id]
105 
106@app.get("/harvest/{harvest_id}")
107def get_harvest(harvest_id: int, authorization: Optional[str] = Header(None)):
108    get_current_user(authorization)
109    if harvest_id not in harvest_history:
110        raise HTTPException(status_code=404, detail="Harvest record not found")
111    return harvest_history[harvest_id]
112 
113@app.get("/plots/{plot_id}/history")
114def get_plot_history(plot_id: int, authorization: Optional[str] = Header(None)):
115    get_current_user(authorization)
116    if plot_id not in plots:
117        raise HTTPException(status_code=404, detail="Plot not found")
118    waterings = [v for v in watering_logs.values() if v["plot_id"] == plot_id]
119    harvests = [v for v in harvest_history.values() if v["plot_id"] == plot_id]
120    return {"plot_id": plot_id, "waterings": waterings, "harvests": harvests}

1fastapi
2uvicorn