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)

Вернуться на верх