Django 将 URL 参数传递给装饰器

Question

如果他们不是工作人员，则尝试将用户重定向到页面。 如何将 url 参数传递给 django 装饰器？

# categories.urls.py

from django.urls import path
from categories.views import CategoryInfoView, CategoryDetailView

app_name = 'categories'

urlpatterns = [
    path('<slug:handle>', CategoryDetailView.as_view(), name = 'category'),
    path('<slug:handle>/info/', CategoryInfoView.as_view(), name = 'category_info'),
]

# categories.view.py

class CategoryInfoView(LoginRequiredMixin, DetailView):
    model = Category
    template_name = 'categories/info.html'
    context_object_name = 'category'

    @redirect_if_not_staff(redirect_to = reverse_lazy('categories:category')) # <-- how do i pass the url parameter here?!
    def get(self, request, *args, **kwargs):
        return super().get(self, request, *args, **kwargs)

    def get_object(self):
        return get_object_or_404(Category, handle = self.kwargs.get('handle'))

# decorator.py

def redirect_if_not_staff(*setting_args, **setting_kwargs):
    """
    A decorator to redirect users if they are not staff
    Can be used as: @decorator(with, arguments, and = kwargs) or @decorator
    """

    no_args = False
    redirect_to = setting_kwargs.get('redirect_to', reverse_lazy('index'))

    if len(setting_args) == 1 and not setting_kwargs and callable(setting_args[0]):
        func = setting_args[0]
        no_args = True

    def decorator(func):
        @wraps(func)
        def redirect_function(self, request, *args, **kwargs):
            if not request.user.is_staff:
                return HttpResponseRedirect(redirect_to)
            
            return func(self, request, *args, **kwargs)

        return redirect_function

    if no_args:
        return decorator(func)
    else:
        return decorator

如果用户不是使用装饰器的员工，我如何让localhost:8000/categories/sample/info/重定向到localhost:8000/categories/sample/

当前NoReverseMatch at /categories/agriculture/info/

Reverse for 'category' with no arguments not found. 1 pattern(s) tried: ['categories/(?P<handle>[-a-zA-Z0-9_]+)\\Z']

Answer 1

不是 django 专家，但是我认为您面临的重定向问题与装饰器有关，但与 django 重定向本身有关。

在您的情况下，我会用“redirect_to”代替def decorator中的get_url类的函数来确定 url：

 def get_url(self):
        dict = {'handle': 'sample'}
        url = reverse('game', kwargs=dict)
        return url

在def redirect_function中：

def redirect_function(self, request, *args, **kwargs):
            if not request.user.is_staff:
                return HttpResponseRedirect(self.get_url())
            
            return func(self, request, *args, **kwargs)

        return redirect_function

    if no_args:
        return decorator(func)
    else:
        return decorator

这就是我用于任何类型重定向的结构，目前我暂时没有遇到任何问题。 我希望它有帮助