How to Order Partition Key Events in Python (Mock Stream)

import itertools
import random


def partition_key_events(keys, events_per_key=3, seed=None):
    """Produce a realistic-looking, but mock, event stream grouped by partition key.

    Args:
        keys: iterable of partition keys (e.g. strings or ints).
        events_per_key: how many events we want per key.
        seed: optional RNG seed for repeatable output.

    Returns:
        List of (partition_key, sequence, event_id) tuples.
    """
    if seed is not None:
        random.seed(seed)

    events = []
    for key in keys:
        for seq in range(1, events_per_key + 1):
            event_id = random.randint(1_000_000, 9_999_999)
            events.append((key, seq, event_id))
    return events


def order_partition_events(events):
    """Sort events deterministically: partition key first, then sequence."""

    return sorted(events, key=lambda ev: (str(ev[0]), ev[1]))


if __name__ == "__main__":
    mock_stream = partition_key_events(["user-a", "user-b", "user-c"], events_per_key=2, seed=42)
    print("Original (shuffled) stream:")
    random.shuffle(mock_stream)
    for e in mock_stream:
        print(e)

    print("\nOrdered by partition key (then sequence):")
    ordered = order_partition_events(mock_stream)
    for e in ordered:
        print(e)