Фильтр Django Rest Framework с использованием внешнего ключа - PullRequest
Новая
       28

Фильтр Django Rest Framework с использованием внешнего ключа

0 голосов
/ 07 сентября 2018

У меня есть следующая структура БД с 3 таблицами, как показано ниже

Таблица лиц 

Id  U_id    PersonName
101 12       Iron Man
201 12       Spider Man
301 15       Thanos

Таблица EarnTypes 

Id  U_id    TypeEarning
10  12       Salary
20  15       Lottery
30  15       Gambling

Таблица доходовEntry 

ID     U_Id P_Id    EarnType_Name   Earn_Amt    Earn_Date
1001    12  101         10           5$         8-Jun-2017
3001    15  301         20           25$        7-Apr-2018
4001    12  201         10           50$        19-Apr-2018

Мой список просмотра API-интерфейса выглядит следующим образом. Я фильтрую данные только по зарегистрированным пользователям. 

    class EarningEntryAPIView(mixins.CreateModelMixin,generics.ListAPIView):
        permission_classes      = [IsOwnerOnly]
        serializer_class        = EarningsSerializer
        #
        def get_queryset(self):
            request = self.request
            #print (request.user)
            qs = EarningsEntry.objects.filter(U_id=self.request.user)
            query = request.GET.get('q')
            if query is not None:
                qs = qs.filter(Earning_Type_id__EarningTypeName__contains=query)
            return qs

        def post(self,request,*args,**kwargs):
            return self.create(request,*args,**kwargs)

        def perform_create(self, serializer):
            serializer.save(id=self.request.user)

Моя модель EarningEntry выглядит следующим образом. 

def upload_file(instance, filename):
    return "earnings/{user}/{filename}".format(user=instance.id, filename=filename)


class EarningsQuerySet(models.QuerySet):
    def serialize(self):
        list_values = list(
            self.values('Id', 'U_id', 'P_id', 'Earning_Type_id', 'Ear_Amt', 'Ear_Img', 'Ear_date', 'Ear_comm'))
        print(list_values)
        return json.dumps(list_values, sort_keys=True, indent=1, cls=DjangoJSONEncoder)


class EarningssManager(models.Manager):
    def get_queryset(self):
        return EarningsQuerySet(self.model, using=self._db)


# Create your models here.
class EarningsEntry(models.Model):
    Id = models.AutoField(primary_key=True)
    U_id=models.ForeignKey(settings.AUTH_USER_MODEL,on_delete=models.CASCADE)
    P_id =models.ForeignKey(personmodel.Person,on_delete=models.CASCADE)
    Earning_Type_id = models.ForeignKey(eartypemodel.EarningTypes,on_delete=models.CASCADE)
    Ear_Amt = models.FloatField(null=False,blank=False)
    Ear_Img = models.ImageField(null=True,blank=True)
    Ear_date = models.DateField("ExpenseDate",null=False,blank=False)
    Ear_comm = models.TextField()
    objects = EarningssManager()

    def __str__(self):
        return str(self.U_id) + str(self.P_id) + str(self.Ear_Amt) + str(self.Ear_Img) +str(self.Ear_Img)+str(self.Earning_Type_id)+ \
               str(self.Ear_date) + str(self.Ear_comm) + str(self.Id) or ""

    def serialize(self):
        data={
            'Id': self.Id,
            'U_id': self.U_id,
            'P_id': self.P_id,
            'Earning_Type_id': self.Earning_Type_id,
            'Ear_Amt': self.Ear_Amt,
            'Ear_Img': self.Ear_Img,
            'Ear_date': self.Ear_date,
            'Ear_comm': self.Ear_comm
        }
        data = json.dumps(data,sort_keys=True,indent=1,cls=DjangoJSONEncoder)
        return data


    @property
    def owner(self):
        return self.Id

Теперь, когда я пытаюсь просмотреть API для вставки или просмотра списка, я получаю все значения для ссылок на внешние ключи U_id, P_id и Earnig_Type_id, что неверно.

Как я могу отфильтровать это, основываясь только на моем зарегистрированном пользователе. В моем текущем представлении я просматриваю все.

Пожалуйста, обратитесь к этому экрану, где nirav и nirav13 - два разных пользователя. enter image description here

1 Ответ

0 голосов
/ 07 сентября 2018

Прежде всего, не называйте поля заглавными буквами.во-вторых, не пишите _id в названии внешних ключей, взгляните на django docs .в любом случае, вы должны переопределить __init__ и отфильтровать набор запросов для этих полей: 

class EarningsSerializer(serializer.Modelserializer):
    def __init__(self, *args, **kwargs):
        super(EarningsSerializer, self).__init__(*args, **kwargs)
        request_user = self.context['request'].user
        self.fields['P_id'].queryset = Person.objects.filter(U_id=request_user)
        self.fields['Earning_Type_id'].queryset = EarningTypes.objects.filter(U_id=request_user)
...