From 59cb0d9e7a1a4201720fd6c5b6c7ec6200d2bb7d Mon Sep 17 00:00:00 2001
From: Max Yakovenko <mxyakovenko9@gmail.com>
Date: Tue, 31 Jul 2018 11:25:15 +0300
Subject: [PATCH] remove captcha field

---
 accounts_ext/forms.py | 22 +++++-----------------
 1 file changed, 5 insertions(+), 17 deletions(-)

diff --git a/accounts_ext/forms.py b/accounts_ext/forms.py
index ede88b2..a44219d 100644
--- a/accounts_ext/forms.py
+++ b/accounts_ext/forms.py
@@ -1,11 +1,9 @@
 import logging
-from captcha.fields import CaptchaField
 from crispy_forms.helper import FormHelper
 from crispy_forms.layout import Layout, Field, Div, Submit, HTML
 from django import forms
 from django.conf import settings
 from django.contrib.auth import get_user_model, password_validation
-from django.forms import ALL_FIELDS
 from django.urls import reverse, reverse_lazy
 from django.utils.translation import ugettext_lazy as _
 from django_email_blacklist import DisposableEmailChecker
@@ -18,18 +16,11 @@ from django.contrib.auth.forms import (
     UserCreationForm as UserCreationFormBase
 )
 from django.contrib.auth.forms import UsernameField
-from core.models import STATUS_ACTIVE, STATUS_CHOICES
-from .models import User, Profile, Company
+from core.models import STATUS_ACTIVE
+from .models import Profile, Company
 
 logger = logging.getLogger(__name__)
 
-
-# class PickUpPointsForm(layout.ModelForm):
-#     class Meta:'
-#         model = PickUpRequest
-#         fields = ('requisites',)
-
-
 class RegistrationForm(RegistrationFormUniqueEmail):
     email = forms.EmailField(label=_("E-mail"), widget=forms.EmailInput(attrs={'class': 'reg__text'}))
     password1 = forms.CharField(
@@ -51,9 +42,6 @@ class RegistrationForm(RegistrationFormUniqueEmail):
                                  widget=forms.TextInput(attrs={'class': 'reg__text'}))
     agreement = forms.BooleanField()
 
-    # captcha = CaptchaField(required=not settings.DEBUG)
-    captcha = CaptchaField(required=not settings.DEBUG)
-
     title = _('Регистрация')
 
     def clean_email(self):
@@ -76,7 +64,7 @@ class RegistrationForm(RegistrationFormUniqueEmail):
         return user
 
     class Meta:
-        model = User
+        model = get_user_model()
         fields = ('username', 'email',)
         labels = {
             'username': _('Логин'),
@@ -269,7 +257,7 @@ class UserChangeForm(UserChangeFormBase):
 
     class Meta:
         model = get_user_model()
-        fields = ('status',)
+        fields = ('status','referral_code',)
 
 
 class UserCreationForm(UserCreationFormBase):
@@ -282,7 +270,7 @@ class UserCreationForm(UserCreationFormBase):
 
     class Meta:
         model = get_user_model()
-        fields = ('email', 'username', 'status', 'is_superuser')
+        fields = ('email', 'username', 'status', 'referral_code', 'is_superuser')
         field_classes = {
             'email': forms.EmailField,
             'username': UsernameField,