Skip to content

Commit dd77299

Browse files
committed
feat: add method to get CVSS object from the scoring system
Signed-off-by: Keshav Priyadarshi <git@keshav.space>
1 parent df926ff commit dd77299

1 file changed

Lines changed: 10 additions & 4 deletions

File tree

vulnerabilities/severity_systems.py

Lines changed: 10 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -82,9 +82,12 @@ def compute(self, scoring_elements: str) -> str:
8282
"""
8383
return str(CVSS3(vector=scoring_elements).base_score)
8484

85-
def get(self, scoring_elements: str) -> dict:
85+
def get_cvss(self, scoring_elements: str) -> CVSS3:
8686
scoring_elements = scoring_elements.strip()
87-
return CVSS3(vector=scoring_elements).as_json()
87+
return CVSS3(vector=scoring_elements)
88+
89+
def get(self, scoring_elements: str) -> dict:
90+
return self.get_cvss(scoring_elements=scoring_elements).as_json()
8891

8992

9093
@dataclasses.dataclass(order=True)
@@ -98,9 +101,12 @@ def compute(self, scoring_elements: str) -> str:
98101
"""
99102
return str(CVSS4(vector=scoring_elements).base_score)
100103

101-
def get(self, scoring_elements: str) -> dict:
104+
def get_cvss(self, scoring_elements: str) -> CVSS4:
102105
scoring_elements = scoring_elements.strip()
103-
return CVSS4(vector=scoring_elements).as_json()
106+
return CVSS4(vector=scoring_elements)
107+
108+
def get(self, scoring_elements: str) -> dict:
109+
return self.get_cvss(scoring_elements=scoring_elements).as_json()
104110

105111

106112
CVSSV3 = Cvssv3ScoringSystem(

0 commit comments

Comments
 (0)