Подсказка типа Python 3 для динамических атрибутов через __getattr__ - PullRequest
Новая
       17

Подсказка типа Python 3 для динамических атрибутов через __getattr__

3 голосов
/ 17 апреля 2019

Чтобы обернуть все вызовы функций модуля и получить к нему доступ через метод __getattr__ класса-оболочки, я попытался использовать библиотеку typing, но не мог понять, как это сделать правильно. 

import interface

"""
>>> print(interface.__all__)
['execute_foo_operation', ...]
"""

class InterfaceWrapper(object):
    def __init__(self, job_queue):
        self.job_queue = job_queue
        self.callbacks = []

    def __getattr__(self, name):
        func = getattr(interface, name)
        return functools.partial(self._wrapper, func)

    def _wrapper(self, func, *args, **kwargs):
        job = func(*args, **kwargs)
        self.job_queue.push(job)
        for callback in self.callbacks:
            callback(job)
        return job

    def register_callback(self, callback):
        self.callbacks.append(callback)


class Operator(object):
    def __init__(self, job_queue):
        self.interface = InterfaceWrapper(job_queue)

    def after_queuing(self):
        # do something

    def execute_foo_operation(self):
        self.interface.register_callback(self.after_queuing)
        self.interface.execute_foo_operation()  # unresolved attribute

Кто-нибудь может направить мой код для правильной работы?

...