2016-09-30 13 views
0

ユーザーサインアップのためにrestframeworkを使用しています 保存する前にランダムなパスワードを生成する必要があります。 ¿これをどのように無効にできますか?ありがとう。AbstractBaseUserを拡張したdjangoモデル保存関数をオーバーライドします

# ViewSet. 
class AgentViewSet(viewsets.ModelViewSet): 
    queryset = Agents.objects.all() 
    serializer_class = AgentSerializer 
    permission_classes = (permissions.AllowAny,) 
    http_method_names = ['post'] 

# Model. 
class Agents(AbstractBaseUser): 
    oldkey = models.CharField(max_length=32, blank=True, null=True) 
    is_main_user = models.IntegerField(blank=True, null=True) 
    name = models.CharField(max_length=45, blank=True, null=True) 
    lastname = models.CharField(max_length=45, blank=True, null=True) 
    email1 = models.CharField(
     max_length=100, blank=True, null=True, unique=True 
    ) 
    email2 = models.CharField(max_length=100, blank=True, null=True) 
    city = models.CharField(max_length=200, blank=True, null=True) 
    country = models.ForeignKey(
     'bdetail.Countries', models.DO_NOTHING, db_column='country', 
     blank=True, null=True 
    ) 
    username = models.CharField(max_length=30, blank=True, null=True) 
    image = models.CharField(max_length=80, blank=True, null=True) 
    num_access = models.IntegerField(blank=True, null=True) 
    phone1 = models.CharField(max_length=20, blank=True, null=True) 
    phone2 = models.CharField(max_length=20, blank=True, null=True) 
    phone3 = models.CharField(max_length=20, blank=True, null=True) 
    whatsapp = models.CharField(max_length=20, blank=True, null=True) 
    permissions = models.CharField(
     max_length=10, blank=True, null=True, default='limited') 
    fb_profile_id = models.CharField(max_length=30, blank=True, null=True) 
    fb_access_token = models.TextField(blank=True, null=True) 
    clients = models.ForeignKey(
     'Clients', models.DO_NOTHING, blank=True, null=True 
    ) 
    last_login_time = models.IntegerField(blank=True, null=True) 
    about = models.TextField('Acerca de mi', blank=True) 

    USERNAME_FIELD = 'email1' 
    REQUIRED_FIELDS = ['name'] 
    objects = MyUserManager() 

    class Meta: 
     db_table = 'agents' 

    def get_full_name(self): 
     return self.name, ' ', self.lastname 

    def get_short_name(self): 
     return self.name 

    def has_perm(self, perm, obj=None): 
     return True 

    def save(self, *args, **kwargs): 
     raise Exception('ingres') 

    def has_module_perms(self, app_label): 
     "Does the user have permissions to view the app `app_label`?" 
     # Simplest possible answer: Yes, always 
     return True 

    def __unicode__(self): 
     return unicode(self.name) or u'' 

    @property 
    def is_staff(self): 
     "Is the user a member of staff?" 
     # Simplest possible answer: All admins are staff 
     return False 

答えて

1

AgentSerializer(serializers.py)models.pyで

class AgentSerializer(serializers.ModelSerializer): 
    class Meta: 
     model = Agents 

    def create(self, validated_data): 
     return Agents.objects.create(**validated_data) 
views.py

class AgentViewSet(viewsets.ModelViewSet): 
    queryset = Agents.objects.all() 
    serializer_class = AgentSerializer 
    permission_classes = (permissions.AllowAny,) 
    http_method_names = ['post'] 

    def create(self, request): 
     serializer = self.serializer_class(data=request.data) 

     if serializer.is_valid(): 
      Agents.objects.create_user(**serializer.validated_data) 

      return Response(serializer.validated_data, status=status.HTTP_201_CREATED) 

     return JSONResponse(serializer.errors, status=400) 

あなたのコードに追加または更新:

ような何か
class MyUserManager(BaseUserManager): 
    def _create_user(self, email, password, 
       is_staff, **extra_fields): 
     password= # Generate random password 
     email = self.normalize_email(email) 
     user = self.model(email=email, 
     is_staff=is_staff, **extra_fields) 
     user.set_password(password) 
     user.save(using=self._db) 
     return user 
    def create_user(self, email, password=None, **extra_fields): 
     return self._create_user(email, password, False 
          **extra_fields) 
+0

キーは、AgentViewSetのcreate createメソッドです。ありがとう –

0

あなたMyUserManagercreate_userメソッドをオーバーライドしてみます。

def create_user(self, **kwargs): 
    kwargs['password'] = # Generate random password 
    return super().create_user(**kwargs) 
+0

これは機能しません、create_userはrestframeworkによって呼び出されません –

+0

RFを使用するプロジェクトがあるのでこれは奇妙です。カスタムサインアップビュー/メソッドを実装しましたか? – lucasnadalutti

関連する問題