Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from datetime import datetime, date
4from typing import Optional
5import uuid
6import secrets
7 
8app = FastAPI()
9 
10users = {}
11tokens = {}
12sittings = {}
13sitting_id_counter = 1
14 
15class SignupRequest(BaseModel):
16    username: str
17    password: str
18 
19class LoginRequest(BaseModel):
20    username: str
21    password: str
22 
23class SittingCreate(BaseModel):
24    pet_name: str
25    owner_contact: str
26    start_date: date
27    end_date: date
28 
29class SittingUpdate(BaseModel):
30    pet_name: Optional[str] = None
31    owner_contact: Optional[str] = None
32    start_date: Optional[date] = None
33    end_date: Optional[date] = None
34 
35@app.post("/signup")
36def signup(req: SignupRequest):
37    if req.username in users:
38        raise HTTPException(status_code=400, detail="User already exists")
39    users[req.username] = {"password": req.password}
40    token = secrets.token_hex(16)
41    tokens[token] = req.username
42    return {"token": token}
43 
44@app.post("/login")
45def login(req: LoginRequest):
46    user = users.get(req.username)
47    if not user or user["password"] != req.password:
48        raise HTTPException(status_code=401, detail="Invalid credentials")
49    token = secrets.token_hex(16)
50    tokens[token] = req.username
51    return {"token": token}
52 
53def get_current_user(authorization: str = Header(...)):
54    token = authorization.replace("Bearer ", "")
55    username = tokens.get(token)
56    if not username:
57        raise HTTPException(status_code=401, detail="Invalid token")
58    return username
59 
60@app.get("/sittings/{sitting_id}")
61def get_sitting(sitting_id: int, authorization: str = Header(...)):
62    get_current_user(authorization)
63    sitting = sittings.get(sitting_id)
64    if not sitting:
65        raise HTTPException(status_code=404, detail="Sitting not found")
66    return sitting
67 
68@app.post("/sittings")
69def create_sitting(sitting: SittingCreate, authorization: str = Header(...)):
70    global sitting_id_counter
71    get_current_user(authorization)
72    new_sitting = {
73        "id": sitting_id_counter,
74        "pet_name": sitting.pet_name,
75        "owner_contact": sitting.owner_contact,
76        "start_date": sitting.start_date.isoformat(),
77        "end_date": sitting.end_date.isoformat(),
78        "created_at": datetime.utcnow().isoformat()
79    }
80    sittings[sitting_id_counter] = new_sitting
81    sitting_id_counter += 1
82    return new_sitting
83 
84@app.get("/sittings")
85def list_sittings(authorization: str = Header(...)):
86    get_current_user(authorization)
87    today = date.today()
88    upcoming = [s for s in sittings.values() if s["start_date"] >= today.isoformat()]
89    upcoming.sort(key=lambda s: s["start_date"])
90    return upcoming
91 
92@app.put("/sittings/{sitting_id}")
93def update_sitting(sitting_id: int, sitting: SittingUpdate, authorization: str = Header(...)):
94    get_current_user(authorization)
95    existing = sittings.get(sitting_id)
96    if not existing:
97        raise HTTPException(status_code=404, detail="Sitting not found")
98    if sitting.pet_name is not None:
99        existing["pet_name"] = sitting.pet_name
100    if sitting.owner_contact is not None:
101        existing["owner_contact"] = sitting.owner_contact
102    if sitting.start_date is not None:
103        existing["start_date"] = sitting.start_date.isoformat()
104    if sitting.end_date is not None:
105        existing["end_date"] = sitting.end_date.isoformat()
106    return existing
107 
108@app.delete("/sittings/{sitting_id}")
109def delete_sitting(sitting_id: int, authorization: str = Header(...)):
110    get_current_user(authorization)
111    if sitting_id not in sittings:
112        raise HTTPException(status_code=404, detail="Sitting not found")
113    del sittings[sitting_id]
114    return {"ok": True}

1fastapi
2uvicorn