Исправить или переписать скрипт на Python библиотека Polars

Добрый день! Есть часть скрипта, который отвечает за подбор документов.
Что он делает: проходи по документам "прихода" и "расхода". На документы "прихода" ставит остаток, а к документам "расхода" должен подобрать документ "прихода". Если к "расходу" подходит 2 и более "приходов" строка разбивается по количеству остатка в "приходе".
Скрипт работает не совсем корректно. Когда товар продан и остаток по нему 0, а позже произошел возврат товара, скрипт данный факт пропускает и просто не проходит по этим строчкам. Возможно, он еще и не очень оптимально написан, т.к. постоянно возвращается к предыдущей строке.
В вложенных файлах:
- Excel - лист "Комплектации (вход)" - входные данные для скрипта; лист "РЕЗУЛЬТАТ" - то, что должно получиться после отработки скрипта
- Word - полное ТЗ и ссылка на gist, где выложен кусочек кода.