为啥我的想法在 python2 中不起作用？

Posted 2023-02-23

【中文标题】为啥我的想法在 python2 中不起作用？

【英文标题】：Why doesn't my idea work in python2?为什么我的想法在 python2 中不起作用？

【发布时间】：2015-01-31 07:57:00

【问题描述】：

这是一个可以改变键的 dict 子类的想法。这是一个简单的自包含示例，类似于 dict，但对 str 键不区分大小写。

from functools import wraps

def key_fix_decorator(f):
    @wraps(f)
    def wrapped(self, *args, **kwargs):
        if args and isinstance(args[0], str):
            args = (args[0].lower(),) + args[1:]
        return f(self, *args, **kwargs)
    return wrapped

class LowerDict(dict):
    pass

for method_name in '__setitem__', '__getitem__', '__delitem__', '__contains__', 'get', 'pop', 'setdefault':
    new_method = key_fix_decorator(getattr(LowerDict, method_name))
    setattr(LowerDict, method_name, new_method)

_{开发说明：如果您复制我的代码以供自己使用，您应该实现 LowerDict.__init__ 以检查是否存在任何键冲突 - 为此目的，我没有费心将其包含在内这个问题的}

在 python3 上似乎一切正常：

>>> d = LowerDict(potato=123, spam='eggs')
>>> d['poTATo']
123
>>> d.pop('SPAm')
'eggs'
>>> d['A']
# KeyError: 'a'

在 python2 中它甚至不导入，这是回溯：

  File "/tmp/thing.py", line 15, in <module>
    new_method = key_fix_decorator(getattr(LowerDict, method_name))
  File "/tmp/thing.py", line 4, in key_fix_decorator
    @wraps(f)
  File "/usr/lib/python2.7/functools.py", line 33, in update_wrapper
    setattr(wrapper, attr, getattr(wrapped, attr))
AttributeError: 'wrapper_descriptor' object has no attribute '__module__'

可能是什么问题？除了str/basestring 之外，我看不到任何特定于版本的代码，这只是一个小细节，而不是代码破坏问题。

【问题讨论】：

区别在于update_wrapper在两个版本中的实现方式。hg.python.org/cpython/file/2.7/Lib/functools.py#l17和hg.python.org/cpython/file/3.4/Lib/functools.py#l43

这可能是相关的：bugs.python.org/issue3445

您可能也对 PEP 455 感兴趣：python.org/dev/peps/pep-0455

【参考方案1】：

Python 3 中的functools.wraps() 版本可以处理函数对象，但它复制的一些属性丢失了； Python 2 中的那个不能。这是因为issue #3445 仅适用于 Python 3； dict的方法是用C代码定义的，没有__module__属性。

省略 @wraps(f) 装饰器可以让 Python 2 中的一切都正常工作：

>>> def key_fix_decorator(f):
...     def wrapped(self, *args, **kwargs):
...         if args and isinstance(args[0], str):
...             args = (args[0].lower(),) + args[1:]
...         return f(self, *args, **kwargs)
...     return wrapped
... 
>>> class LowerDict(dict):
...     pass
... 
>>> for method_name in '__setitem__', '__getitem__', '__delitem__', '__contains__', 'get', 'pop', 'setdefault':
...     new_method = key_fix_decorator(getattr(LowerDict, method_name))
...     setattr(LowerDict, method_name, new_method)
... 
>>> d = LowerDict(potato=123, spam='eggs')
>>> d['poTATo']
123
>>> d.pop('SPAm')
'eggs'
>>> d['A']
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "<stdin>", line 5, in wrapped
KeyError: 'a'

您可以复制 足够 wraps 手动执行的操作：

def key_fix_decorator(f):
    def wrapped(self, *args, **kwargs):
        if args and isinstance(args[0], str):
            args = (args[0].lower(),) + args[1:]
        return f(self, *args, **kwargs)
    wrapped.__name__ = f.__name__
    wrapped.__doc__ = f.__doc__
    return wrapped

或限制wraps 尝试复制的属性：

def key_fix_decorator(f):
    @wraps(f, assigned=('__name__', '__doc__'))
    def wrapped(self, *args, **kwargs):
        if args and isinstance(args[0], str):
            args = (args[0].lower(),) + args[1:]
        return f(self, *args, **kwargs)
    return wrapped

您实际上不需要在此处更新__module__ 属性；这主要只对内省有用。