perf: change i18n

apps/accounts/serializers/automations/check_accounts.py

@@ -1,9 +1,12 @@
 # -*- coding: utf-8 -*-
 #
+from django.utils.translation import gettext_lazy as _
 from rest_framework import serializers
 
 from accounts.const import AutomationTypes
 from accounts.models import AccountCheckAutomation, AccountRisk, RiskChoice
+from assets.models import Asset
+from common.serializers.fields import ObjectRelatedField, LabeledChoiceField
 from common.utils import get_logger
 from .base import BaseAutomationSerializer
 
@@ -18,9 +21,15 @@ __all__ = [
 
 
 class AccountRiskSerializer(serializers.ModelSerializer):
+    asset = ObjectRelatedField(queryset=Asset.objects.all(), required=False,label=_("Asset"))
+    risk = LabeledChoiceField(choices=RiskChoice.choices, required=False, read_only=True, label=_("Risk"))
+
     class Meta:
         model = AccountRisk
-        fields = '__all__'
+        fields = [
+            'id', 'asset', 'username', 'risk', 'confirmed',
+            'date_created'
+        ]
 
 
 class RiskSummarySerializer(serializers.Serializer):

apps/assets/api/tree.py

@@ -146,7 +146,8 @@ class NodeChildrenAsTreeApi(SerializeToTreeNodeMixin, NodeChildrenApi):
 
     def list(self, request, *args, **kwargs):
         nodes = self.filter_queryset(self.get_queryset()).order_by('value')
-        nodes = self.serialize_nodes(nodes, with_asset_amount=True)
+        with_asset_amount = request.query_params.get('asset_amount', '1') == '1'
+        nodes = self.serialize_nodes(nodes, with_asset_amount=with_asset_amount)
         assets = self.filter_queryset_for_assets(self.get_queryset_for_assets())
         node_key = self.instance.key if self.instance else None
         assets = self.serialize_assets(assets, node_key=node_key)