Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional
4import secrets
5 
6app = FastAPI()
7 
8users = {}
9tokens = {}
10plants = {}
11swaps = {}
12watering_logs = {}
13reminders = {}
14 
15user_id_counter = 1
16plant_id_counter = 1
17swap_id_counter = 1
18log_id_counter = 1
19reminder_id_counter = 1
20 
21class SignupRequest(BaseModel):
22    username: str
23    password: str
24 
25class LoginRequest(BaseModel):
26    username: str
27    password: str
28 
29class PlantCreate(BaseModel):
30    name: str
31    species: Optional[str] = None
32    owner_id: int
33 
34class SwapCreate(BaseModel):
35    plant_id: int
36    swapper_id: int
37 
38class WateringLogCreate(BaseModel):
39    plant_id: int
40    caretaker_id: int
41 
42class ReminderCreate(BaseModel):
43    plant_id: int
44    caretaker_id: int
45    interval_hours: int
46 
47def get_user_from_token(authorization: str = Header(None)):
48    if not authorization:
49        raise HTTPException(status_code=401, detail="No auth header")
50    token = authorization.replace("Bearer ", "")
51    if token not in tokens:
52        raise HTTPException(status_code=401, detail="Invalid token")
53    return tokens[token]
54 
55@app.post("/signup")
56def signup(req: SignupRequest):
57    global user_id_counter
58    if req.username in users:
59        raise HTTPException(status_code=400, detail="Username exists")
60    user_id = user_id_counter
61    user_id_counter += 1
62    users[req.username] = {"id": user_id, "username": req.username, "password": req.password}
63    return {"id": user_id, "username": req.username}
64 
65@app.post("/login")
66def login(req: LoginRequest):
67    if req.username not in users or users[req.username]["password"] != req.password:
68        raise HTTPException(status_code=401, detail="Invalid credentials")
69    token = secrets.token_hex(16)
70    tokens[token] = users[req.username]
71    return {"token": token}
72 
73@app.get("/plants/{plant_id}")
74def get_plant(plant_id: int, authorization: str = Header(None)):
75    get_user_from_token(authorization)
76    if plant_id not in plants:
77        raise HTTPException(status_code=404, detail="Plant not found")
78    return plants[plant_id]
79 
80@app.post("/plants")
81def create_plant(plant: PlantCreate, authorization: str = Header(None)):
82    global plant_id_counter
83    user = get_user_from_token(authorization)
84    if plant.owner_id != user["id"]:
85        raise HTTPException(status_code=403, detail="Not your plant")
86    plant_id = plant_id_counter
87    plant_id_counter += 1
88    plants[plant_id] = {"id": plant_id, "name": plant.name, "species": plant.species, "owner_id": plant.owner_id, "caretaker_id": None}
89    return plants[plant_id]
90 
91@app.get("/swaps/{swap_id}")
92def get_swap(swap_id: int, authorization: str = Header(None)):
93    get_user_from_token(authorization)
94    if swap_id not in swaps:
95        raise HTTPException(status_code=404, detail="Swap not found")
96    return swaps[swap_id]
97 
98@app.post("/swaps")
99def create_swap(swap: SwapCreate, authorization: str = Header(None)):
100    global swap_id_counter
101    user = get_user_from_token(authorization)
102    if swap.swapper_id != user["id"]:
103        raise HTTPException(status_code=403, detail="Not your swap")
104    if swap.plant_id not in plants:
105        raise HTTPException(status_code=404, detail="Plant not found")
106    plant = plants[swap.plant_id]
107    if plant["owner_id"] != user["id"]:
108        raise HTTPException(status_code=403, detail="Not your plant")
109    swap_id = swap_id_counter
110    swap_id_counter += 1
111    swaps[swap_id] = {"id": swap_id, "plant_id": swap.plant_id, "swapper_id": swap.swapper_id}
112    return swaps[swap_id]
113 
114@app.post("/plants/{plant_id}/promote")
115def promote_caretaker(plant_id: int, swapper_id: int, authorization: str = Header(None)):
116    user = get_user_from_token(authorization)
117    if plant_id not in plants:
118        raise HTTPException(status_code=404, detail="Plant not found")
119    plant = plants[plant_id]
120    if plant["owner_id"] != user["id"]:
121        raise HTTPException(status_code=403, detail="Not your plant")
122    if swapper_id not in [u["id"] for u in users.values()]:
123        raise HTTPException(status_code=404, detail="Swapper not found")
124    plant["caretaker_id"] = swapper_id
125    return plant
126 
127@app.get("/watering_logs/{log_id}")
128def get_watering_log(log_id: int, authorization: str = Header(None)):
129    get_user_from_token(authorization)
130    if log_id not in watering_logs:
131        raise HTTPException(status_code=404, detail="Log not found")
132    return watering_logs[log_id]
133 
134@app.post("/watering_logs")
135def create_watering_log(log: WateringLogCreate, authorization: str = Header(None)):
136    global log_id_counter
137    user = get_user_from_token(authorization)
138    if log.plant_id not in plants:
139        raise HTTPException(status_code=404, detail="Plant not found")
140    plant = plants[log.plant_id]
141    if plant["caretaker_id"] != user["id"]:
142        raise HTTPException(status_code=403, detail="Not the caretaker")
143    log_id = log_id_counter
144    log_id_counter += 1
145    watering_logs[log_id] = {"id": log_id, "plant_id": log.plant_id, "caretaker_id": log.caretaker_id, "timestamp": "now"}
146    return watering_logs[log_id]
147 
148@app.get("/reminders/{reminder_id}")
149def get_reminder(reminder_id: int, authorization: str = Header(None)):
150    get_user_from_token(authorization)
151    if reminder_id not in reminders:
152        raise HTTPException(status_code=404, detail="Reminder not found")
153    return reminders[reminder_id]
154 
155@app.post("/reminders")
156def create_reminder(reminder: ReminderCreate, authorization: str = Header(None)):
157    global reminder_id_counter
158    user = get_user_from_token(authorization)
159    if reminder.plant_id not in plants:
160        raise HTTPException(status_code=404, detail="Plant not found")
161    plant = plants[reminder.plant_id]
162    if plant["caretaker_id"] != user["id"]:
163        raise HTTPException(status_code=403, detail="Not the caretaker")
164    reminder_id = reminder_id_counter
165    reminder_id_counter += 1
166    reminders[reminder_id] = {"id": reminder_id, "plant_id": reminder.plant_id, "caretaker_id": reminder.caretaker_id, "interval_hours": reminder.interval_hours}
167    return reminders[reminder_id]

1fastapi
2uvicorn