Как узнать арность метода в Python

В идеале, вам нужно обезьяно исправить функцию arity как метод на функторах Python. Вот как:

def arity(self, method):
    return getattr(self.__class__, method).func_code.co_argcount - 1

functor = arity.__class__
functor.arity = arity
arity.__class__.arity = arity

Но CPython реализует функторы в C, вы не можете их изменить. Однако это может работать в PyPy.

Это все, если ваша функция arity () верна. А как насчет вариативных функций? Вы вообще хотите получить ответ?

Используйте декоратор для украшения методов, например

def arity(method):

  def _arity():
    return method.func_code.co_argcount - 1 # remove self

  method.arity = _arity

  return method

class Foo:
  @arity
  def bar(self, bla):
    pass

print Foo().bar.arity()

Теперь реализуйте функцию _arity , чтобы вычислить количество аргументов в зависимости от ваших потребностей

Это единственный способ, который, как мне кажется, должен быть на 100% эффективным (по крайней мере, в отношении того, является ли функция определяемой пользователем или написанной на C) при определении функции (минимум ) арность. Однако вы должны быть уверены, что эта функция не вызовет никаких побочных эффектов и не вызовет ошибку TypeError:

from functools import partial

def arity(func):
    pfunc = func
    i = 0
    while True:
        try:
            pfunc()
        except TypeError:
            pfunc = partial(pfunc, '')
            i += 1
        else:
            return i

def foo(x, y, z):
    pass

def varfoo(*args):
    pass

class klass(object):
    def klassfoo(self):
        pass

print arity(foo)
print arity(varfoo)

x = klass()
print arity(x.klassfoo)

# output
# 3
# 0
# 0

Как видите, это определит минимальную арность, если функция принимает переменное количество аргументов. Он также не будет принимать во внимание аргумент self или cls метода класса или экземпляра.

Если быть полностью честным, я бы не стал использовать эту функцию в производственной среде, если бы не знал точно, какие функции будут вызываться, хотя так как есть много места для глупых ошибок. Это может свести на нет цель.

вот еще одна попытка использования метакласса, так как я использую python 2.5, но с 2.6 вы можете легко украсить класс

метакласс также может быть определен на уровне модуля, поэтому он работает для всех классов

from types import FunctionType

def arity(unboundmethod):
    def _arity():
        return unboundmethod.func_code.co_argcount - 1 # remove self
    unboundmethod.arity = _arity

    return unboundmethod

class AirtyMetaclass(type):
    def __new__(meta, name, bases, attrs):
        newAttrs = {}
        for attributeName, attribute in attrs.items():
            if type(attribute) == FunctionType:
                attribute = arity(attribute)

            newAttrs[attributeName] = attribute

        klass = type.__new__(meta, name, bases, newAttrs)

        return klass

class Foo:
    __metaclass__ = AirtyMetaclass
    def bar(self, bla):
        pass

print Foo().bar.arity()