Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional
4import random
5import string
6 
7app = FastAPI()
8 
9users = {}
10tokens = {}
11reports = {}
12report_id_counter = 1
13 
14def generate_token():
15    return ''.join(random.choices(string.ascii_letters + string.digits, k=32))
16 
17def get_current_user(authorization: Optional[str] = Header(None)):
18    if not authorization:
19        raise HTTPException(status_code=401, detail="Missing auth header")
20    token = authorization.replace("Bearer ", "")
21    user_id = tokens.get(token)
22    if not user_id:
23        raise HTTPException(status_code=401, detail="Invalid token")
24    return user_id
25 
26class SignupRequest(BaseModel):
27    username: str
28    password: str
29 
30class LoginRequest(BaseModel):
31    username: str
32    password: str
33 
34class ReportCreate(BaseModel):
35    title: str
36    description: str
37    severity: str
38    status: str = "open"
39    latitude: float
40    longitude: float
41 
42class ReportAttach(BaseModel):
43    image_data: str
44    latitude: float
45    longitude: float
46 
47@app.post("/signup")
48def signup(req: SignupRequest):
49    if req.username in users:
50        raise HTTPException(status_code=400, detail="User exists")
51    users[req.username] = req.password
52    token = generate_token()
53    tokens[token] = req.username
54    return {"token": token}
55 
56@app.post("/login")
57def login(req: LoginRequest):
58    stored = users.get(req.username)
59    if not stored or stored != req.password:
60        raise HTTPException(status_code=401, detail="Invalid credentials")
61    token = generate_token()
62    tokens[token] = req.username
63    return {"token": token}
64 
65@app.get("/reports/{report_id}")
66def get_report(report_id: int, authorization: Optional[str] = Header(None)):
67    get_current_user(authorization)
68    report = reports.get(report_id)
69    if not report:
70        raise HTTPException(status_code=404, detail="Report not found")
71    return report
72 
73@app.post("/reports")
74def create_report(report: ReportCreate, authorization: Optional[str] = Header(None)):
75    user = get_current_user(authorization)
76    global report_id_counter
77    new_id = report_id_counter
78    report_id_counter += 1
79    reports[new_id] = {
80        "id": new_id,
81        "title": report.title,
82        "description": report.description,
83        "severity": report.severity,
84        "status": report.status,
85        "latitude": report.latitude,
86        "longitude": report.longitude,
87        "created_by": user
88    }
89    return reports[new_id]
90 
91@app.get("/reports/nearby")
92def get_nearby_reports(lat: float, lng: float, radius_km: float = 1.0, authorization: Optional[str] = Header(None)):
93    get_current_user(authorization)
94    nearby = []
95    for report in reports.values():
96        if report.get("latitude") and report.get("longitude"):
97            dlat = report["latitude"] - lat
98            dlng = report["longitude"] - lng
99            dist_km = (dlat**2 + dlng**2)**0.5 * 111
100            if dist_km <= radius_km:
101                nearby.append(report)
102    return nearby
103 
104@app.post("/reports/attach")
105def attach_report(attach: ReportAttach, authorization: Optional[str] = Header(None)):
106    user = get_current_user(authorization)
107    global report_id_counter
108    new_id = report_id_counter
109    report_id_counter += 1
110    reports[new_id] = {
111        "id": new_id,
112        "title": "Attached image",
113        "description": "",
114        "severity": "unknown",
115        "status": "open",
116        "latitude": attach.latitude,
117        "longitude": attach.longitude,
118        "image_data": attach.image_data,
119        "created_by": user
120    }
121    return reports[new_id]

1fastapi
2uvicorn
3pydantic