Есть ли способ проверить подпись функции в Python?

Question

Я ищу способ проверить количество аргументов, которые данная функция принимает в Python. Целью является создание более надежного метода исправления моих классов для тестов. Итак, я хочу сделать что-то вроде этого:

class MyClass (object): 
    def my_function(self, arg1, arg2): 
     result = ... # Something complicated 
     return result 

def patch(object, func_name, replacement_func): 
    import new 

    orig_func = getattr(object, func_name) 
    replacement_func = new.instancemethod(replacement_func, 
          object, object.__class__) 

    # ... 
    # Verify that orig_func and replacement_func have the 
    # same signature. If not, raise an error. 
    # ... 

    setattr(object, func_name, replacement_func) 

my_patched_object = MyClass() 
patch(my_patched_object, "my_function", lambda self, arg1: "dummy result") 
# The above line should raise an error! 

Спасибо.

Answer 1

Вы можете использовать:

import inspect 
len(inspect.getargspec(foo_func)[0]) 

Это не будет признавать параметры переменной длины, как:

def foo(a, b, *args, **kwargs): 
    pass 

Answer 2

Вы должны использовать inspect.getargspec.

Answer 3

Модуль inspect позволяет изучить аргументы функции. Об этом несколько раз просили о переполнении стека; попробуйте найти некоторые из этих ответов. Например:

Getting method parameter names in python