如何控制Python的+=运算符，实现自定义映射值的更新逻辑？

一、元组+=能写回字典的原因 #

Python对x += y的处理逻辑是：

1. 优先调用x.__iadd__(y)（in-place加法），如果对象支持该方法则直接修改原对象。
2. 若对象不支持__iadd__（比如元组这种不可变类型），则退化为x = x + y：先通过__add__生成新对象，再将新对象赋值回原位置。

对于d['a'] += (3,4)，这里的d['a']是字典的元素引用，Python的处理步骤是：

• 调用d.__getitem__('a')取出元组(1,2)
• 因为元组无__iadd__，执行(1,2) + (3,4)得到新元组(1,2,3,4)
• 自动调用d.__setitem__('a', 新元组)，把结果写回字典对应键

本质上，d['a'] += y等价于：

temp = d['a']
temp = temp + y
d['a'] = temp

这是Python对下标操作+=的统一处理，无论元素是否可变，都会触发后续的__setitem__完成写回。

二、自定义Files类的+=追加逻辑 #

要实现files['foo.txt'] += b'man'直接追加文件而非读-拼-写，推荐用代理对象方案，既不破坏正常取值行为，又能拦截+=操作：

实现代码 #

from collections.abc import MutableMapping
import os

class FileAppendProxy(bytes):
    def __new__(cls, files_instance, key):
        # 继承bytes，复用其所有原生方法
        content = files_instance._get_file_content(key)
        self = super().__new__(cls, content)
        self.files = files_instance
        self.key = key
        return self
    
    def __iadd__(self, other):
        # 拦截+=操作，执行文件追加逻辑
        if not isinstance(other, bytes):
            raise TypeError("仅支持追加bytes类型内容")
        file_path = os.path.join(self.files.root_dir, self.key)
        with open(file_path, 'ab') as f:
            f.write(other)
        # 返回更新后的代理对象，保持链式调用能力
        return FileAppendProxy(self.files, self.key)

class Files(MutableMapping):
    def __init__(self, root_dir):
        self.root_dir = os.path.expanduser(root_dir)
        os.makedirs(self.root_dir, exist_ok=True)
    
    def _get_file_content(self, key):
        # 内部方法：读取文件内容
        file_path = os.path.join(self.root_dir, key)
        if not os.path.exists(file_path):
            raise KeyError(key)
        with open(file_path, 'rb') as f:
            return f.read()
    
    def __getitem__(self, key):
        # 返回代理对象，而非直接返回bytes
        return FileAppendProxy(self, key)
    
    def __setitem__(self, key, value):
        # 普通赋值：覆盖写入文件
        if not isinstance(value, bytes):
            raise TypeError("文件内容必须为bytes类型")
        file_path = os.path.join(self.root_dir, key)
        with open(file_path, 'wb') as f:
            f.write(value)
    
    def __delitem__(self, key):
        file_path = os.path.join(self.root_dir, key)
        if not os.path.exists(file_path):
            raise KeyError(key)
        os.remove(file_path)
    
    def __iter__(self):
        return iter(os.listdir(self.root_dir))
    
    def __len__(self):
        return len(os.listdir(self.root_dir))

方案说明 #

1. 代理对象FileAppendProxy：
   • 继承自bytes，完全复用原生bytes的所有行为（比如取值、拼接、比较等），不破坏files['foo.txt']的正常使用。
   • 重写__iadd__方法，当执行+=时直接打开文件追加内容，而非先读取全量内容再重写。
2. Files类：
   • 保持__getitem__的返回值兼容bytes类型，同时通过代理对象拦截+=操作。
   • 普通赋值files['foo.txt'] = b'xxx'仍执行覆盖写入逻辑，不影响原有功能。

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