Django : Как написать декоратор, который принимает входные данные после функции перенаправления
<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(id_contrat)
return view_func(request, *args, **kwargs)
return wrapper
return wrapper_func
Вам следует удалить аргумент 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)
Вы также можете найти некоторую ценность в комментариях к ответу на Having trouble making a custom django view decorator (with args)