Как написать собственный декоратор в views Django?

Question

[sportik174]

Добрый день!
Есть такой код, который используется для проверки авторизации пользователя (знаю что есть стандартный функционал но нужно именно кастомно)

def decorator_login_check(func):
    def wrapper(request):
        try:
            user = User.objects.get(id=request.session['loged_user'])
        except Exception as ex:
            return redirect('auth_login')
        func(request)
    return wrapper


@decorator_login_check
def index(request):
    """ Главная страница """
    user = User.objects.get(id=request.session['loged_user'])
    context = {
        'user': user
    }
    return render(request, 'main/index.html', context)

Выдает вот такую ошибку:


Что я не правильно написал? Помогите пожалуйста

Answer 1

[Сергей Горностаев]

return func(request)

Comments

[sportik174]

Спасибо!) Капец одна ошибка и в итоге не работает ничего)

[sportik174]

Теперь возникла еще одна проблема если кроме request есть еще аргумент

@decorator_login_check
def users_edit(request, id):
    """ Редактирование пользователя """
    user = User.objects.get(id=id)
    roles = UserRole.objects.all()
    cities = City.objects.all()
    context = {
        'user': user,
        'roles': roles,
        'cities': cities
    }
    return render(request, 'main/users_edit.html', context)

То выдает такую ошибку:

[sportik174]

Решил этот вопрос вот так:

def decorator_login_check(func):
    def wrapper(request, *args, **kwargs):
        try:
            user = User.objects.get(id=request.session['loged_user'])
        except Exception as ex:
            return redirect('auth_login')
        return func(request, *args, **kwargs)
    return wrapper

Правильно ли это?

[Сергей Горностаев]

sportik174, да.