Аргументы в пользу декораторов Python

Оберните вашу функцию другой функцией:

import functools
def makeBold(**kwargs):
    attribstrings = str(kwargs) # figure out what to do with the dict yourself 
    def actualdecorator(fn):
        @functools.wraps(fn)
        def wrapped():
            return '<b'+attribstrings+'>'+fn()+'</b>'
        return wrapped
    return actualdecorator

Я оставляю работу над построением струны в качестве упражнения для читателя.

Обратите внимание, что структура выражения декоратора - @ <callable object of one parameter, w> <declaration of callable object, f>. Это эффект f = w(f). Соответственно, w (декоратор) должен возвращать вызываемый объект того же типа, что и f.

В @makebold(foo)def bar(x):pass выражение makebold(foo) является декоратором, то есть конечным эффектом декоратора является bar = makebold(foo)(bar), и поэтому bar в конечном итоге содержит wrapped.

Цель functools.wraps - исправить свойства декорированной функции для копирования метаданных (таких как имя и строка документации) из функции аргумента в упакованную функцию, чтобы весь процесс упаковки был прозрачным.

Я, возможно, сомневаюсь, что это хороший вариант использования декораторов, но здесь:

import string

SelfClosing = object()

def escapeAttr(attr):
    # WARNING: example only, security not guaranteed for any of these functions
    return attr.replace('"', '\\"')

def tag(name, content='', **attributes):
    # prepare attributes
    for attr,value in attributes.items():
        assert all(c.isalnum() for c in attr)  # probably want to check xml spec
    attrString = ' '.join('{}="{}"'.format(k,escapeAttr(v)) for k,v in attributes.items())

    if not content==SelfClosing:
        return '<{name} {attrs}>{content}</{name}>'.format(
            name = name,
            attrs = attrString,
            content = content
        )
    else:  # self-closing tag
        return '<{name} {attrs}/>'

Пример:

def makeBoldWrapper(**attributes):
    def wrapWithBold(origFunc):
        def composed(*args, **kw):
            result = origFunc(*args, **kw)
            postprocessed = tag('b', content=result, **attributes)
            return postprocessed
        return composed
    return wrapWithBold

Демо:

@makeBoldWrapper(attr1='1', attr2='2')
def helloWorld(text):
    return text

>>> print( helloWorld('Hello, world!') )
<b attr2="2" attr1="1">Hello, world!</b>

Распространенное заблуждение декораторов состоит в том, что параметры (attr1=...) являются параметрами декоратора @myDecorator; Это не относится к делу. Скорее результат вызова функции myDecoratorFactory(attr1=...) вычисляется как someresult и становится анонимным декоратором @someresult. Следовательно, «декораторы с аргументами» на самом деле являются фабриками декораторов, которым необходимо возвращать декоратор в качестве значения.

Чтобы сделать что-то подобное, вам понадобится функция, возвращающая функцию декоратора. Итак, в этом случае (при условии, что вы хотите принимать произвольные атрибуты) вы должны написать

def format_attribs(kwargs):
    """Properly formats HTML attributes from a dictionary"""
    return ' '.join('{}="{}"'.format(key, val) for key,val in kwargs.iteritems())

def makeBold(**kwargs):
    attribs = format_attribs(kwargs)
    def _makeBold(fn):
        def wrapped():
            return '<b ' + attribs + '>' + fn() + '</b>'
        return wrapped
    return _makeBold

Чтобы сделать эту makeBold функцию немного более общей, вы хотите передать аргументы в fn и сохранить другую информацию, такую ​​как имя функции, с помощью _ 4_:

import functools
def makeBold(**kwargs):
    attribs = format_attribs(kwargs)
    def _makeBold(fn):
        @wraps(fn)
        def wrapped(*args, **kwargs):
            return '<b ' + attribs + '>' + fn(*args, **kwargs) + '</b>'
        return wrapped
    return _makeBold