feat: add chinese sca

dongtai_common/migrations/0004_auto_20230731_1556.py

@@ -0,0 +1,22 @@
+# Generated by Django 3.2.20 on 2023-07-31 15:56
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+    dependencies = [
+        ("dongtai_common", "0003_auto_20230728_1444"),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name="iastassetvulv2",
+            name="vul_detail_zh",
+            field=models.TextField(blank=True),
+        ),
+        migrations.AddField(
+            model_name="iastassetvulv2",
+            name="vul_name_zh",
+            field=models.CharField(blank=True, max_length=255),
+        ),
+    ]

dongtai_common/models/asset_vul_v2.py

@@ -6,7 +6,9 @@
 
 class IastAssetVulV2(models.Model):
     vul_name = models.CharField(max_length=255, blank=True)
+    vul_name_zh = models.CharField(max_length=255, blank=True)
     vul_detail = models.TextField()
+    vul_detail_zh = models.TextField(blank=True)
     # 漏洞类型等级
     level = models.IntegerField(
         choices=AssetRiskLevel.choices,

dongtai_common/serializers/assetvulv2.py

@@ -1,12 +1,25 @@
 from rest_framework import serializers
 
 from dongtai_common.models.asset_vul_v2 import IastAssetVulV2
+from dongtai_web.dongtai_sca.scan.utils import get_sca_language_profile
 
 
 class PackageVulSerializer(serializers.ModelSerializer):
+    vul_name = serializers.SerializerMethodField()
+    vul_detail = serializers.SerializerMethodField()
     level_name = serializers.CharField(source="get_level_display")
     level_id = serializers.IntegerField(source="level")
 
     class Meta:
         model = IastAssetVulV2
-        fields = "__all__"
+        exclude = ["vul_name_zh", "vul_detail_zh"]
+
+    def get_vul_name(self, obj: IastAssetVulV2) -> str:
+        if get_sca_language_profile()["language"] == "zh":
+            return obj.vul_name_zh or obj.vul_name
+        return obj.vul_name
+
+    def get_vul_detail(self, obj: IastAssetVulV2) -> str:
+        if get_sca_language_profile()["language"] == "zh":
+            return obj.vul_detail_zh or obj.vul_detail
+        return obj.vul_detail

dongtai_conf/celery.py

@@ -94,7 +94,11 @@
         "exchange": "dongtai-replay-vul-scan",
         "routing_key": "dongtai-replay-vul-scan",
     },
-    "dongtai_web.dongtai_sca.scan.utils.update_one_sca": {
+    "dongtai_web.dongtai_sca.scan.tasks.update_one_sca": {
+        "exchange": "dongtai-sca-task",
+        "routing_key": "dongtai-sca-task",
+    },
+    "dongtai_web.dongtai_sca.scan.tasks.new_update_one_sca": {
         "exchange": "dongtai-sca-task",
         "routing_key": "dongtai-sca-task",
     },

dongtai_protocol/report/handler/sca_handler.py

@@ -8,7 +8,7 @@
 from dongtai_common.utils import const
 from dongtai_protocol.report.handler.report_handler_interface import IReportHandler
 from dongtai_protocol.report.report_handler_factory import ReportHandler
-from dongtai_web.dongtai_sca.scan.utils import new_update_one_sca, update_one_sca
+from dongtai_web.dongtai_sca.scan.tasks import new_update_one_sca, update_one_sca
 
 logger = logging.getLogger("dongtai.openapi")
 

dongtai_web/dongtai_sca/common/dataclass.py

@@ -9,11 +9,19 @@
 # Be careful with potentially nullable types when using them temporarily.
 
 
+@dataclass_json
+@dataclass
+class I18nStr:
+    en: str = ""
+    zh: str | None = None
+
+
 @dataclass_json
 @dataclass
 class Reference:
     type: str = ""
     url: str = ""
+    language: str = ""
 
 
 @dataclass_json
@@ -29,8 +37,8 @@ class VulInfo:
     vul_id: str = ""
     cvss_v3: str = ""
     cwe: tuple[str, ...] = ()
-    title: str = ""
-    description: str = ""
+    title: I18nStr = field(default_factory=I18nStr)
+    description: I18nStr = field(default_factory=I18nStr)
     references: tuple[Reference, ...] = ()
     severity: str = ""
     published_time: datetime | None = field(