Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from typing import Optional
3import hashlib
4import random
5import string
6from datetime import datetime, timedelta
7 
8app = FastAPI()
9 
10users = {}
11tokens = {}
12plots = {}
13progress_logs = {}
14water_reminders = {}
15plot_id_counter = 1
16progress_id_counter = 1
17reminder_id_counter = 1
18user_id_counter = 1
19 
20def generate_token():
21    return ''.join(random.choices(string.ascii_letters + string.digits, k=32))
22 
23def get_current_user(authorization: Optional[str] = Header(None)):
24    if not authorization or not authorization.startswith("Bearer "):
25        raise HTTPException(status_code=401, detail="Invalid auth")
26    token = authorization[7:]
27    if token not in tokens:
28        raise HTTPException(status_code=401, detail="Invalid token")
29    return tokens[token]
30 
31@app.post("/signup")
32def signup(username: str, password: str):
33    global user_id_counter
34    if any(u["username"] == username for u in users.values()):
35        raise HTTPException(status_code=400, detail="Username taken")
36    user_id = user_id_counter
37    user_id_counter += 1
38    users[user_id] = {
39        "id": user_id,
40        "username": username,
41        "password": hashlib.sha256(password.encode()).hexdigest()
42    }
43    return {"user_id": user_id, "username": username}
44 
45@app.post("/login")
46def login(username: str, password: str):
47    for u in users.values():
48        if u["username"] == username and u["password"] == hashlib.sha256(password.encode()).hexdigest():
49            token = generate_token()
50            tokens[token] = u["id"]
51            return {"token": token}
52    raise HTTPException(status_code=401, detail="Invalid credentials")
53 
54@app.post("/plots")
55def create_plot(plant_type: str, plot_name: str, authorization: Optional[str] = Header(None)):
56    user_id = get_current_user(authorization)
57    global plot_id_counter
58    plot_id = plot_id_counter
59    plot_id_counter += 1
60    plots[plot_id] = {
61        "id": plot_id,
62        "plot_name": plot_name,
63        "plant_type": plant_type,
64        "user_id": user_id,
65        "created_at": datetime.now().isoformat()
66    }
67    return plots[plot_id]
68 
69@app.get("/plots/{plot_id}")
70def get_plot(plot_id: int, authorization: Optional[str] = Header(None)):
71    get_current_user(authorization)
72    if plot_id not in plots:
73        raise HTTPException(status_code=404, detail="Plot not found")
74    return plots[plot_id]
75 
76@app.post("/progress")
77def create_progress(plot_id: int, note: str, authorization: Optional[str] = Header(None)):
78    user_id = get_current_user(authorization)
79    if plot_id not in plots:
80        raise HTTPException(status_code=404, detail="Plot not found")
81    global progress_id_counter
82    progress_id = progress_id_counter
83    progress_id_counter += 1
84    progress_logs[progress_id] = {
85        "id": progress_id,
86        "plot_id": plot_id,
87        "note": note,
88        "user_id": user_id,
89        "timestamp": datetime.now().isoformat()
90    }
91    return progress_logs[progress_id]
92 
93@app.get("/progress/{progress_id}")
94def get_progress(progress_id: int, authorization: Optional[str] = Header(None)):
95    get_current_user(authorization)
96    if progress_id not in progress_logs:
97        raise HTTPException(status_code=404, detail="Progress log not found")
98    return progress_logs[progress_id]
99 
100@app.post("/water-reminders")
101def create_water_reminder(plot_id: int, schedule: str, authorization: Optional[str] = Header(None)):
102    user_id = get_current_user(authorization)
103    if plot_id not in plots:
104        raise HTTPException(status_code=404, detail="Plot not found")
105    global reminder_id_counter
106    reminder_id = reminder_id_counter
107    reminder_id_counter += 1
108    water_reminders[reminder_id] = {
109        "id": reminder_id,
110        "plot_id": plot_id,
111        "schedule": schedule,
112        "user_id": user_id,
113        "created_at": datetime.now().isoformat()
114    }
115    return water_reminders[reminder_id]
116 
117@app.get("/water-reminders/{reminder_id}")
118def get_water_reminder(reminder_id: int, authorization: Optional[str] = Header(None)):
119    get_current_user(authorization)
120    if reminder_id not in water_reminders:
121        raise HTTPException(status_code=404, detail="Water reminder not found")
122    return water_reminders[reminder_id]
123 
124@app.get("/plots/{plot_id}/reminders")
125def get_plot_reminders(plot_id: int, authorization: Optional[str] = Header(None)):
126    get_current_user(authorization)
127    if plot_id not in plots:
128        raise HTTPException(status_code=404, detail="Plot not found")
129    return [r for r in water_reminders.values() if r["plot_id"] == plot_id]

1fastapi
2uvicorn