В моей модели у меня есть семейная таблица, в которой пользователь может вводить сведения о своей семье и сохранять свою информацию. Но как мне получить / запросить всех членов семьи на основе текущего userId?

Например: userId = 1 добавил 2 члена семьи, мать и отца. Как мне получить этих 2 членов семьи на основе запроса userId текущего пользователя?

Вот мой код:

< STRONG> моделей

class MyUser(AbstractUser):
    userId = models.AutoField(primary_key=True)
    gender = models.CharField(max_length=6, blank=True, null=True)
    nric = models.CharField(max_length=40, blank=True, null=True)
    birthday = models.DateField(blank=True, null=True)
    birthTime = models.TimeField(blank=True, null=True)

class Family(models.Model):
    userId = models.ForeignKey(MyUser)
    relationship = models.CharField(max_length=100)
    first_name = models.CharField(max_length=50)
    last_name = models.CharField(max_length=50)
    gender = models.CharField(max_length=6, blank=True, null=True)

< Сильный > сериализаторов

class MyUserSerializer(serializers.ModelSerializer):
    valid_time_formats = ['%H:%M', '%I:%M%p', '%I:%M %p']
    birthTime = serializers.TimeField(format='%I:%M %p', input_formats=valid_time_formats, allow_null=True, required=False)

    class Meta:
        model = MyUser
        fields = ['userId', 'username', 'email', 'first_name', 'last_name', 'gender', 'nric', 'birthday', 'birthTime']
        read_only_fields = ('userId',)
        extra_kwargs = {"password": {"write_only": True}}

    def update(self, instance, validated_data):
        for attr, value in validated_data.items():
            if attr == 'password':
                instance.set_password(value)
            else:
                setattr(instance, attr, value)
        instance.save()
        return instance


class FamilySerializer(serializers.ModelSerializer):

    class Meta:
        model = Family
        fields = ('id', 'userId', 'first_name', 'last_name', 'gender', 'relationship')

просмотры

class MyUserViewSet(viewsets.ModelViewSet):
    permission_classes = [AllowAny]
    queryset = MyUser.objects.all()
    serializer_class = MyUserSerializer
    filter_backends = (filters.SearchFilter,)
    search_fields = ('=userId', 'username', 'email', 'first_name', 'last_name')


class FamilyViewSet(viewsets.ModelViewSet):
    permission_classes = [AllowAny]
    queryset = Family.objects.all()
    serializer_class = FamilySerializer


def get_object(self, *args, **kwargs):
    return self.request.user

def perform_update(self, serializer):
    super(CurrentFamilyView, self).perform_update(serializer)
    user = serializer.instance
    if settings.SEND_ACTIVATION_EMAIL and not user.is_active:
        context = {'user': user}
        to = [MyUser(user)]
        email.ActivationEmail(self.request, context).send(to)

Может ли кто-нибудь помочь мне получить семью текущего пользователя

0
Jin Nii Sama 3 Янв 2018 в 11:07

2 ответа

Лучший ответ

Вы можете попробовать это:

class Family(models.Model):
    userId = models.ForeignKey(MyUser, related_name='user_family')
    relationship = models.CharField(max_length=100)
    first_name = models.CharField(max_length=50)
    last_name = models.CharField(max_length=50)
    gender = models.CharField(max_length=6, blank=True, null=True)

serializers.py

class FamilySerializer(serializers.ModelSerializer):

    class Meta:
        model = Family
        fields = ('id', 'first_name', 'last_name', 'gender', 'relationship')


class MyUserSerializer(serializers.ModelSerializer):

    valid_time_formats = ['%H:%M', '%I:%M%p', '%I:%M %p']
    birthTime = serializers.TimeField(format='%I:%M %p', input_formats=valid_time_formats, allow_null=True, required=False)
    my_family = FamilySerializer(many=True, source='user_family')

    class Meta:
        model = MyUser
        fields = ('userId', 'username', 'email', 'first_name', 'last_name', 'gender', 'nric', 'birthday', 'birthTime', 'my_family')
        read_only_fields = ('userId',)
        extra_kwargs = {"password": {"write_only": True}}

MyUserSerializer теперь будет показывать userid связанных членов семьи. Если вы хотите получить текущего члена семьи пользователя, вам необходимо перезаписать метод get_queryset.

Ваш views.py :

class MyUserViewSet(viewsets.ModelViewSet):

    serializer_class = MyUserSerializer
    get_queryset(self):
        return  MyUser.objects.get(id=self.request.user.id)

Теперь представление вернет текущих членов семьи пользователей.

Обновления

  1. Добавлен related_name в модель Family.
  2. Добавлено source в MyUserSerializer
2
mohammadjh 4 Янв 2018 в 08:20

Предположим, у нас есть пользователь с id = 1

usr = MyUser.objects.get(id=1)
usr_famly = usr.family_set.values()

usr_famly будет иметь все семейные данные, связанные с user with id=1

1
JPG 3 Янв 2018 в 09:45