Adjustments

Signed-off-by: Tushar Goel <tushar.goel.dav@gmail.com>

Author: TG1999

vulnerabilities/importers/gitlab.py

@@ -31,6 +31,7 @@
 from packageurl import PackageURL
 from univers.version_range import RANGE_CLASS_BY_SCHEMES
 from univers.version_range import VersionRange
+from univers.versions import Version
 
 from vulnerabilities.importer import AdvisoryData
 from vulnerabilities.importer import AffectedPackage
@@ -132,15 +133,15 @@ def get_purl(package_slug):
 
 
 def extract_affected_packages(
-    affected_version_range: VersionRange, fixed_versions: List[str], purl: PackageURL
+    affected_version_range: VersionRange, fixed_versions: List[str], purl: PackageURL, version_class: Version
 ) -> Iterable[AffectedPackage]:
     """
     Yield a list of AffectedPackage objects
     """
     for fixed_version in fixed_versions or []:
         yield AffectedPackage(
             package=purl,
-            fixed_version=affected_version_range.version_class(fixed_version),
+            fixed_version=version_class(fixed_version),
             affected_version_range=affected_version_range,
         )
 
@@ -163,6 +164,7 @@ def parse_yaml_file(file):
     vrc: VersionRange = RANGE_CLASS_BY_SCHEMES[purl.type]
     affected_range = yaml_file.get("affected_range")
     affected_version_range = None
+    version_class = vrc.version_class
     try:
         affected_version_range = vrc.from_native(affected_range) if affected_range else None
     except Exception:
@@ -173,7 +175,7 @@ def parse_yaml_file(file):
         affected_version_range = None
 
     affected_packages = list(
-        extract_affected_packages(affected_version_range, fixed_versions, purl)
+        extract_affected_packages(affected_version_range, fixed_versions, purl, version_class)
     )
 
     return AdvisoryData(