add new field

core/apps/finance/serializers/income.py

@@ -1,103 +1,133 @@
 from django.db import transaction
-
 from rest_framework import serializers
 
 from core.apps.finance.models import Income
 
 
 class IncomeListSerializer(serializers.ModelSerializer):
-    cash_transaction = serializers.SerializerMethodField(method_name='get_cash_transaction')
-    payment_type = serializers.SerializerMethodField(method_name='get_payment_type')
-    project_folder = serializers.SerializerMethodField(method_name='get_project_folder')
-    project = serializers.SerializerMethodField(method_name='get_project')
-    counterparty = serializers.SerializerMethodField(method_name='get_counterparty')
-    type_income = serializers.SerializerMethodField(method_name='get_type_income')
-    user = serializers.SerializerMethodField(method_name='get_user')
+    cash_transaction = serializers.SerializerMethodField(
+        method_name="get_cash_transaction"
+    )
+    payment_type = serializers.SerializerMethodField(method_name="get_payment_type")
+    project_folder = serializers.SerializerMethodField(method_name="get_project_folder")
+    project = serializers.SerializerMethodField(method_name="get_project")
+    counterparty = serializers.SerializerMethodField(method_name="get_counterparty")
+    type_income = serializers.SerializerMethodField(method_name="get_type_income")
+    user = serializers.SerializerMethodField(method_name="get_user")
 
     class Meta:
         model = Income
         fields = [
-            'id', 'cash_transaction', 'payment_type', 'project_folder', 'project',
-            'counterparty', 'type_income', 'currency', 'price', 'exchange_rate', 'date',
-            'comment', 'file', 'audit', 'user', 'income_chat'
+            "id",
+            "cash_transaction",
+            "payment_type",
+            "project_folder",
+            "project",
+            "counterparty",
+            "type_income",
+            "currency",
+            "price",
+            "exchange_rate",
+            "date",
+            "comment",
+            "file",
+            "audit",
+            "user",
+            "income_chat",
+            "created_at",
         ]
 
     def get_cash_transaction(self, obj):
-        return {
-            'id': obj.cash_transaction.id,
-            'name': obj.cash_transaction.name
-        }
+        return {"id": obj.cash_transaction.id, "name": obj.cash_transaction.name}
 
     def get_payment_type(self, obj):
-        return {
-            'id': obj.payment_type.id,
-            'name': obj.payment_type.name
-        }
+        return {"id": obj.payment_type.id, "name": obj.payment_type.name}
 
     def get_project_folder(self, obj):
-        return {
-            'id': obj.project_folder.id,
-            'name': obj.project_folder.name
-        } if obj.project_folder else None
+        return (
+            {"id": obj.project_folder.id, "name": obj.project_folder.name}
+            if obj.project_folder
+            else None
+        )
 
     def get_project(self, obj):
-        return {
-            'id': obj.project.id,
-            'name': obj.project.name
-        } if obj.project else None
+        return {"id": obj.project.id, "name": obj.project.name} if obj.project else None
 
     def get_counterparty(self, obj):
-        return {
-            'id': obj.counterparty.id,
-            'name': obj.counterparty.name
-        } if obj.counterparty else None
+        return (
+            {"id": obj.counterparty.id, "name": obj.counterparty.name}
+            if obj.counterparty
+            else None
+        )
 
     def get_type_income(self, obj):
-        return {
-            'id': obj.type_income.id,
-            'name': obj.type_income.name
-        } if obj.type_income else None
+        return (
+            {"id": obj.type_income.id, "name": obj.type_income.name}
+            if obj.type_income
+            else None
+        )
 
     def get_user(self, obj):
-        return {
-            'id': obj.user.id,
-            'full_name': obj.user.full_name,
-        } if obj.user else None
+        return (
+            {
+                "id": obj.user.id,
+                "full_name": obj.user.full_name,
+            }
+            if obj.user
+            else None
+        )
 
 
 class IncomeCreateSerializer(serializers.ModelSerializer):
     class Meta:
         model = Income
         fields = [
-            'cash_transaction', 'payment_type', 'project_folder', 'project',
-            'counterparty', 'type_income', 'currency', 'price', 'exchange_rate', 'date',
-            'comment', 'file', 'audit'
+            "cash_transaction",
+            "payment_type",
+            "project_folder",
+            "project",
+            "counterparty",
+            "type_income",
+            "currency",
+            "price",
+            "exchange_rate",
+            "date",
+            "comment",
+            "file",
+            "audit",
         ]
 
     def create(self, validated_data):
         with transaction.atomic():
             income = Income.objects.create(
-                user=self.context.get('user'),
-                cash_transaction=validated_data['cash_transaction'],
-                payment_type=validated_data['payment_type'],
-                project_folder=validated_data.get('project_folder'),
-                project=validated_data.get('project'),
-                counterparty=validated_data.get('counterparty'),
-                type_income=validated_data.get('type_income'),
-                currency=validated_data.get('currency'),
-                price=validated_data.get('price') * validated_data.get('exchange_rate') if validated_data.get('exchange_rate') else validated_data.get('price'),
-                exchange_rate=validated_data.get('exchange_rate'),
-                date=validated_data.get('date'),
-                comment=validated_data.get('comment'),
-                file=validated_data.get('file'),
-                audit=validated_data.get('audit')
+                user=self.context.get("user"),
+                cash_transaction=validated_data["cash_transaction"],
+                payment_type=validated_data["payment_type"],
+                project_folder=validated_data.get("project_folder"),
+                project=validated_data.get("project"),
+                counterparty=validated_data.get("counterparty"),
+                type_income=validated_data.get("type_income"),
+                currency=validated_data.get("currency"),
+                price=(
+                    validated_data.get("price") * validated_data.get("exchange_rate")
+                    if validated_data.get("exchange_rate")
+                    else validated_data.get("price")
+                ),
+                exchange_rate=validated_data.get("exchange_rate"),
+                date=validated_data.get("date"),
+                comment=validated_data.get("comment"),
+                file=validated_data.get("file"),
+                audit=validated_data.get("audit"),
             )
             cash_transaction = income.cash_transaction
             payment_type = income.payment_type
 
-            if validated_data.get('currency') == 'uzs':
+            if validated_data.get("currency") == "uzs":
                 cash_transaction.income_balance_uzs += income.price
-                cash_transaction.total_balance_uzs = cash_transaction.income_balance_uzs - cash_transaction.expence_balance_uzs
+                cash_transaction.total_balance_uzs = (
+                    cash_transaction.income_balance_uzs
+                    - cash_transaction.expence_balance_uzs
+                )
                 payment_type.total_uzs += income.price
 
                 if income.counterparty:
@@ -105,26 +135,33 @@ class IncomeCreateSerializer(serializers.ModelSerializer):
                         income.counterparty.debit_uzs -= income.price
                         income.counterparty.total_debit -= income.price
 
-                        income.counterparty.kredit_uzs += income.counterparty.debit_uzs - income.price
+                        income.counterparty.kredit_uzs += (
+                            income.counterparty.debit_uzs - income.price
+                        )
                         income.counterparty.total_kredit += income.price
                     else:
                         income.counterparty.kredit_uzs += income.price
                         income.counterparty.total_kredit += income.price
                     income.counterparty.save()
 
-            elif validated_data.get('currency') == 'usd':
+            elif validated_data.get("currency") == "usd":
                 cash_transaction.income_balance_usd += income.price
-                cash_transaction.total_balance_usd = cash_transaction.income_balance_usd - cash_transaction.expence_balance_usd
+                cash_transaction.total_balance_usd = (
+                    cash_transaction.income_balance_usd
+                    - cash_transaction.expence_balance_usd
+                )
                 payment_type.total_usd += income.price
                 if income.counterparty:
                     if income.counterparty.debit_usd != 0:
-                        income.counterparty.debit_usd -= validated_data.get('price')
+                        income.counterparty.debit_usd -= validated_data.get("price")
                         income.counterparty.total_debit -= income.price
 
-                        income.counterparty.kredit_usd += income.counterparty.debit_usd - validated_data.get('price')
+                        income.counterparty.kredit_usd += (
+                            income.counterparty.debit_usd - validated_data.get("price")
+                        )
                         income.counterparty.total_kredit += income.price
                     else:
-                        income.counterparty.kredit_usd += validated_data.get('price')
+                        income.counterparty.kredit_usd += validated_data.get("price")
                         income.counterparty.total_kredit += income.price
                     income.counterparty.save()
 
@@ -141,46 +178,83 @@ class IncomeUpdateSerializer(serializers.ModelSerializer):
     class Meta:
         model = Income
         fields = [
-            'project_folder', 'project', 'price', 'type_income', 'counterparty',
-            'date', 'comment', 'audit', 'file', 
+            "project_folder",
+            "project",
+            "price",
+            "type_income",
+            "counterparty",
+            "date",
+            "comment",
+            "audit",
+            "file",
         ]
         extra_kwargs = {
-            'price': {'required': False},
+            "price": {"required": False},
         }
 
     def update(self, instance, validated_data):
         old_price = instance.price
-        instance.project_folder = validated_data.get('project_folder', instance.project_folder)
-        instance.project = validated_data.get('project', instance.project)
-        instance.price = validated_data.get('price', instance.price)
-        instance.type_income = validated_data.get('type_income', instance.type_income)
-        instance.counterparty = validated_data.get('counterparty', instance.counterparty)
-        instance.date = validated_data.get('date', instance.date)
-        instance.comment = validated_data.get('comment', instance.comment)
-        instance.audit = validated_data.get('audit', instance.audit)
-        instance.file = validated_data.get('file', instance.file)
-        if validated_data.get('price'):
+        instance.project_folder = validated_data.get(
+            "project_folder", instance.project_folder
+        )
+        instance.project = validated_data.get("project", instance.project)
+        instance.price = validated_data.get("price", instance.price)
+        instance.type_income = validated_data.get("type_income", instance.type_income)
+        instance.counterparty = validated_data.get(
+            "counterparty", instance.counterparty
+        )
+        instance.date = validated_data.get("date", instance.date)
+        instance.comment = validated_data.get("comment", instance.comment)
+        instance.audit = validated_data.get("audit", instance.audit)
+        instance.file = validated_data.get("file", instance.file)
+        if validated_data.get("price"):
             cash_transaction = instance.cash_transaction
             payment_type = instance.payment_type
-            if old_price > validated_data.get('price'):
-                if instance.currency == 'uzs':
-                    cash_transaction.income_balance_uzs -= (old_price - validated_data.get('price'))
-                    cash_transaction.total_balance_uzs -= (old_price - validated_data.get('price'))
-                    payment_type.total_uzs -= (old_price - validated_data.get('price')) if payment_type.total_uzs > (old_price - validated_data.get('price')) else 0
+            if old_price > validated_data.get("price"):
+                if instance.currency == "uzs":
+                    cash_transaction.income_balance_uzs -= (
+                        old_price - validated_data.get("price")
+                    )
+                    cash_transaction.total_balance_uzs -= (
+                        old_price - validated_data.get("price")
+                    )
+                    payment_type.total_uzs -= (
+                        (old_price - validated_data.get("price"))
+                        if payment_type.total_uzs
+                        > (old_price - validated_data.get("price"))
+                        else 0
+                    )
                 else:
-                    cash_transaction.income_balance_usd -= (old_price - validated_data.get('price'))
-                    cash_transaction.total_balance_usd -= (old_price - validated_data.get('price'))
-                    payment_type.total_usd -= (old_price - validated_data.get('price')) if payment_type.total_usd > (old_price - validated_data.get('price')) else 0
+                    cash_transaction.income_balance_usd -= (
+                        old_price - validated_data.get("price")
+                    )
+                    cash_transaction.total_balance_usd -= (
+                        old_price - validated_data.get("price")
+                    )
+                    payment_type.total_usd -= (
+                        (old_price - validated_data.get("price"))
+                        if payment_type.total_usd
+                        > (old_price - validated_data.get("price"))
+                        else 0
+                    )
 
-            elif old_price < validated_data.get('price'):
-                if instance.currency == 'uzs':
-                    cash_transaction.income_balance_uzs += (old_price - validated_data.get('price'))
-                    cash_transaction.total_balance_uzs += (old_price - validated_data.get('price'))
-                    payment_type.total_uzs += (old_price - validated_data.get('price'))
+            elif old_price < validated_data.get("price"):
+                if instance.currency == "uzs":
+                    cash_transaction.income_balance_uzs += (
+                        old_price - validated_data.get("price")
+                    )
+                    cash_transaction.total_balance_uzs += (
+                        old_price - validated_data.get("price")
+                    )
+                    payment_type.total_uzs += old_price - validated_data.get("price")
                 else:
-                    cash_transaction.income_balance_usd += (old_price - validated_data.get('price'))
-                    cash_transaction.total_balance_usd += (old_price - validated_data.get('price'))
-                    payment_type.total_usd += (old_price - validated_data.get('price'))
+                    cash_transaction.income_balance_usd += (
+                        old_price - validated_data.get("price")
+                    )
+                    cash_transaction.total_balance_usd += (
+                        old_price - validated_data.get("price")
+                    )
+                    payment_type.total_usd += old_price - validated_data.get("price")
         cash_transaction.save()
         payment_type.save()
         instance.save()