Нахождение параметров функции в Python - PullRequest
Новая
       33

Нахождение параметров функции в Python

5 голосов
/ 31 июля 2010

Я хочу иметь возможность спросить метод класса __init__, каковы его параметры. Простой подход заключается в следующем: 

cls.__init__.__func__.__code__.co_varnames[:code.co_argcount]

Однако это не сработает, если в классе есть декораторы. Это даст список параметров для функции, возвращаемой декоратором. Я хочу перейти к исходному методу __init__ и получить эти исходные параметры. В случае декоратора, функция декоратора будет найдена в закрытии функции, возвращаемой декоратором: 

cls.__init__.__func__.__closure__[0]

Однако, если в замыкании есть другие вещи, которые декораторы могут делать время от времени, это будет сложнее: 

def Something(test):
    def decorator(func):
        def newfunc(self):
            stuff = test
            return func(self)
        return newfunc
    return decorator

def test():
    class Test(object):
        @Something(4)
        def something(self):
            print Test
    return Test

test().something.__func__.__closure__
(<cell at 0xb7ce7584: int object at 0x81b208c>, <cell at 0xb7ce7614: function object at 0xb7ce6994>)

И тогда я должен решить, хочу ли я получить параметры из декоратора или параметры из исходной функции. Функция, возвращаемая декоратором, может иметь *args и **kwargs для своих параметров. Что если есть несколько декораторов, и я должен решить, какой из них мне нужен?

Так каков наилучший способ найти параметры функции, даже если функция может быть оформлена? Кроме того, каков наилучший способ вернуться по цепочке декораторов к декорированной функции?

Обновление:

Вот как я это делаю сейчас (имена изменены, чтобы защитить личность обвиняемого): 

import abc
import collections

IGNORED_PARAMS = ("self",)
DEFAULT_PARAM_MAPPING = {}
DEFAULT_DEFAULT_PARAMS = {}

class DICT_MAPPING_Placeholder(object):
    def __get__(self, obj, type):
        DICT_MAPPING = {}
        for key in type.PARAMS:
            DICT_MAPPING[key] = None
        for cls in type.mro():
            if "__init__" in cls.__dict__:
                cls.DICT_MAPPING = DICT_MAPPING
                break
        return DICT_MAPPING

class PARAM_MAPPING_Placeholder(object):
    def __get__(self, obj, type):
        for cls in type.mro():
            if "__init__" in cls.__dict__:
                cls.PARAM_MAPPING = DEFAULT_PARAM_MAPPING
                break
        return DEFAULT_PARAM_MAPPING

class DEFAULT_PARAMS_Placeholder(object):
    def __get__(self, obj, type):
        for cls in type.mro():
            if "__init__" in cls.__dict__:
                cls.DEFAULT_PARAMS = DEFAULT_DEFAULT_PARAMS
                break
        return DEFAULT_DEFAULT_PARAMS

class PARAMS_Placeholder(object):
    def __get__(self, obj, type):
        func = type.__init__.__func__
        # unwrap decorators here
        code = func.__code__
        keys = list(code.co_varnames[:code.co_argcount])
        for name in IGNORED_PARAMS:
            try: keys.remove(name)
            except ValueError: pass
        for cls in type.mro():
            if "__init__" in cls.__dict__:
                cls.PARAMS = tuple(keys)
                break
        return tuple(keys)

class BaseMeta(abc.ABCMeta):
    def __init__(self, name, bases, dict):
        super(BaseMeta, self).__init__(name, bases, dict)
        if "__init__" not in dict:
            return
        if "PARAMS" not in dict:
            self.PARAMS = PARAMS_Placeholder()
        if "DEFAULT_PARAMS" not in dict:
            self.DEFAULT_PARAMS = DEFAULT_PARAMS_Placeholder()
        if "PARAM_MAPPING" not in dict:
            self.PARAM_MAPPING = PARAM_MAPPING_Placeholder()
        if "DICT_MAPPING" not in dict:
            self.DICT_MAPPING = DICT_MAPPING_Placeholder()


class Base(collections.Mapping):
    __metaclass__ = BaseMeta
    """
    Dict-like class that uses its __init__ params for default keys.

    Override PARAMS, DEFAULT_PARAMS, PARAM_MAPPING, and DICT_MAPPING
    in the subclass definition to give non-default behavior.

    """
    def __init__(self):
        pass
    def __nonzero__(self):
        """Handle bool casting instead of __len__."""
        return True
    def __getitem__(self, key):
        action = self.DICT_MAPPING[key]
        if action is None:
            return getattr(self, key)
        try:
            return action(self)
        except AttributeError:
            return getattr(self, action)
    def __iter__(self):
        return iter(self.DICT_MAPPING)
    def __len__(self):
        return len(self.DICT_MAPPING)

print Base.PARAMS
# ()
print dict(Base())
# {}

В этот момент Base сообщает неинтересные значения для четырех констант, а dict-версия экземпляров пуста. Однако, если вы подкласс, вы можете переопределить любой из четырех, или вы можете включить другие параметры в __init__: 

class Sub1(Base):
    def __init__(self, one, two):
        super(Sub1, self).__init__()
        self.one = one
        self.two = two

Sub1.PARAMS
# ("one", "two")
dict(Sub1(1,2))
# {"one": 1, "two": 2}

class Sub2(Base):
    PARAMS = ("first", "second")
    def __init__(self, one, two):
        super(Sub2, self).__init__()
        self.first = one
        self.second = two

Sub2.PARAMS
# ("first", "second")
dict(Sub2(1,2))
# {"first": 1, "second": 2}

1 Ответ

3 голосов
/ 17 августа 2010

Рассмотрим этот декоратор: 

def rickroll(old_function):
    return lambda junk, junk1, junk2: "Never Going To Give You Up"

class Foo(object):
    @rickroll
    def bar(self, p1, p2):
        return p1 * p2

print Foo().bar(1, 2)

В нем декоратор rickroll берет метод bar, отбрасывает его, заменяет его новой функцией, которая игнорирует параметры с разными именами (и, возможно, нумеруется!) И вместо этого возвращает строку из классической песни.

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

В таком случае я не вижу, как вы можете найти имена параметров p1 и p2. В моем понимании, даже сам интерпретатор Python понятия не имеет, как они назывались.

...