diff --git a/core/apps/finance/serializers/income.py b/core/apps/finance/serializers/income.py index 7ea3861..14fbe16 100644 --- a/core/apps/finance/serializers/income.py +++ b/core/apps/finance/serializers/income.py @@ -1,137 +1,174 @@ 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: if income.counterparty.debit_uzs != 0: - income.counterparty.debit_uzs -= income.price + 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() cash_transaction.save() payment_type.save() return income - + class IncomeDeleteSerializer(serializers.Serializer): comment = serializers.CharField() @@ -141,47 +178,84 @@ 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 + 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 + ) 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() - return instance \ No newline at end of file + return instance