Получение реального имени метода вызывающего при использовании setattr () - PullRequest
Новая
       53

Получение реального имени метода вызывающего при использовании setattr ()

0 голосов
/ 27 июня 2019

У меня есть простой класс, в котором я хочу генерировать методы на основе унаследованных полей класса: 

class Parent:
    def __init__(self, *args, **kwargs):
        self.fields = getattr(self, 'TOGGLEABLE')
        self.generate_methods()

    def _toggle(self, instance):
        print(self, instance)  # Prints correctly

        # Here I need to have the caller method, which is:
        # toggle_following()

    def generate_methods(self):
        for field_name in self.fields:
             setattr(self, f'toggle_{field_name}', self._toggle)


class Child(Parent):
    following = ['a', 'b', 'c']

    TOGGLEABLE = ('following',)

На данный момент существует toggle_following функция, успешно сгенерированная в Child.

Затем я вызываю его с одним параметром: 

>>> child = Child()
>>> child.toggle_following('b')
<__main__.Child object at 0x104d21b70> b

И он выводит ожидаемый результат в операторе print.

Но мне нужно получитьимя вызывающего абонента toggle_following в моей общей функции _toggle.

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

Ответы [ 2 ]

2 голосов
/ 27 июня 2019

может быть, это слишком хакерски, и, может быть, есть более элегантный (то есть выделенный) способ добиться этого, но:

Вы можете создать функцию-оболочку, которая передает func_name во внутренний _toggle func: 

class Parent:
    def __init__(self, *args, **kwargs):
        self.fields = getattr(self, 'TOGGLEABLE')
        self.generate_methods()

    def _toggle(self, instance, func_name):
        print(self, instance)  # Prints correctly
        print(func_name)

    def generate_methods(self):
        for field_name in self.fields:
            func_name = 'toggle_{}'.format(field_name)  # I don't have python 3.7 on this computer :P
            setattr(self, func_name, lambda instance: self._toggle(instance, func_name))


class Child(Parent):
    following = ['a', 'b', 'c']

    TOGGLEABLE = ('following',)


child = Child()
child.toggle_following('b')

Выход: 

<__main__.Child object at 0x7fbe566c0748> b
toggle_following
1 голос
/ 27 июня 2019

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

class Parent:
    def _toggle(self, instance, func_name):
        print(self, instance)  # Prints correctly
        print(func_name)

    def __getattr__(self, attr):

        if not attr.startswith("toggle_"):
            raise AttributeError("Attribute {} not found".format(attr))

        tmp = attr.replace("toggle_", "")
        if tmp not in self.TOGGLEABLE:
            raise AttributeError(
                "You cannot toggle the untoggleable {}".format(attr)
            )

        return lambda x: self._toggle(x, attr)


class Child(Parent):
    following = ['a', 'b', 'c']
    TOGGLEABLE = ('following',)


child = Child()
# This can toggle
child.toggle_following('b')

# This cannot toggle
child.toggle_something('b')

, что дает: 

(<__main__.Child instance at 0x107a0e290>, 'b')
toggle_following

Traceback (most recent call last):
  File "/Users/urban/tmp/test.py", line 26, in <module>
    child.toggle_something('b')
  File "/Users/urban/tmp/test.py", line 13, in __getattr__
    raise AttributeError("You cannot toggle the untoggleable {}".format(attr))
AttributeError: You cannot toggle the untoggleable toggle_something
...