Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from datetime import datetime, timedelta
3import hashlib
4import secrets
5 
6app = FastAPI()
7 
8users = {}
9tokens = {}
10orders = {}
11shipments = {}
12delays = {}
13next_user_id = 1
14next_token_id = 1
15next_order_id = 1
16next_shipment_id = 1
17next_delay_id = 1
18 
19def hash_password(password):
20    return hashlib.sha256(password.encode()).hexdigest()
21 
22def generate_token():
23    return secrets.token_hex(32)
24 
25def get_current_user(authorization: str = Header(None)):
26    if not authorization:
27        raise HTTPException(status_code=401, detail="Missing Authorization header")
28    token = authorization.replace("Bearer ", "")
29    for uid, t in tokens.items():
30        if t == token:
31            return uid
32    raise HTTPException(status_code=401, detail="Invalid token")
33 
34@app.post("/signup")
35def signup(username: str, password: str):
36    global next_user_id
37    if username in users:
38        raise HTTPException(status_code=400, detail="Username already exists")
39    user_id = next_user_id
40    next_user_id += 1
41    users[username] = {"id": user_id, "password": hash_password(password)}
42    return {"id": user_id, "username": username}
43 
44@app.post("/login")
45def login(username: str, password: str):
46    if username not in users:
47        raise HTTPException(status_code=400, detail="Invalid credentials")
48    if users[username]["password"] != hash_password(password):
49        raise HTTPException(status_code=400, detail="Invalid credentials")
50    token = generate_token()
51    tokens[users[username]["id"]] = token
52    return {"token": token}
53 
54@app.post("/orders")
55def create_order(destination: str, current_location: str, eta: str, authorization: str = Header(None)):
56    get_current_user(authorization)
57    global next_order_id
58    order_id = next_order_id
59    next_order_id += 1
60    orders[order_id] = {
61        "id": order_id,
62        "destination": destination,
63        "current_location": current_location,
64        "eta": eta,
65        "created_at": datetime.now().isoformat()
66    }
67    return orders[order_id]
68 
69@app.get("/orders/{order_id}")
70def get_order(order_id: int, authorization: str = Header(None)):
71    get_current_user(authorization)
72    if order_id not in orders:
73        raise HTTPException(status_code=404, detail="Order not found")
74    return orders[order_id]
75 
76@app.get("/shipments")
77def get_shipments(authorization: str = Header(None)):
78    get_current_user(authorization)
79    result = []
80    for oid, order in orders.items():
81        result.append({
82            "order_id": oid,
83            "current_location": order["current_location"],
84            "eta": order["eta"]
85        })
86    return result
87 
88@app.post("/update-location")
89def update_location(order_id: int, latitude: float, longitude: float, authorization: str = Header(None)):
90    get_current_user(authorization)
91    if order_id not in orders:
92        raise HTTPException(status_code=404, detail="Order not found")
93    global next_shipment_id
94    shipment_id = next_shipment_id
95    next_shipment_id += 1
96    shipments[shipment_id] = {
97        "id": shipment_id,
98        "order_id": order_id,
99        "latitude": latitude,
100        "longitude": longitude,
101        "timestamp": datetime.now().isoformat()
102    }
103    orders[order_id]["current_location"] = f"{latitude},{longitude}"
104    return shipments[shipment_id]
105 
106@app.get("/delays")
107def get_delays(authorization: str = Header(None)):
108    get_current_user(authorization)
109    result = []
110    now = datetime.now()
111    for oid, order in orders.items():
112        try:
113            eta_dt = datetime.fromisoformat(order["eta"])
114        except:
115            continue
116        if eta_dt < now - timedelta(hours=2):
117            result.append({
118                "order_id": oid,
119                "eta": order["eta"],
120                "current_location": order["current_location"]
121            })
122    return result

1fastapi
2uvicorn