我们正在开发一个
python库,并希望改变某些函数中某些函数参数的命名方式.
我们希望保持向后兼容性,因此我们希望找到一种为函数参数创建别名的方法.
这是一个例子：
旧版：
class MyClass(object):
def __init__(self,object_id):
self.id = object_id
新版本：
class MyClass(object):
def __init__(self,id_object):
self.id = id_object
我们如何使类与两种调用方式兼容：
object1 = MyClass(object_id=1234) object2 = MyClass(id_object=1234)
我当然可以创建这样的东西：
class MyClass(object):
def __init__(self,object_id=None,id_object=None):
if id_object is not None:
self.id = id_object
else:
self.id = object_id
但是,它会改变参数的数量,我们严格要避免这种情况.
有没有办法声明方法别名或参数别名？
解决方法
你可以写一个装饰者：
import functools
import warnings
def deprecated_alias(**aliases):
def deco(f):
@functools.wraps(f)
def wrapper(*args,**kwargs):
rename_kwargs(f.__name__,kwargs,aliases)
return f(*args,**kwargs)
return wrapper
return deco
def rename_kwargs(func_name,aliases):
for alias,new in aliases.items():
if alias in kwargs:
if new in kwargs:
raise TypeError('{} received both {} and {}'.format(
func_name,alias,new))
warnings.warn('{} is deprecated; use {}'.format(alias,new),DeprecationWarning)
kwargs[new] = kwargs.pop(alias)
class MyClass(object):
@deprecated_alias(object_id='id_object')
def __init__(self,id_object):
self.id = id_object
或者,由于您使用的是Python 3,因此可以使object_id成为仅限关键字的参数：
import warnings
class MyClass(object):
# v Look here
def __init__(self,id_object=None,*,object_id=None):
if id_object is not None and object_id is not None:
raise TypeError("MyClass received both object_id and id_object")
elif id_object is not None:
self.id = id_object
elif object_id is not None:
warnings.warn("object_id is deprecated; use id_object",DeprecationWarning)
self.id = object_id
else:
raise TypeError("MyClass missing id_object argument")