python - 如何动态更改子类中方法的签名？

在使用类方法动态更改子类中的方法时，如何动态更改方法的签名？

例

import inspect

class ModelBase(object):

    @classmethod
    def method_one(cls, *args):
        raise NotImplementedError

    @classmethod
    def method_two(cls, *args):
        return cls.method_one(*args) + 1

class SubClass(ModelBase):
    @staticmethod
    def method_one(a, b):
        return a + b

test = SubClass()

try:
    print(inspect.signature(test.method_two))
except AttributeError:
    print(inspect.getargspec(test.method_two).args)

我希望test.method_two获得test.method_one的签名。如何重写父类ModelBase？

我已经阅读了有关Preserving signatures of decorated functions的信息。在python3.4 +中，functools.wraps有助于保留修饰函数的签名。我想将其应用于类方法。

当使用functools.wraps时，我需要分配修饰方法的名称。但是在这种情况下如何在classmethod之外访问修饰方法？

from functools import wraps

class ModelBase(object):

    @classmethod
    def method_one(cls, *args):
        raise NotImplementedError

    @classmethod
    def method_two(cls):
        @wraps(cls.method_one)
        def fun(*args):
            return cls.method_one(*args) + 1
        return fun

method_two返回一个包装的函数，但是我必须将其与test.method_two()(*arg)一起使用。此方法不是直接的。

最佳答案

如果仅出于自省目的，则可以覆盖__getattribute__上的ModelBase，并且每次访问method_two时，我们都会返回带有method_one签名的函数。

import inspect

def copy_signature(frm, to):
    def wrapper(*args, **kwargs):
        return to(*args, **kwargs)
    wrapper.__signature__ = inspect.signature(frm)
    return wrapper


class ModelBase(object):

    @classmethod
    def method_one(cls, *args):
        raise NotImplementedError

    @classmethod
    def method_two(cls, *args):
        return cls.method_one(*args) + 1

    def __getattribute__(self, attr):
        value = object.__getattribute__(self, attr)
        if attr == 'method_two':
            value = copy_signature(frm=self.method_one, to=value)
        return value


class SubClass(ModelBase):
    @staticmethod
    def method_one(a, b):
        return a + b


class SubClass2(ModelBase):
    @staticmethod
    def method_one(a, b, c, *arg):
        return a + b

演示：

>>> test1 = SubClass()
>>> print(inspect.signature(test1.method_two))
(a, b)
>>> test2 = SubClass2()
>>> print(inspect.signature(test2.method_two))
(a, b, c, *arg)