Skip to content

Commit 27b64d3

Browse files
committed
calculating CVSS score
Signed-off-by: Ziad <ziadhany2016@gmail.com>
1 parent d218443 commit 27b64d3

1 file changed

Lines changed: 8 additions & 2 deletions

File tree

vulnerabilities/models.py

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -278,7 +278,6 @@ def get_absolute_url(self):
278278

279279

280280
class PackageRelatedVulnerability(models.Model):
281-
282281
# TODO: Fix related_name
283282
package = models.ForeignKey(
284283
Package,
@@ -351,7 +350,6 @@ def update_or_create(self):
351350

352351

353352
class VulnerabilitySeverity(models.Model):
354-
355353
reference = models.ForeignKey(VulnerabilityReference, on_delete=models.CASCADE)
356354

357355
scoring_system_choices = tuple(
@@ -371,6 +369,14 @@ class VulnerabilitySeverity(models.Model):
371369

372370
value = models.CharField(max_length=50, help_text="Example: 9.0, Important, High")
373371

372+
@property
373+
def scoring_element(self):
374+
try:
375+
score = SCORING_SYSTEMS[self.scoring_system].as_score(self.value)
376+
return score
377+
except NotImplementedError:
378+
return
379+
374380
class Meta:
375381
unique_together = ["reference", "scoring_system", "value"]
376382
ordering = ["reference", "scoring_system", "value"]

0 commit comments

Comments
 (0)