diff --git a/db.sqlite3 b/db.sqlite3 index b50042d..b2fa287 100644 Binary files a/db.sqlite3 and b/db.sqlite3 differ diff --git a/filme/views.py b/filme/views.py index 656ce32..bae5a40 100644 --- a/filme/views.py +++ b/filme/views.py @@ -1,7 +1,7 @@ from django.shortcuts import render, redirect, reverse from .models import Filme, Usuario from django.views.generic import TemplateView, ListView, DetailView, FormView, UpdateView -from django.contrib.auth.mixins import LoginRequiredMixin +from django.contrib.auth.mixins import LoginRequiredMixin, UserPassesTestMixin from django.contrib.auth.views import LoginView from .forms import CustomLoginForm, CriarContaForm, FormHomepage @@ -76,12 +76,17 @@ def get_queryset(self): return None -class EditarPerfil(LoginRequiredMixin, UpdateView): +class EditarPerfil(LoginRequiredMixin, UserPassesTestMixin, UpdateView): template_name = 'editarperfil.html' model = Usuario # Irá criar um form com os fields definidos na lista fields = ['first_name', 'last_name', 'email'] + # Validação do usuário on link + def test_func(self): + user = self.get_object() + return self.request.user == user + # Assim que editar será redirecionado para homefilmes def get_success_url(self): return reverse('filme:homefilmes')