Merge pull request #715 from Bidaya0/bugfix/fix-vul-assetname-v2

Bugfix/fix vul assetname v2

Author: Bidaya0

dongtai_web/dongtai_sca/utils.py

@@ -317,14 +317,15 @@ def _add_vul_data(asset, asset_package, cve_relation):
 
 
 def _add_asset_vul_relation(asset_vul):
-    vul_assets = Asset.objects.filter(package_name=asset_vul.aql, version=asset_vul.package_version,
+    vul_assets = Asset.objects.filter(version=asset_vul.package_version,
                                       signature_value=asset_vul.package_hash).values('id').all()
     asset_vul_relations = []
     if vul_assets:
         for asset_vl in vul_assets:
-            relation_exist = IastVulAssetRelation.objects.filter(asset_vul=asset_vul, asset_id=asset_vl['id']).exists()
+            relation_exist = IastVulAssetRelation.objects.filter(asset_vul_id=asset_vul.id,
+                                                                 asset_id=asset_vl['id']).first()
             if not relation_exist:
-                asset_vul_relations.append(IastVulAssetRelation(asset_vul=asset_vul, asset_id=asset_vl['id'],
+                asset_vul_relations.append(IastVulAssetRelation(asset_vul_id=asset_vul.id, asset_id=asset_vl['id'],
                                                                 create_time=int(time.time()), status_id=1))
 
     if asset_vul_relations: