Python中如何优雅地重写父类方法并修改部分逻辑？

如题

我自己的下载 pipeline，继承了 scrapy 的 FilesPipeline 类，我现在想修改 media_to_download 这个方法

其实就是想加一点 log，逻辑上有一点小小的修改，大体上的逻辑还是 scrapy 中的逻辑

我只会复制粘贴 FilesPipeline 中的方法代码过来，然后自己修改

有没有什么比较好的方法能做到加一点 log 这个要求吗？
Python中如何优雅地重写父类方法并修改部分逻辑？

bupafengyu 1楼

装饰器？

ionicwang 2楼

class Parent:
    def process_data(self, data):
        """父类处理逻辑"""
        # 基础验证
        if not data:
            raise ValueError("数据不能为空")
        
        # 核心处理逻辑
        result = []
        for item in data:
            processed = self._transform_item(item)
            result.append(processed)
        
        return self._finalize(result)
    
    def _transform_item(self, item):
        """转换单个数据项（可被子类重写）"""
        return str(item).upper()
    
    def _finalize(self, result):
        """最终处理（可被子类重写）"""
        return list(filter(None, result))


class Child(Parent):
    def process_data(self, data):
        """重写父类方法，修改部分逻辑"""
        # 1. 先执行父类的验证逻辑
        if not data:
            raise ValueError("数据不能为空")
        
        # 2. 添加子类特有的预处理
        filtered_data = [x for x in data if x is not None]
        
        # 3. 调用父类核心逻辑，但使用子类的_transform_item
        result = []
        for item in filtered_data:
            processed = self._transform_item(item)  # 这里会调用子类重写的方法
            result.append(processed)
        
        # 4. 修改最终处理逻辑
        return sorted(result)  # 子类添加排序功能
    
    def _transform_item(self, item):
        """重写转换逻辑"""
        # 调用父类方法获取基础转换
        base_result = super()._transform_item(item)
        # 添加子类特有的转换
        return f"processed: {base_result}"


# 使用示例
if __name__ == "__main__":
    parent = Parent()
    child = Child()
    
    test_data = ["hello", None, "world", "python"]
    
    print("父类处理结果:", parent.process_data(test_data))
    # 输出: ['HELLO', 'WORLD', 'PYTHON']
    
    print("子类处理结果:", child.process_data(test_data))
    # 输出: ['processed: HELLO', 'processed: PYTHON', 'processed: WORLD']

关键点：