Ошибка при использовании decorator: decorator () принимает 1 позиционный аргумент, но задано 2 - PullRequest
Новая
       21

Ошибка при использовании decorator: decorator () принимает 1 позиционный аргумент, но задано 2

1 голос
/ 10 апреля 2019

Я читаю и слежу за Pro Python книгой, я создаю два декоратора annotation_decorator и типов безопасных , используя тот же код из книги, но когда я пытаюсь для запуска этого кода я получаю:

TypeError: decorator () принимает 1 позиционный аргумент, но 2 даны

Код такой же, как в книге, и я не могу понять, почему это работает, вы, ребята, можете заметить, что не так? Я провел здесь PoC на случай, если вы захотите проверить: https://repl.it/repls/GlassNotedPayware

import functools
import inspect
from itertools import chain


def annotation_decorator(process):
    """
    Creates a decorator that processes annotations for each argument passed
    into its target function, raising an exception if there's a problem.
    """
    @functools.wraps(process)
    def decorator(func):
        spec = inspect.getfullargspec(func)
        annotations = spec.annotations

        defaults = spec.defaults or ()
        defaults_zip = zip(spec.args[-len(defaults):], defaults)
        kwonlydefaults = spec.kwonlydefaults or {}

        for name, value in chain(defaults_zip, kwonlydefaults.items()):
            if name in annotations:
                process(value, annotations[name])

        @functools.wraps(func)
        def wrapper(*args, **kwargs):
            # Populate a dictionary of explicit arguments passed positionally
            explicit_args = dict(zip(spec.args, args))
            new_args = []
            new_kwargs = {}
            keyword_args = kwargs.copy()

            # Deal with explicit arguments passed positionally
            for name, arg in explicit_args:
                if name in annotations:
                    new_args.append(process(arg, annotations[name]))

            # Add all explicit arguments passed by keyword
            for name in chain(spec.args, spec.kwonlyargs):
                if name in kwargs:
                    new_kwargs[name] = process(keyword_args.pop(name),
                                               annotations[name])

            # Deal with variable positional arguments
            if spec.varargs and spec.varargs in annotations:
                annotation = annotations[spec.varargs]
                for arg in args[len(spec.args):]:
                    new_args.append(process(arg, annotation))

            # Deal with variable keyword arguments
            if spec.varkw and spec.varkw in annotations:
                annotation = annotations[spec.varkw]
                for name, arg in keyword_args.items():
                    new_kwargs[name] = process(arg, annotation)

            r = func(*new_args, **new_kwargs)

            if 'return' in annotations:
                r = process(r, annotations['return'])

            return r

        return wrapper

    return decorator


@annotation_decorator
def typesafe(value, annotation):
    """
    Verify that the function is called with the right argument types and
    that it returns a value of the right type, according to its annotations
    """
    if not isinstance(value, annotation):
        raise TypeError("Expected %s, got %s." % (annotation.__name__,
                                                  type(value).__name__))

    return value


@annotation_decorator
def coerce_arguments(value, annotation):
    return annotation(value)


@typesafe(str, str)
def combine(a, b):
    return a + b


combine('spam', 'alot')

1 Ответ

0 голосов
/ 10 апреля 2019

Я изменил две вещи на основе комментария @chepner:

Вместо использования typesafe декоратора для функции combine я просто украшаю annotation_decorator передачей typesafe в качестве аргумента, а также добавляем аннотацию функции к переменным функции. 

@annotation_decorator(process=typesafe)
def combine(a: str, b: str):
    return a + b
...