Django:如何编写一个在重定向函数后接受输入的装饰器

Question

我想写一个decorator，它接受来自重定向函数的输入(值：id_contrat(。如果我再解释一下。用户在表单中选择一个值，然后将该值重定向到另一个视图"；家"；。我想检查用户是否拥有此值的权限！所以我需要把它放在我的装饰师那里。如何将此值添加到我的装饰器中。

I tried something but it is not working : `TypeError: wrapper_func() got an unexpected keyword argument 'id_contrat'`

我的看法：

@authenticated_user
def selectcontrat(request) :
context = initialize_context(request)
form_client = SelectClient(request.POST, user=request.user)
if form_client.is_valid():
id_contrat = request.POST.get("ID_Customer")

return redirect(reverse('home', args=(id_contrat,)))
context['form_client'] = form_client
return render(request, 'base/selectcontrat.html', context)

@authenticated_user
@check_user_rights
def home(request, id_contrat=None):
context = initialize_context(request)
return render(request, 'home.html', context)

url定义：

from django.urls import path
from . import views
urlpatterns = [
path('home/<int:id_contrat>/', views.home, name="home"),
path('', views.loginAD, name="login"),
path('signin', views.sign_in, name='signin'),
path('callback', views.callback, name='callback'),
path('selectcontrat', views.selectcontrat, name='selectcontrat')

形式：

class SelectClient(forms.Form):
ID_Customer = forms.ChoiceField(label="Company :")  


def __init__(self, *args, **kwargs) :
self.user = kwargs.pop('user')
super(SelectClient, self).__init__(*args, **kwargs)
id_client_list = AADJNTGroup.objects.filter(ID_User_id=self.user.id).values_list('ID_Group_id', flat=True)
id_client_list = list(id_client_list)
client_choices = Groups.objects.all().filter(ID__in=id_client_list).values_list('IDCustomer','GroupName')

self.fields['ID_Customer'].choices = client_choices

编辑：装饰器

def check_user_rights(id_contrat(:

def wrapper_func(view_func) :
@wraps(view_func)
def wrapper(request, *args, **kwargs) :
print(kwargs["id_contrat"])
return view_func(request, *args, **kwargs)
return wrapper
return wrapper_func

Answer 1

您应该从装饰器中删除id_contrat参数，除非您计划在装饰视图时传递此参数，例如@check_user_rights(my_id_contrat)

def check_user_rights():
def wrapper_func(view_func):
@wraps(view_func)
def wrapper(request, *args, **kwargs):
# This prints all the keyword arguments passed to your view
# and should contain id_contrat from your url
print(kwargs)
return view_func(request, *args, **kwargs)
return wrapper
return wrapper_func
@check_user_rights()
def home(request, id_contrat=None):
context = initialize_context(request)
return render(request, 'home.html', context)

您还可以在"制作自定义django视图装饰器(带args(时遇到问题"的答案的注释中找到一些价值