Django добавляет необязательные аргументы декоратору - PullRequest
Новая
       6

Django добавляет необязательные аргументы декоратору

4 голосов
/ 27 января 2012

У меня есть следующий декоратор и вид, который отлично работает.

декоратор 

def event_admin_only(func):
    """
    Checks if the current role for the user is an Event Admin or not
    """
    def decorator(request, *args, **kwargs):
        event = get_object_or_404(Event, slug=kwargs['event_slug'])

        allowed_roles = [role[1] for role in Role.ADMIN_ROLES]

        # get user current role
        current_role = request.session.get('current_role')

        if current_role not in allowed_roles:
            url = reverse('no_perms')
            return redirect(url)
        else:       
            return func(request, *args, **kwargs)
    return decorator

View 

@event_admin_only
def event_dashboard(request, event_slug):
    pass

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

@event_admin_only(obj1,[...])
def event_dashboard(request, event_slug):
    pass

1 Ответ

10 голосов
/ 27 января 2012

Вам нужно обернуть создание функции декоратора в другую функцию: 

def the_decorator(arg1, arg2):

    def _method_wrapper(view_method):

        def _arguments_wrapper(request, *args, **kwargs) :
            """
            Wrapper with arguments to invoke the method
            """

            #do something with arg1 and arg2

            return view_method(request, *args, **kwargs)

        return _arguments_wrapper

    return _method_wrapper

Это можно затем назвать так: 

@the_decorator("an_argument", "another_argument")
def event_dashboard(request, event_slug):

Я бы порекомендовал ответ отУдовлетворительно по этому вопросу, чтобы понять это: Как создать цепочку функций декораторов?

...