Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional, Dict, List
4from datetime import date, timedelta
5import uuid
6import uvicorn
7 
8app = FastAPI()
9 
10users = {}
11equipment = {}
12rentals = {}
13next_user_id = 1
14next_equipment_id = 1
15next_rental_id = 1
16tokens = {}
17 
18class SignupRequest(BaseModel):
19    username: str
20    password: str
21 
22class LoginRequest(BaseModel):
23    username: str
24    password: str
25 
26class EquipmentCreate(BaseModel):
27    name: str
28    daily_rate: float
29    owner: str
30 
31class RentalCreate(BaseModel):
32    equipment_id: int
33    start_date: date
34    end_date: date
35 
36def get_current_user(token: str = Header(...)):
37    if token not in tokens:
38        raise HTTPException(status_code=401, detail="Invalid token")
39    return tokens[token]
40 
41@app.post("/signup")
42def signup(req: SignupRequest):
43    global next_user_id
44    user_id = next_user_id
45    next_user_id += 1
46    users[user_id] = {"id": user_id, "username": req.username, "password": req.password}
47    token = str(uuid.uuid4())
48    tokens[token] = user_id
49    return {"user_id": user_id, "token": token}
50 
51@app.post("/login")
52def login(req: LoginRequest):
53    for uid, u in users.items():
54        if u["username"] == req.username and u["password"] == req.password:
55            token = str(uuid.uuid4())
56            tokens[token] = uid
57            return {"token": token}
58    raise HTTPException(status_code=401, detail="Invalid credentials")
59 
60@app.get("/equipment")
61def list_equipment():
62    result = []
63    for eid, e in equipment.items():
64        availability = {}
65        today = date.today()
66        for d in range(30):
67            day = today + timedelta(days=d)
68            available = True
69            for rid, r in rentals.items():
70                if r["equipment_id"] == eid and r["start_date"] <= day <= r["end_date"]:
71                    available = False
72                    break
73            availability[day.isoformat()] = available
74        result.append({
75            "id": eid,
76            "name": e["name"],
77            "daily_rate": e["daily_rate"],
78            "owner": e["owner"],
79            "availability_calendar": availability
80        })
81    return result
82 
83@app.post("/equipment")
84def create_equipment(req: EquipmentCreate):
85    global next_equipment_id
86    eid = next_equipment_id
87    next_equipment_id += 1
88    equipment[eid] = {"id": eid, "name": req.name, "daily_rate": req.daily_rate, "owner": req.owner}
89    return {"id": eid}
90 
91@app.get("/equipment/{equipment_id}")
92def get_equipment(equipment_id: int):
93    if equipment_id not in equipment:
94        raise HTTPException(status_code=404, detail="Equipment not found")
95    e = equipment[equipment_id]
96    availability = {}
97    today = date.today()
98    for d in range(30):
99        day = today + timedelta(days=d)
100        available = True
101        for rid, r in rentals.items():
102            if r["equipment_id"] == equipment_id and r["start_date"] <= day <= r["end_date"]:
103                available = False
104                break
105        availability[day.isoformat()] = available
106    return {
107        "id": equipment_id,
108        "name": e["name"],
109        "daily_rate": e["daily_rate"],
110        "owner": e["owner"],
111        "availability_calendar": availability
112    }
113 
114@app.post("/rentals")
115def create_rental(req: RentalCreate, token: str = Header(...)):
116    global next_rental_id
117    user_id = get_current_user(token)
118    if req.equipment_id not in equipment:
119        raise HTTPException(status_code=404, detail="Equipment not found")
120    if req.start_date > req.end_date:
121        raise HTTPException(status_code=400, detail="Start date must be before end date")
122    for rid, r in rentals.items():
123        if r["equipment_id"] == req.equipment_id:
124            if not (req.end_date < r["start_date"] or req.start_date > r["end_date"]):
125                raise HTTPException(status_code=400, detail="Equipment not available for those dates")
126    rid = next_rental_id
127    next_rental_id += 1
128    rentals[rid] = {
129        "id": rid,
130        "user_id": user_id,
131        "equipment_id": req.equipment_id,
132        "start_date": req.start_date,
133        "end_date": req.end_date
134    }
135    return {"id": rid}
136 
137@app.get("/rentals/active")
138def get_active_rentals(token: str = Header(...)):
139    user_id = get_current_user(token)
140    result = []
141    today = date.today()
142    for rid, r in rentals.items():
143        if r["user_id"] == user_id and r["end_date"] >= today:
144            result.append({
145                "id": rid,
146                "equipment_id": r["equipment_id"],
147                "equipment_name": equipment[r["equipment_id"]]["name"],
148                "start_date": r["start_date"],
149                "end_date": r["end_date"],
150                "return_date": r["end_date"]
151            })
152    return result
153 
154@app.get("/rentals/{rental_id}")
155def get_rental(rental_id: int, token: str = Header(...)):
156    if rental_id not in rentals:
157        raise HTTPException(status_code=404, detail="Rental not found")
158    r = rentals[rental_id]
159    return {
160        "id": rental_id,
161        "user_id": r["user_id"],
162        "equipment_id": r["equipment_id"],
163        "start_date": r["start_date"],
164        "end_date": r["end_date"]
165    }

1fastapi
2uvicorn