Декоратор функции, который записывает значение указанных аргументов функции в доступный объект python - PullRequest
Новая
       0

Декоратор функции, который записывает значение указанных аргументов функции в доступный объект python

0 голосов
/ 13 сентября 2018

Я пытаюсь создать декоратор функции, который записывает значение аргументов указанной функции в доступный объект python. У меня уже есть рабочий код, но мне не хватает части, чтобы закончить это.

Во-первых, у меня есть журнал объектов, в котором я буду правильно сохранять вещи: 

class Borg:
    _shared_state = {}

    def __init__(self):
        self.__dict__ = self._shared_state

class Log(Borg):

    def __init__(self):
        Borg.__init__(self)
        if not hasattr(self, 'tape'):
            self.tape = []

    def add(self, this):
        self.tape.append(this)

    def __str__(self):
        return '\n'.join([str(line) for line in self.tape])

Тогда у меня есть универсальный объект вызова и реализация декоратора (с отсутствующим кодом): 

import inspect
import functools

class Call:

    def __init__(self, name, **saved_arguments):
        self.name = name
        self.saved_arguments = saved_arguments

    def __str__(self):
        return f'Call(name={self.name}, saved_arguments={self.saved_arguments})'

def record(func, save_args_names=None):
    if save_args_names is None:
        save_args_names = {}
    name = func.__name__
    args = inspect.getfullargspec(func).args
    if save_args_names and not set(save_args_names).issubset(set(args)):
        raise ValueError(f'Arguments not present in function: {set(save_args_names) - set(args)}')
    log = Log()

    @functools.wraps(func)
    def wrapper(*func_args, **func_kwargs):
        # **here** I am missing something to replace 0 with the correct values!
        saved_arguments = {a: 0 for a in save_args_names}
        log.add(Call(name, **saved_arguments))
        return_value = func(*func_args, **func_kwargs)
        return return_value

    return wrapper

Чтобы проверить это, у меня настроены следующие функции: 

def inner(x, add=0):
    return sum(x) + add

def outer(number, add=0):
    x = range(number)
    return inner(x, add)

и основной вариант использования (без сохранения аргументов) работает: 

inner = record(inner)
print(outer(1), outer(2), outer(3))
print(Log())

Выводит правильно: 

0 1 3
Call(name=inner, saved_arguments={})
Call(name=inner, saved_arguments={})
Call(name=inner, saved_arguments={})

Чего мне не хватает это способ использовать этот вариант использования: 

inner = record(inner, save_args_names=['x'])
print(outer(1), outer(2), outer(3))
print(Log())

для вывода: 

0 1 3
Call(name=inner, saved_arguments={'x': range(0, 1)})
Call(name=inner, saved_arguments={'x': range(0, 2)})
Call(name=inner, saved_arguments={'x': range(0, 3)})

Это также должно работать для аргументов ключевых слов, например ::1010* 

inner = record(inner, save_args_names=['x', 'add'])
print(outer(1, 2), outer(2, 3), outer(3, 4))
print(Log())

должен вывести: 

2 4 7
Call(name=inner, saved_arguments={'x': range(0, 1), 'add': 2})
Call(name=inner, saved_arguments={'x': range(0, 2), 'add': 3})
Call(name=inner, saved_arguments={'x': range(0, 3), 'add': 4})

Я чувствую, что я близко, и что библиотека inspect должна помочь мне закрыть это, но небольшая помощь будет высоко ценится!

1 Ответ

0 голосов
/ 13 сентября 2018

Функция, которую вы ищете: Signature.bind.Определите вашу wrapper функцию следующим образом: 

@functools.wraps(func)
def wrapper(*func_args, **func_kwargs):
    signature = inspect.signature(func)
    bound_args = signature.bind(*func_args, **func_kwargs)
    saved_arguments = {a: bound_args.arguments[a] for a in save_args_names}

    log.add(Call(name, **saved_arguments))
    return_value = func(*func_args, **func_kwargs)
    return return_value
...