example.py

"""Example script."""


def main() -> int:
    """Main function."""
    order_totals = []
    for order_id in range(120):
        item_count = 12 + (order_id % 9)
        subtotal = estimate_order_subtotal(order_id, item_count)
        route_score = estimate_delivery_route(order_id, zone_count=7)
        order_totals.append(subtotal + route_score)
    return sum(order_totals)


def estimate_order_subtotal(order_id: int, item_count: int) -> int:
    """Estimate the subtotal of an order."""
    subtotal = 0
    for item_idx in range(item_count):
        base_price = 15 + ((order_id * 3 + item_idx * 5) % 45)
        quantity = 1 + ((order_id + item_idx) % 4)
        discount = (item_idx % 3) * 2
        line_total = max(5, base_price - discount) * quantity
        subtotal += line_total
        if line_total > 120:
            subtotal -= 3
    return subtotal


def calculate_congestion_penalty(order_id: int, step: int, next_zone: int) -> int:
    """Calculate the congestion penalty for an order."""
    penalty = 0
    for minute in range(5):
        wave = (step + minute + order_id) % 11
        penalty += (wave * (next_zone + 1)) % 13
    return penalty


def estimate_delivery_route(order_id: int, zone_count: int) -> int:
    """Estimate the delivery route for an order."""
    score = 0
    last_zone = order_id % zone_count
    for step in range(300):
        next_zone = (last_zone * 2 + step + order_id) % zone_count
        traffic_factor = 1 + ((step + order_id) % 5)
        zone_distance = 8 + (next_zone * 3 + step % 4)
        score += zone_distance * traffic_factor
        score += calculate_congestion_penalty(order_id, step, next_zone)
        if next_zone == last_zone:
            score += 6
        else:
            score -= 1
        last_zone = next_zone
    return score


if __name__ == "__main__":
    PROFILE = True
    PROFILE_DIR = "profile"
    if PROFILE:
        from line_profiler import LineProfiler

        from line_profiler_web import save_stats

        lp = LineProfiler()
        lp.add_function(main)
        lp.add_function(estimate_order_subtotal)
        lp.add_function(estimate_delivery_route)
        lp.add_function(calculate_congestion_penalty)
        lp.run("main()")  # type: ignore[no-untyped-call]
        save_stats(lp, PROFILE_DIR)
    else:
        main()