Получение имен параметров метода в Python - PullRequest
202 голосов
/ 20 октября 2008

Учитывая функцию Python:

def aMethod(arg1, arg2):
    pass

Как мне извлечь количество и имена аргументов. То есть, учитывая, что у меня есть ссылка на func, я хочу, чтобы func. [Что-то] возвращало ("arg1", "arg2").

Сценарий использования этого заключается в том, что у меня есть декоратор, и я хочу использовать аргументы метода в том же порядке, в котором они отображаются для фактической функции в качестве ключа. Т.е. как будет выглядеть декоратор, который печатает «a, b», когда я вызываю aMethod («a», «b»)?

Ответы [ 13 ]

315 голосов
/ 20 октября 2008

Посмотрите на модуль inspect - он проведет проверку различных свойств объекта кода для вас.

>>> inspect.getfullargspec(aMethod)
(['arg1', 'arg2'], None, None, None)

Другими результатами являются имена переменных * args и ** kwargs и предоставленные значения по умолчанию. то есть.

>>> def foo(a,b,c=4, *arglist, **keywords): pass
>>> inspect.getfullargspec(foo)
(['a', 'b', 'c'], 'arglist', 'keywords', (4,))

Обратите внимание, что некоторые вызовы могут не быть интроспективными в некоторых реализациях Python. Например, в CPython некоторые встроенные функции, определенные в C, не предоставляют метаданных об их аргументах. В результате вы получите ValueError в случае использования inspect.getfullargspec() со встроенной функцией.

Начиная с Python 3.3, вы также можете использовать inspect.signature () , чтобы узнать сигнатуру вызываемого объекта:

>>> inspect.signature(foo)
<Signature (a, b, c=4, *arglist, **keywords)>
94 голосов
/ 20 октября 2008

В CPython количество аргументов равно

aMethod.func_code.co_argcount

и их имена в начале

aMethod.func_code.co_varnames

Это детали реализации CPython, так что это, вероятно, не работает в других реализациях Python, таких как IronPython и Jython.

Одним из переносимых способов допустить "сквозные" аргументы является определение вашей функции с подписью func(*args, **kwargs). Это часто используется, например, matplotlib, где внешний уровень API передает множество аргументов ключевых слов в API более низкого уровня.

21 голосов
/ 14 мая 2013

В методе декоратора вы можете перечислить аргументы исходного метода следующим образом:

import inspect, itertools 

def my_decorator():

        def decorator(f):

            def wrapper(*args, **kwargs):

                # if you want arguments names as a list:
                args_name = inspect.getargspec(f)[0]
                print(args_name)

                # if you want names and values as a dictionary:
                args_dict = dict(itertools.izip(args_name, args))
                print(args_dict)

                # if you want values as a list:
                args_values = args_dict.values()
                print(args_values)

Если **kwargs важен для вас, то это будет немного сложно:

        def wrapper(*args, **kwargs):

            args_name = list(OrderedDict.fromkeys(inspect.getargspec(f)[0] + kwargs.keys()))
            args_dict = OrderedDict(list(itertools.izip(args_name, args)) + list(kwargs.iteritems()))
            args_values = args_dict.values()

Пример:

@my_decorator()
def my_function(x, y, z=3):
    pass


my_function(1, y=2, z=3, w=0)
# prints:
# ['x', 'y', 'z', 'w']
# {'y': 2, 'x': 1, 'z': 3, 'w': 0}
# [1, 2, 3, 0]
13 голосов
/ 21 октября 2008

Вот то, что я думаю, будет работать для того, что вы хотите, используя декоратор.

class LogWrappedFunction(object):
    def __init__(self, function):
        self.function = function

    def logAndCall(self, *arguments, **namedArguments):
        print "Calling %s with arguments %s and named arguments %s" %\
                      (self.function.func_name, arguments, namedArguments)
        self.function.__call__(*arguments, **namedArguments)

def logwrap(function):
    return LogWrappedFunction(function).logAndCall

@logwrap
def doSomething(spam, eggs, foo, bar):
    print "Doing something totally awesome with %s and %s." % (spam, eggs)


doSomething("beans","rice", foo="wiggity", bar="wack")

Запустите его, он выдаст следующий вывод:

C:\scripts>python decoratorExample.py
Calling doSomething with arguments ('beans', 'rice') and named arguments {'foo':
 'wiggity', 'bar': 'wack'}
Doing something totally awesome with beans and rice.
12 голосов
/ 07 июня 2010

Я думаю, вы ищете метод местных жителей -


In [6]: def test(a, b):print locals()
   ...: 

In [7]: test(1,2)              
{'a': 1, 'b': 2}
8 голосов
/ 01 ноября 2016

Версия Python 3:

def _get_args_dict(fn, args, kwargs):
    args_names = fn.__code__.co_varnames[:fn.__code__.co_argcount]
    return {**dict(zip(args_names, args)), **kwargs}

Метод возвращает словарь, содержащий как args, так и kwargs.

6 голосов
/ 26 июля 2017

Python 3.5 +:

Предупреждение об устаревании: inspect.getargspec () устарела, вместо этого используйте inspect.signature ()

Так ранее:

func_args = inspect.getargspec(function).args

Сейчас:

func_args = list(inspect.signature(function).parameters.keys())

Для проверки:

'arg' in list(inspect.signature(function).parameters.keys())

Учитывая, что у нас есть функция 'function', которая принимает аргумент 'arg', она будет оцениваться как True, иначе как False.

Пример из консоли Python:

Python 3.6.0 (v3.6.0:41df79263a11, Dec 23 2016, 07:18:10) [MSC v.1900 32 bit (Intel)] on win32
>>> import inspect
>>> 'iterable' in list(inspect.signature(sum).parameters.keys())
True
4 голосов
/ 20 августа 2017

В Python 3. + с имеющимся объектом Signature простой способ получить соответствие между именами аргументов и значениями - это использование метода bind() в Signature!

Например, вот декоратор для печати такой карты:

import inspect

def decorator(f):
    def wrapper(*args, **kwargs):
        bound_args = inspect.signature(f).bind(*args, **kwargs)
        bound_args.apply_defaults()
        print(dict(bound_args.arguments))

        return f(*args, **kwargs)

    return wrapper

@decorator
def foo(x, y, param_with_default="bars", **kwargs):
    pass

foo(1, 2, extra="baz")
# This will print: {'kwargs': {'extra': 'baz'}, 'param_with_default': 'bars', 'y': 2, 'x': 1}
2 голосов
/ 30 мая 2017

Возвращает список имен аргументов, заботится о частичках и обычных функциях:

def get_func_args(f):
    if hasattr(f, 'args'):
        return f.args
    else:
        return list(inspect.signature(f).parameters)
2 голосов
/ 22 марта 2017

Обновление для Ответ Брайана :

Если функция в Python 3 имеет аргументы только для ключевых слов, вам нужно использовать inspect.getfullargspec:

def yay(a, b=10, *, c=20, d=30):
    pass
inspect.getfullargspec(yay)

дает это:

FullArgSpec(args=['a', 'b'], varargs=None, varkw=None, defaults=(10,), kwonlyargs=['c', 'd'], kwonlydefaults={'c': 20, 'd': 30}, annotations={})
...