Skip to content

Commit 22fb93d

Browse files
committed
Update retiredotnet to use the new AdvisoryDataV2
Drop duplicates in affected and fixed version range Signed-off-by: ziad hany <ziadhany2016@gmail.com>
1 parent 74ffe5e commit 22fb93d

2 files changed

Lines changed: 13 additions & 13 deletions

File tree

vulnerabilities/pipelines/v2_importers/retiredotnet_importer.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@
1616
from packageurl import PackageURL
1717
from univers.version_range import NugetVersionRange
1818

19-
from vulnerabilities.importer import AdvisoryData
19+
from vulnerabilities.importer import AdvisoryDataV2
2020
from vulnerabilities.importer import AffectedPackageV2
2121
from vulnerabilities.importer import ReferenceV2
2222
from vulnerabilities.pipelines import VulnerableCodeBaseImporterPipelineV2
@@ -66,7 +66,7 @@ def collect_advisories(self):
6666
# group by package name `id`
6767
# { pkg_id: {'affected_versions': [], 'fixed': []} }
6868
grouped_packages = defaultdict(
69-
lambda: {"affected_versions": [], "fixed_versions": []}
69+
lambda: {"affected_versions": set(), "fixed_versions": set()}
7070
)
7171
for pkg in json_doc.get("packages") or []:
7272
name = pkg.get("id")
@@ -75,11 +75,11 @@ def collect_advisories(self):
7575

7676
affected_version = pkg.get("affected")
7777
if affected_version:
78-
grouped_packages[name]["affected_versions"].append(affected_version)
78+
grouped_packages[name]["affected_versions"].add(affected_version)
7979

8080
fixed_version = pkg.get("fix")
8181
if fixed_version:
82-
grouped_packages[name]["fixed_versions"].append(fixed_version)
82+
grouped_packages[name]["fixed_versions"].add(fixed_version)
8383

8484
for pkg in grouped_packages:
8585
affected_version_range = None
@@ -109,12 +109,12 @@ def collect_advisories(self):
109109
)
110110
]
111111

112-
yield AdvisoryData(
112+
yield AdvisoryDataV2(
113113
advisory_id=advisory_id,
114114
aliases=[aliases] if aliases else [],
115115
summary=description,
116116
affected_packages=affected_packages,
117-
references_v2=vuln_reference,
117+
references=vuln_reference,
118118
url=advisory_url,
119119
)
120120

vulnerabilities/tests/test_data/retiredotnet_v2/expected_file.json

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@
1616
"subpath": ""
1717
},
1818
"affected_version_range": "vers:nuget/2.1.0|2.1.1|2.2.0",
19-
"fixed_version_range": "vers:nuget/2.1.7|2.1.7|2.2.1",
19+
"fixed_version_range": "vers:nuget/2.1.7|2.2.1",
2020
"introduced_by_commit_patches": [],
2121
"fixed_by_commit_patches": []
2222
},
@@ -30,7 +30,7 @@
3030
"subpath": ""
3131
},
3232
"affected_version_range": "vers:nuget/2.1.0|2.1.1|2.1.2|2.1.3",
33-
"fixed_version_range": "vers:nuget/2.1.7|2.1.7|2.1.7|2.1.7",
33+
"fixed_version_range": "vers:nuget/2.1.7",
3434
"introduced_by_commit_patches": [],
3535
"fixed_by_commit_patches": []
3636
},
@@ -44,7 +44,7 @@
4444
"subpath": ""
4545
},
4646
"affected_version_range": "vers:nuget/4.5.0|4.5.1|4.5.2",
47-
"fixed_version_range": "vers:nuget/4.5.3|4.5.3|4.5.3",
47+
"fixed_version_range": "vers:nuget/4.5.3",
4848
"introduced_by_commit_patches": [],
4949
"fixed_by_commit_patches": []
5050
},
@@ -58,7 +58,7 @@
5858
"subpath": ""
5959
},
6060
"affected_version_range": "vers:nuget/2.1.0|2.1.1|2.1.2|2.1.3|2.1.4|2.1.5|2.1.6|2.2.0",
61-
"fixed_version_range": "vers:nuget/2.1.7|2.1.7|2.1.7|2.1.7|2.1.7|2.1.7|2.1.7|2.2.1",
61+
"fixed_version_range": "vers:nuget/2.1.7|2.2.1",
6262
"introduced_by_commit_patches": [],
6363
"fixed_by_commit_patches": []
6464
},
@@ -72,7 +72,7 @@
7272
"subpath": ""
7373
},
7474
"affected_version_range": "vers:nuget/2.1.0|2.1.1|2.1.2|2.1.3|2.1.4|2.1.5|2.1.6|2.2.0",
75-
"fixed_version_range": "vers:nuget/2.1.7|2.1.7|2.1.7|2.1.7|2.1.7|2.1.7|2.1.7|2.2.1",
75+
"fixed_version_range": "vers:nuget/2.1.7|2.2.1",
7676
"introduced_by_commit_patches": [],
7777
"fixed_by_commit_patches": []
7878
},
@@ -86,12 +86,12 @@
8686
"subpath": ""
8787
},
8888
"affected_version_range": "vers:nuget/2.1.0|2.1.1|2.1.2|2.1.3|2.1.4|2.1.5|2.1.6|2.2.0",
89-
"fixed_version_range": "vers:nuget/2.1.7|2.1.7|2.1.7|2.1.7|2.1.7|2.1.7|2.1.7|2.2.1",
89+
"fixed_version_range": "vers:nuget/2.1.7|2.2.1",
9090
"introduced_by_commit_patches": [],
9191
"fixed_by_commit_patches": []
9292
}
9393
],
94-
"references_v2": [
94+
"references": [
9595
{
9696
"reference_id": "",
9797
"reference_type": "",

0 commit comments

Comments
 (0)