Session window gap mock in Python

from datetime import datetime, timedelta


def session_windows(timestamps, gap_seconds=300):
    """Group timestamps into sessions where gaps > gap_seconds start new sessions."""
    if not timestamps:
        return []

    # Sort timestamps chronologically to ensure correct windowing
    timestamps = sorted(timestamps)
    sessions = [[timestamps[0]]]
    session_start = timestamps[0]

    for ts in timestamps[1:]:
        if (ts - session_start).total_seconds() > gap_seconds:
            # Start a new session if gap from session start exceeds threshold
            sessions.append([ts])
            session_start = ts
        else:
            # Same session, append to current
            sessions[-1].append(ts)

    return sessions


if __name__ == "__main__":
    base = datetime(2025, 1, 1, 10, 0, 0)
    mock_timestamps = [
        base,
        base + timedelta(minutes=2),   # same session
        base + timedelta(minutes=10),  # > 5 min gap → new session
        base + timedelta(minutes=11),  # same as previous
        base + timedelta(hours=1),     # > 5 min gap → new session
    ]

    for i, session in enumerate(session_windows(mock_timestamps, gap_seconds=300), 1):
        print(f"Session {i}: {[ts.strftime('%H:%M:%S') for ts in session]}")