Function format_number_ranges

agents/feedback-agent/main.py:344–367

Format selected numbers as compact ranges such as 01-03, 07, 09.

(numbers: Set[int])

def format_number_ranges(numbers: Set[int]) -> str:
    """Format selected numbers as compact ranges such as 01-03, 07, 09."""
    if not numbers:
        return "无"

    sorted_numbers = sorted(numbers)
    ranges = []
    start = end = sorted_numbers[0]

    for number in sorted_numbers[1:]:
        if number == end + 1:
            end = number
            continue
        ranges.append((start, end))
        start = end = number
    ranges.append((start, end))

    formatted = []
    for start, end in ranges:
        if start == end:
            formatted.append(f"{start:02d}")
        else:
            formatted.append(f"{start:02d}-{end:02d}")
    return ", ".join(formatted)


def get_existing_selected_numbers(user_id: str, push_id: str, papers: List[Dict]) -> Set[int]:
    """Return paper numbers already selected for this user/push.