From 6151c73feb01a0c39e143e8b18991ddc32d71458 Mon Sep 17 00:00:00 2001 From: Edward Ribeiro Date: Tue, 23 Feb 2016 09:15:03 -0300 Subject: [PATCH] =?UTF-8?q?Adiciona=20gera=C3=A7=C3=A3o=20de=20"hash".?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- materia/views.py | 12 +++++++++++- 1 file changed, 11 insertions(+), 1 deletion(-) diff --git a/materia/views.py b/materia/views.py index 5449f88e9..8b810eb32 100644 --- a/materia/views.py +++ b/materia/views.py @@ -1,5 +1,7 @@ from datetime import datetime +from random import choice from re import sub +from string import ascii_letters, digits from django.contrib import messages from django.core.exceptions import ObjectDoesNotExist @@ -1495,6 +1497,10 @@ class AcompanhamentoMateriaView(FormMixin, materia_legislativa_crud.CrudDetailView): template_name = "materia/acompanhamento_materia.html" + def get_random_chars(self): + s = ascii_letters + digits + return ''.join(choice(s) for i in range(choice([6, 7]))) + def get(self, request, *args, **kwargs): pk = self.kwargs['pk'] materia = MateriaLegislativa.objects.get(id=pk) @@ -1511,12 +1517,16 @@ class AcompanhamentoMateriaView(FormMixin, email = form.cleaned_data['email'] usuario = request.user + hash_txt = self.get_random_chars() + try: AcompanhamentoMateria.objects.get( email=email, - materia=materia) + materia=materia, + hash=hash_txt) except ObjectDoesNotExist: acompanhar = form.save(commit=False) + acompanhar.hash = hash_txt acompanhar.materia = materia acompanhar.usuario = usuario.username acompanhar.save()