Фильтрация потока Python: исключения и гарантированное закрытие генератора

Шлюз очистки потока

Шлюз очистки данных принимает ленивый генератор записей и отбирает из него только те элементы, которые удовлетворяют готовому предикату схемы. Если же во входном потоке встречается специальный маркер неисправности 'CORRUPTED', шлюз сразу прерывает обработку и возбуждает исключение ValueError('corrupted stream').

При любом завершении работы — будь то штатное исчерпание записей, аварийный маркер или прямой вызов close() у самого шлюза — валидатор обязан в блоке finally вызвать close() у входящего генератора. Однако процедура очистки стороннего источника может сама завершиться ошибкой: в таком случае валидатор должен подавить вторичное исключение стороннего close(), сохраняя первичное исключение потока или нормальный выход без подмены причины.

Реализуйте генератор-валидатор, который фильтрует поток по схеме, вовремя реагирует на маркер сбоя и гарантированно закрывает исходный генератор с защитой от ошибок его финализации.

Что уже дано

В среде выполнения доступны вспомогательные фабрики источников и проверочные инструменты (например, make_validation_source), которые используются в тестах и примере. Реализовывать их заново не требуется: вам нужно написать только саму генераторную функцию validate_stream.

Что нужно сделать

Реализуйте генераторную функцию validate_stream(source, predicate).

Функция принимает два аргумента:

• source — однократный итератор, содержащий от 0 до 1000 записей (числа int/float в диапазоне от -1 000 000 до 1 000 000 или строки длиной от 0 до 100 символов, включая их обычные подклассы). Источник предоставляет метод close(), освобождающий базовый ресурс.
• predicate — вызываемый объект (функция), истинность результата которого определяет допуск обычной записи.

Функция должна возвращать неначатный (unstarted) нативный генератор Python со следующим поведением:

• Создание генератора не выполняет чтений из source и не вызывает predicate. Если созданный генератор закрывается вызовом метода close() до первого запроса значения, тело генератора не начинает выполняться, а метод source.close() не вызывается.
• Чтение из source происходит строго по требованию: ровно настолько, насколько необходимо для выдачи очередного прошедшего проверку значения. Каждая подходящая запись отдаётся без изменений через yield с сохранением исходного порядка и дубликатов.
• Если очередная запись из source равна литералу 'CORRUPTED' (включая строковые подклассы), генератор обязан возбудить исключение ValueError (или его подкласс) с точным сообщением 'corrupted stream' ещё до вызова predicate и навсегда прекратить чтение.
• Для обычных записей вызывается predicate(record). Решение о пропуске принимается по логической истинности (bool(predicate(record))), не требуя строгого объекта True.
• При любом выходе из начатого генератора (штатное исчерпание source, маркер 'CORRUPTED', сбой при чтении или в предикате, исключение со стороны вызывающего кода или закрытие активного генератора) необходимо в блоке finally вызвать source.close().
• Если вызов source.close() возбуждает любое исключение (включая подклассы BaseException), это исключение должно быть полностью подавлено, чтобы не подменять первичное исключение источника, предиката, потребителя или ошибку повреждения данных. Если завершение было штатным, подавление исключения закрытия оставляет выход нормальным.
• Во время финализации запрещено выполнять yield или предпринимать повторные попытки чтения из source.

Ввод и вывод

Стандартные потоки ввода stdin и вывода stdout в решении не используются: считывать ввод и вызывать print() не нужно. После выполнения кода учащегося проверяющая система считывает из пространства имён модуля функцию верхнего уровня validate_stream и вызывает её с проверочными аргументами.

Пример использования

source = make_validation_source([-1, 0, 2], close_failure='exception')
print(list(validate_stream(source, lambda value: value >= 0)))

Вывод:

[0, 2]