Преглед на файлове

Restauro código de invitación

Celestino Rey преди 5 месеца
родител
ревизия
aab1c24b10
променени са 2 файла, в които са добавени 5 реда и са изтрити 6 реда
  1. 1 0
      src/accounts/views.py
  2. 4 6
      src/reymotausers/forms.py

+ 1 - 0
src/accounts/views.py

@@ -38,4 +38,5 @@ class SignUpView(CreateView):
                 messages.error(self.request, 'El código no es válido.')
                 messages.error(self.request, 'El código no es válido.')
         else:
         else:
             context['invitation_valid'] = False
             context['invitation_valid'] = False
+            messages.error(self.request, 'Se requiere un código de invitación.')
         return context
         return context

+ 4 - 6
src/reymotausers/forms.py

@@ -9,9 +9,8 @@ from accounts.models import InvitationCode
 class ReyMotaUserCreationForm(UserCreationForm):
 class ReyMotaUserCreationForm(UserCreationForm):
     invitation_code = forms.CharField(
     invitation_code = forms.CharField(
         max_length=32,
         max_length=32,
-        required=False,
         label="Código de Invitación",
         label="Código de Invitación",
-        help_text="Introduce el código que has recibido (opcional)"
+        help_text="Introduce el código que has recibido"
     )
     )
 
 
     class Meta:
     class Meta:
@@ -28,7 +27,7 @@ class ReyMotaUserCreationForm(UserCreationForm):
     def clean_invitation_code(self):
     def clean_invitation_code(self):
         code = self.cleaned_data.get('invitation_code')
         code = self.cleaned_data.get('invitation_code')
         if not code:
         if not code:
-            return code
+            raise ValidationError("El código de invitación es obligatorio.")
 
 
         try:
         try:
             invitation = InvitationCode.objects.get(code=code)
             invitation = InvitationCode.objects.get(code=code)
@@ -47,9 +46,8 @@ class ReyMotaUserCreationForm(UserCreationForm):
         if commit:
         if commit:
             user.save()
             user.save()
             code = self.cleaned_data.get('invitation_code')
             code = self.cleaned_data.get('invitation_code')
-            if code:
-                invitation = InvitationCode.objects.get(code=code)
-                invitation.mark_as_used(user)
+            invitation = InvitationCode.objects.get(code=code)
+            invitation.mark_as_used(user)
         return user
         return user