Skip to content

Commit 5e98ecd

Browse files
committed
Add avid in api
Signed-off-by: Tushar Goel <tushar.goel.dav@gmail.com>
1 parent 3a9496a commit 5e98ecd

1 file changed

Lines changed: 6 additions & 3 deletions

File tree

vulnerabilities/api_v3.py

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -227,8 +227,7 @@ def get_affected_by_vulnerabilities(self, package):
227227
result = []
228228

229229
for adv in advisories:
230-
fixed = impact_map.get(adv["avid"])
231-
adv.pop("avid", None)
230+
fixed = impact_map.get(adv["avid"]) or []
232231
resource_url = None
233232

234233
if request := self.context.get("request", None):
@@ -293,6 +292,7 @@ def get_affected_by_vulnerabilities(self, package):
293292

294293
result.append(
295294
{
295+
"avid": advisory.avid,
296296
"advisory_id": advisory.advisory_id.split("/")[-1],
297297
"advisory_uid": advisory.avid,
298298
"aliases": [alias.alias for alias in advisory.aliases.all()],
@@ -338,6 +338,7 @@ def get_fixing_vulnerabilities(self, package):
338338
{
339339
"advisory_id": advisory["advisory_id"],
340340
"resource_url": resource_url,
341+
"avid": advisory["avid"],
341342
}
342343
)
343344
if results:
@@ -362,6 +363,7 @@ def get_fixing_vulnerabilities(self, package):
362363
"advisory_id": advisory.advisory_id.split("/")[-1],
363364
"advisory_uid": advisory.avid,
364365
"resource_url": resource_url,
366+
"avid": advisory.avid,
365367
}
366368
)
367369
return results
@@ -393,6 +395,7 @@ def return_fixing_advisories_data(self, advisories):
393395
"advisory_id": advisory.identifier,
394396
"advisory_uid": advisory.advisory.avid,
395397
"resource_url": resource_url,
398+
"avid": advisory.advisory.avid,
396399
}
397400
)
398401

@@ -420,7 +423,7 @@ def return_advisories_data(self, package, advisories_qs, advisories):
420423
location=advisory.advisory.get_absolute_url()
421424
)
422425
impact = impact_by_avid.get(advisory.advisory.avid)
423-
if not impact:
426+
if impact:
424427
fixed_by_packages = list(set([pkg.purl for pkg in impact.fixed_by_packages.all()]))
425428

426429
result.append(

0 commit comments

Comments
 (0)