Design Uber/Ride Sharing

import redis

r = redis.Redis()

# Driver location update (প্রতি 4 sec)
def update_driver_location(driver_id: str, lat: float, lng: float):
    # Redis GEO: GEOADD key longitude latitude member
    r.geoadd("drivers:active", {driver_id: (lng, lat)})
    # Driver status update
    r.hset(f"driver:{driver_id}", mapping={
        "lat": lat, "lng": lng,
        "status": "available",
        "updated_at": now()
    })

# Rider request — nearby drivers খুঁজুন
def find_nearby_drivers(rider_lat: float, rider_lng: float, radius_km: float = 5):
    # Redis GEORADIUS: center থেকে radius-এর মধ্যে সব drivers
    nearby = r.georadius(
        "drivers:active",
        rider_lng, rider_lat,
        radius_km,
        unit='km',
        withcoord=True,
        withdist=True,
        sort='ASC',  # Closest first
        count=10     # Top 10 closest
    )
    return [
        {"driver_id": d[0], "distance_km": d[1], "location": d[2]}
        for d in nearby
    ]

# Example output:
# [{"driver_id": "D123", "distance_km": 0.8, "location": (90.4, 23.8)},
#  {"driver_id": "D456", "distance_km": 1.2, ...}]

def match_driver(rider_request, nearby_drivers):
    def score_driver(driver):
        # Multi-factor scoring
        distance_score = 1.0 / (driver["distance_km"] + 0.1)
        rating_score = driver["rating"] / 5.0
        acceptance_score = driver["acceptance_rate"] / 100.0

        # Weighted final score
        return (
            distance_score   * 0.5 +  # 50% weight — distance most important
            rating_score     * 0.3 +  # 30% weight
            acceptance_score * 0.2    # 20% weight
        )

    # Car type filter করুন আগে
    compatible = [
        d for d in nearby_drivers
        if d["car_type"] == rider_request["ride_type"]
    ]

    if not compatible:
        raise NoDriverAvailable()

    # Best score = best match
    best_driver = max(compatible, key=score_driver)

    # Driver-কে request push করুন (WebSocket)
    websocket.push(best_driver["driver_id"], {
        "type": "RIDE_REQUEST",
        "rider": rider_request,
        "expires_in": 15  # 15 seconds to accept
    })
    return best_driver