Есть ли способ агрегировать поле в Django Rest Framework, в котором будет суммирование одного поля - PullRequest
Новая
       31

Есть ли способ агрегировать поле в Django Rest Framework, в котором будет суммирование одного поля

0 голосов
/ 29 апреля 2019

В настоящее время я выяснил, как агрегировать один столбец в моем serializers.py , но сумма моего поля salary перейдет в total_salary *Поле 1006 * в моем сериализаторе и total_salary не в моих моделях.Теперь моя проблема в том, как я могу сделать что-то подобное в своем API ниже: 

"total_salary": 1422.05,
{
        "id": "8c1810d9-b799-46a9-8506-3c18ef0067f8",
        "date": "2019-04-27",
        "virtual_assistant": "Joevie",
        "time_in": "2019-04-27T22:20:13+08:00",
        "time_out": "2019-04-28T05:20:13+08:00",
        "hours": "7.00",
        "client_name": "landmaster",
        "rate": "90.00",
        "salary": "630.00",
        "status": "APPROVED-BY-THE-MANAGER",
        "notes": ""
    },

Существующий похож на это. 

 {
        "id": "8c1810d9-b799-46a9-8506-3c18ef0067f8",
        "total_salary": 1422.05,
        "date": "2019-04-27",
        "virtual_assistant": "Joevie",
        "time_in": "2019-04-27T22:20:13+08:00",
        "time_out": "2019-04-28T05:20:13+08:00",
        "hours": "7.00",
        "client_name": "landmaster",
        "rate": "90.00",
        "salary": "630.00",
        "status": "APPROVED-BY-THE-MANAGER",
        "notes": ""
 },

В настоящее время я нашел обходной путь: сумма зарплаты в ListView на данный момент, и каждый раз, когда пользователь будет искать определенный месяц.Расчет будет меняться в зависимости от месяца, который искал пользователь.Код ниже. 

def get(self, request, *args, **kwargs):
        search = request.GET.get('search')
        user = request.user.staffs.full_name
        current_month = datetime.date.today().month
        current_year = datetime.date.today().year
        payroll_list = VaPayroll.objects.all()
        payroll_data = payroll_list.filter(Q(virtual_assistant=user),
                                           Q(date__month=current_month),
                                           Q(status='APPROVED-BY-THE-MANAGER'))
        total_salary = VaPayroll.objects.filter(Q(virtual_assistant=user), 
                                                Q(date__month=current_month),
                                                Q(status='APPROVED-BY-THE-MANAGER'),
                                                Q(date__year=current_year)).aggregate(Sum('salary'))
        if search:
            payroll_data = payroll_list.filter(Q(virtual_assistant=user),
                                               Q(status='APPROVED-BY-THE-MANAGER'),
                                               Q(date__icontains=search))
            total_salary = VaPayroll.objects.filter(Q(virtual_assistant=user),
                                                    Q(status='APPROVED-BY-THE-MANAGER'),
                                                    Q(date__month=search),
                                                    Q(date__year=current_year)).aggregate(Sum('salary'))
        context = {
            'total_salary': total_salary,
            'payroll_data': payroll_data
        }
        return render(request, self.template_name, context)

Это из моего serializers.py 

class VaPayrollSerializer(serializers.ModelSerializer):
    total_salary = serializers.SerializerMethodField()

    class Meta:
        model = VaPayroll
        fields = '__all__'

    def get_total_salary(self, obj):
        user = self.context['request'].user.staffs.full_name
        totalsalary = VaPayroll.objects.filter(Q(status='APPROVED-BY-THE-MANAGER'),
                                               Q(virtual_assistant=user),
                                               Q(date__month=datetime.date.today().month),
                                               Q(date__year=datetime.date.today().year)).aggregate(total_salary=Sum('salary'))
        return totalsalary['total_salary']

Это из моего набора моделей a get_queryset 

def get_queryset(self):
        current_month = datetime.date.today().month
        current_year = datetime.date.today().year
        queryset = VaPayroll.objects.filter(Q(virtual_assistant=self.request.user.staffs.full_name), 
                                            Q(date__month=current_month), 
                                            Q(date__year=current_year))
        return queryset

В моем запланированном сценарии использования я ожидаю, что в моем API будет один столбец, в котором будет указана сумма всей зарплаты, чтобы я мог по отдельности вызывать это поле во внешнем интерфейсе без необходимости обновлять страницу для повторного вычисления общей зарплаты.Или это правильный способ сделать это?Или я должен просто придерживаться View, чтобы просто пересчитать общую зарплату.

1 Ответ

1 голос
/ 29 апреля 2019

Вы не можете получить json так, как хотите, потому что это недопустимый формат json.Я рекомендую попробовать что-то вроде этого: 

"salary": {
           "total": 1422.05,
           "detail": {
                      "id": "8c1810d9-b799-46a9-8506-3c18ef0067f8",
                      "date": "2019-04-27",
                      "virtual_assistant": "Joevie",
                      "time_in": "2019-04-27T22:20:13+08:00",
                      "time_out": "2019-04-28T05:20:13+08:00",
                      "hours": "7.00",
                      "client_name": "landmaster",
                      "rate": "90.00",
                      "salary": "630.00",
                      "status": "APPROVED-BY-THE-MANAGER",
                      "notes": ""
             }
      }

Для этого вы должны проинструктировать ваш сериализатор, используя to_representation() метод: 

class VaPayrollSerializer(serializers.ModelSerializer):
    class Meta:
        model = VaPayroll
        fields = '__all__'

    def to_representation(self, instance):
        original_representation = super().to_representation(instance)

        representation = {
            'total': self.get_total_salary(instance),
            'detail': original_representation,
        }

        return representation

    def get_total_salary(self, obj):
        user = self.context['request'].user.staffs.full_name
        totalsalary = VaPayroll.objects.filter(Q(status='APPROVED-BY-THE-MANAGER'),
            Q(virtual_assistant=user),
            Q(date__month=datetime.date.today().month),
        Q(date__year=datetime.date.today().year)).aggregate(total_salary=Sum('salary'))

        return totalsalary['total_salary']
...