Django Rest Framework создает пользователя и профиль пользователя

Question

Я пытаюсь создать представление, которое создаст нового пользователя, используя мой API. Я использую пользовательскую модель для своего Пользователя, а также создал модель под названием Профиль для управления данными, которые не связаны с проверкой подлинности.

Я новичок в мире Django, и это может быть довольно сложно.

Вот мой models.py

class UserManager(BaseUserManager): 
    def create_user(self, email, password=None): 
     if not email: 
      raise ValueError('User must have an email address') 

     user = self.model(
      email = self.normalize_email(email), 
     ) 
     user.set_password(password) 
     user.save() 
     return user 

    def create_superuser(self, email, password): 
     user = self.create_user(email, password=password) 
     user.is_admin = True 
     user.save() 
     return user 


class User(AbstractBaseUser): 
    objects = UserManager() 
    email = models.EmailField(unique=True, db_index=True) 
    created = models.DateTimeField('created', auto_now_add=True) 
    modified = models.DateTimeField(auto_now=True) 

    is_active = models.BooleanField('active', default=True) 
    is_admin = models.BooleanField('admin', default=False) 

    USERNAME_FIELD = 'email' 

    ordering = ('created',) 

    def is_staff(self): 
     return self.is_admin 

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

    def has_module_perms(self, app_label): 
     return True 

    def get_short_name(self): 
     return self.email 

    def get_full_name(self): 
     return self.email 

    def __unicode__(self): 
     return self.email 


class Profile(models.Model): 
    GENDER = (
     ('M', 'Homme'), 
     ('F', 'Femme'), 
    ) 

    user = models.OneToOneField(settings.AUTH_USER_MODEL) 
    first_name = models.CharField(max_length=120, blank=False) 
    last_name = models.CharField(max_length=120, blank=False) 
    gender = models.CharField(max_length=1, choices=GENDER) 
    zip_code = models.CharField(max_length=5, validators=[MinLengthValidator(5)], blank=False) 

    def __unicode__(self): 
     return u'Profile of user: {0}'.format(self.user.email) 


def create_profile(sender, instance, created, **kwargs): 
    if created: 
     Profile.objects.create(user=instance) 
post_save.connect(create_profile, sender=User) 


def delete_user(sender, instance=None, **kwargs): 
    try: 
     instance.user 
    except User.DoesNotExist: 
     pass 
    else: 
     instance.user.delete() 
post_delete.connect(delete_user, sender=Profile) 

Вот мой serializers.py

class ProfileSerializer(serializers.ModelSerializer): 
    class Meta: 
     model = Profile 
     fields = ('first_name', 'last_name', 'gender', 'zip_code',) 


class UserSerializer(serializers.ModelSerializer): 
    profile = ProfileSerializer(required=True) 
    class Meta: 
     model = User 
     fields = ('url', 'email', 'profile', 'created',) 

А вот views.py

class UserList(generics.ListCreateAPIView): 
    permission_classes = (IsAuthenticatedOrWriteOnly,) 
    serializer_class = UserSerializer 

    def get_queryset(self): 
     if self.request.user.is_staff: 
      return User.objects.all() 
     else: 
      return self.request.user 

    def post(self, request, format=None): 
     serializer = ProfileSerializer(data=request.data) 
     print serializer.__dict__ 
     if serializer.is_valid(): 
      print "valid" 
     return Response("placeholder", status=status.HTTP_201_CREATED) 

Моя цель состоит в том, чтобы быть в состоянии для создания пользователя и профиля одновременно, когда вы можете все это сделать, как я могу это достичь?

Answer 1

Действительно, Django Rest Framework не может справиться с этой работой с вложенными отношениями, тогда вы должны сами реализовать эти методы. Я приведу вам пример того, как выглядит ваш код.

Ваше мнение:

class UserList(generics.ListCreateAPIView): 
    permission_classes = (IsAuthenticatedOrWriteOnly,) 
    serializer_class = UserSerializer 

    def post(self, request, format=None): 
     serializer = UserSerializer(data=request.data) 
     if serializer.is_valid(): 
      serializer.save() 
      return Response(serializer.data, status=status.HTTP_201_CREATED) 
     return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST) 

Теперь save метод вашего сериализатором будет вызывать метод create, когда вы хотите создать объект и update метод, когда вы хотите обновить объект. Итак, давайте реализуем метод create вашего UserSerializer, который создаст профиль и пользователя. Вот простой пример того, что ваш UserSerializer должен выглядеть следующим образом:

class UserSerializer(serializers.ModelSerializer): 
    profile = ProfileSerializer(required=True) 
    class Meta: 
     model = User 
     fields = ('url', 'email', 'profile', 'created',) 

    def create(self, validated_data): 

     # create user 
     user = User.objects.create(
      url = validated_data['url'], 
      email = validated_data['email'], 
      # etc ... 
     ) 

     profile_data = validated_data.pop('profile') 
     # create profile 
     profile = Profile.objects.create(
      user = user 
      first_name = profile_data['first_name'], 
      last_name = profile_data['last_name'], 
      # etc... 
     ) 

     return user 

Как я уже сказал, это пример, вы должны закончить это делать то, что вы хотите сделать, но сейчас, вы знаете, как это сделать :) Чтобы определить поведение во время обновления, выполните метод update.