为什么我的想法在python2中不起作用？
这是一个可以改变密钥的字典子类的想法。 这是一个简单的自包含的例子，就像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)
dev注意：如果你为自己的用途复制我的代码，你应该执行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东西，这只是一个小细节而不是破解代码的问题。
Python 3中的functools.wraps()版本可以处理函数对象，其中有一些属性会被复制丢失; Python 2中的那个不能。 这是因为＃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__属性; 这大多只用于反思。