소스 검색

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.')
         else:
             context['invitation_valid'] = False
+            messages.error(self.request, 'Se requiere un código de invitación.')
         return context

+ 4 - 6
src/reymotausers/forms.py

@@ -9,9 +9,8 @@ from accounts.models import InvitationCode
 class ReyMotaUserCreationForm(UserCreationForm):
     invitation_code = forms.CharField(
         max_length=32,
-        required=False,
         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:
@@ -28,7 +27,7 @@ class ReyMotaUserCreationForm(UserCreationForm):
     def clean_invitation_code(self):
         code = self.cleaned_data.get('invitation_code')
         if not code:
-            return code
+            raise ValidationError("El código de invitación es obligatorio.")
 
         try:
             invitation = InvitationCode.objects.get(code=code)
@@ -47,9 +46,8 @@ class ReyMotaUserCreationForm(UserCreationForm):
         if commit:
             user.save()
             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