如何在Python循环中管理副作用？实现通用化副作用处理逻辑

嘿，这个需求我太有共鸣了！把循环里的副作用逻辑（比如这种按需打印的规则）抽出来复用，确实能让代码清爽不少。用上下文管理器来封装这个跟踪逻辑是个绝佳的思路，我给你写个实用的实现，你可以直接套用或者按需调整：

先上核心代码，用contextlib.contextmanager来简化上下文管理器的编写：

from contextlib import contextmanager

@contextmanager
def tracked_printer(show_first=10, show_last=1):
    """
    上下文管理器，负责跟踪迭代过程并按规则输出条目：
    - 打印前`show_first`个条目
    - 如果总条目数超过`show_first + show_last`，中间打印省略号
    - 最后打印`show_last`个条目
    """
    # 存储所有迭代过的条目，用于收尾处理
    collected_items = []
    
    def process_item(item):
        """循环内部调用的处理函数，负责即时打印前N个条目"""
        collected_items.append(item)
        if len(collected_items) <= show_first:
            print(item)
    
    try:
        # 将处理函数交给外部循环使用
        yield process_item
    finally:
        # 迭代结束后，处理收尾逻辑
        total_count = len(collected_items)
        if total_count > show_first + show_last:
            print("...")
            # 打印最后show_last个条目
            for item in collected_items[-show_last:]:
                print(item)
        elif total_count > show_first:
            # 如果总条目数介于show_first和show_first+show_last之间，直接打印剩余的
            for item in collected_items[show_first:]:
                print(item)

用法示例 #

你可以在任何需要这个打印规则的循环里直接用：

# 示例1：遍历15个元素，打印前3个、最后2个，中间用省略号
with tracked_printer(show_first=3, show_last=2) as print_item:
    for num in range(15):
        print_item(num)

# 输出结果：
# 0
# 1
# 2
# ...
# 13
# 14

# 示例2：如果条目数很少，比如只有5个，不会出现省略号
with tracked_printer(show_first=3, show_last=2) as print_item:
    for num in range(5):
        print_item(num)

# 输出结果：
# 0
# 1
# 2
# 3
# 4

扩展思路 #

这个上下文管理器很容易扩展成其他副作用逻辑，比如替换成日志记录、统计计数等：

import logging
logging.basicConfig(level=logging.INFO)

@contextmanager
def tracked_logger(show_first=5, show_last=2):
    collected = []
    count = 0
    
    def log_item(item):
        nonlocal count
        count +=1
        collected.append(item)
        if count <= show_first:
            logging.info(f"处理第{count}个条目: {item}")
    
    yield log_item
    finally:
        logging.info(f"共处理{count}个条目")
        if count > show_first + show_last:
            logging.info("...")
            for idx, item in enumerate(collected[-show_last:], start=count-show_last+1):
                logging.info(f"处理第{idx}个条目: {item}")

# 调用示例
with tracked_logger() as log_item:
    for num in range(12):
        log_item(f"数据-{num}")

这种方式的好处是，主循环只需要专注于业务逻辑，副作用（打印/日志）完全由上下文管理器接管，而且可以在多个项目里直接复用这个管理器。

内容的提问来源于stack exchange，提问作者Matt Ko