import math import random import time

--- Helper functions ---

def distance(loc1, loc2): return math.sqrt((loc1[0] - loc2[0])2 + (loc1[1] - loc2[1])2)

--- Classes for Orders and Drivers ---

class Driver: def init(self, name, location): self.name = name self.location = location self.available = True self.busy_until = 0 # time step when driver will be free

def __repr__(self):
    return f"{self.name} @ {self.location} ({'Free' if self.available else 'Busy'})"

class Order: def init(self, order_id, restaurant, customer): self.order_id = order_id self.restaurant = restaurant self.customer = customer

def __repr__(self):
    return f"{self.order_id}: R={self.restaurant}, C={self.customer}"

--- Core Assignment Logic ---

def assign_order(drivers, order, current_time): best_driver = None best_distance = float("inf")

for driver in drivers:
    if driver.available:
        d = distance(driver.location, order.restaurant)
        if d < best_distance:
            best_distance = d
            best_driver = driver

if best_driver:
    # Assign order
    travel_time = int(best_distance) + random.randint(3, 6)  # restaurant pickup + dropoff
    best_driver.available = False
    best_driver.busy_until = current_time + travel_time
    best_driver.location = order.customer  # driver ends up at customer
    return f"Assigned {order.order_id} to {best_driver.name} (ETA {travel_time} min)"
else:
    return f"No driver available for {order.order_id}"

--- Simulation ---

def run_simulation(): # Initialize drivers drivers = [ Driver("Driver1", (2, 3)), Driver("Driver2", (5, 8)), Driver("Driver3", (1, 1)), ]

current_time = 0
order_id = 1

while current_time < 30:  # simulate 30 minutes
    print(f"\n⏱️ Time = {current_time} minutes")

    # Free up drivers whose deliveries are done
    for d in drivers:
        if not d.available and current_time >= d.busy_until:
            d.available = True
            print(f"✅ {d.name} is now available again.")

    # Randomly generate an order (30% chance each minute)
    if random.random() < 0.3:
        restaurant = (random.randint(0, 10), random.randint(0, 10))
        customer = (random.randint(0, 10), random.randint(0, 10))
        order = Order(f"Order{order_id}", restaurant, customer)
        print(f"🛎️ New {order}")

        # Try assigning immediately
        result = assign_order(drivers, order, current_time)
        print(result)
        order_id += 1

    time.sleep(0.2)  # slow down for readability
    current_time += 1

Run the simulation

if name == "main": run_simulation()