fixed code style

Signed-off-by: Tushar912 <tushar.912u@gmail.com>

Author: tushar912

vulnerabilities/importer_yielder.py

@@ -227,7 +227,7 @@
         'data_source_cfg': {},
     },
     {
-    'name': 'elixir_security' ,
+    'name': 'elixir_security',
     'license': '',
     'last_run': None,
     'data_source': 'ElixirSecurityDataSource',

vulnerabilities/importers/__init__.py

@@ -43,4 +43,4 @@
 from vulnerabilities.importers.kaybee import KaybeeDataSource
 from vulnerabilities.importers.nginx import NginxDataSource
 from vulnerabilities.importers.postgresql import PostgreSQLDataSource
-from vulnerabilities.importers.elixir_security import ElixirSecurityDataSource
+from vulnerabilities.importers.elixir_security import ElixirSecurityDataSource

vulnerabilities/importers/elixir_security.py

@@ -34,16 +34,16 @@
 from vulnerabilities.data_source import Advisory
 from vulnerabilities.data_source import Reference
 
+
 class ElixirSecurityDataSource(GitDataSource):
     def __enter__(self):
         super(ElixirSecurityDataSource, self).__enter__()
 
         if not getattr(self, "_added_files", None):
             self._added_files, self._updated_files = self.file_changes(
-                recursive=True, file_ext="yml",subdir="./packages"
+                recursive=True, file_ext="yml", subdir="./packages"
             )
 
-        
     def updated_advisories(self) -> Set[Advisory]:
         files = self._updated_files
         advisories = []
@@ -62,61 +62,67 @@ def added_advisories(self) -> Set[Advisory]:
                 advisories.append(processed_data)
         return self.batch_advisories(advisories)
 
-    @staticmethod    
+    @staticmethod
     def generate_all_versions_list(pkg_name):
         resp = requests.get(f"https://hex.pm/api/packages/{pkg_name}")
         resp = resp.content
         json_resp = json.loads(resp)
         versions_list = []
-        for release in json_resp['releases']:
-            versions_list.append(release['version'])
+        for release in json_resp["releases"]:
+            versions_list.append(release["version"])
         return versions_list
 
-    
-    def get_pkg_from_range(self,versions_list,pkg_name):
+    def get_pkg_from_range(self, versions_list, pkg_name):
         pkg_versions = []
         all_versions_list = self.generate_all_versions_list(pkg_name)
         if versions_list is None:
             return
         for version in versions_list:
-            if re.match('^>=',version):
+            if re.match("^>=", version):
                 index = all_versions_list.index(version[3:])
-                pkg_versions = pkg_versions + all_versions_list[0:index+1]
-            elif re.match('^>',version):
+                pkg_versions = pkg_versions + all_versions_list[0: index + 1]
+            elif re.match("^>", version):
                 index = all_versions_list.index(version[2:])
                 pkg_versions = pkg_versions + all_versions_list[0:index]
-            elif re.match('^<',version):
+            elif re.match("^<", version):
                 index = all_versions_list.index(version[2:])
-                pkg_versions = pkg_versions + all_versions_list[index+1:-1]
-            else :
+                pkg_versions = pkg_versions + all_versions_list[index + 1: -1]
+            else:
                 pkg_versions.append(version[3:])
         return pkg_versions
-                
-    
-    def process_file(self,path):
+
+    def process_file(self, path):
         with open(path) as f:
             yaml_file = yaml.safe_load(f)
-            pkg_name = yaml_file['package']
-            safe_pkg_versions =[]
-            if yaml_file.get('unaffected_versions'):
-                safe_pkg_versions = self.get_pkg_from_range(yaml_file['patched_versions']+yaml_file['unaffected_versions'],pkg_name)
+            pkg_name = yaml_file["package"]
+            safe_pkg_versions = []
+            if yaml_file.get("unaffected_versions"):
+                safe_pkg_versions = self.get_pkg_from_range(
+                    yaml_file["patched_versions"] + yaml_file["unaffected_versions"],
+                    pkg_name,
+                )
             else:
-                safe_pkg_versions = self.get_pkg_from_range(yaml_file['patched_versions'],pkg_name)
-            cve_id = yaml_file['cve']
+                safe_pkg_versions = self.get_pkg_from_range(
+                    yaml_file["patched_versions"], pkg_name
+                )
+            cve_id = yaml_file["cve"]
             safe_purls = []
             if safe_pkg_versions is not None:
-                safe_purls ={ PackageURL(name=pkg_name,
-                type='hex',
-                version=version)
-                for version in safe_pkg_versions}
-            
-            vuln_reference = [Reference(
-                url=yaml_file['link'],
-            )]
+                safe_purls = {
+                    PackageURL(name=pkg_name, type="hex", version=version)
+                    for version in safe_pkg_versions
+                }
+
+            vuln_reference = [
+                Reference(
+                    url=yaml_file["link"],
+                )
+            ]
 
             return Advisory(
-                summary=yaml_file['description'],
+                summary=yaml_file["description"],
                 impacted_package_urls=[],
                 resolved_package_urls=safe_purls,
                 cve_id=cve_id,
-                vuln_references=vuln_reference)
+                vuln_references=vuln_reference,
+            )

vulnerabilities/tests/test_elixir_security.py

@@ -34,41 +34,52 @@
 
 
 class TestElixirSecurityDataSource(TestCase):
-
     @classmethod
     def setUpClass(cls):
         data_source_cfg = {
-            'repository_url': 'https://test.net', }
+            "repository_url": "https://test.net",
+        }
         cls.data_src = ElixirSecurityDataSource(1, config=data_source_cfg)
 
     def test_generate_all_versions_list(self):
-        package = 'coherence'
+        package = "coherence"
         actual_list = self.data_src.generate_all_versions_list(package)
-        expected_list = ['0.5.2', '0.5.1', '0.5.0', '0.4.0', '0.3.1', '0.3.0', '0.2.0', '0.1.3', '0.1.2', '0.1.1', '0.1.0']
+        expected_list = [
+            "0.5.2",
+            "0.5.1",
+            "0.5.0",
+            "0.4.0",
+            "0.3.1",
+            "0.3.0",
+            "0.2.0",
+            "0.1.3",
+            "0.1.2",
+            "0.1.1",
+            "0.1.0",
+        ]
         assert actual_list == expected_list
 
     def test_process_file(self):
 
         path = os.path.join(BASE_DIR, "test_data/elixir_security/test_file.yml")
         expected_data = Advisory(
-            summary=('The Coherence library has "Mass Assignment"-like vulnerabilities.\n'),
-            
+            summary=(
+                'The Coherence library has "Mass Assignment"-like vulnerabilities.\n'
+            ),
             impacted_package_urls=[],
-
             resolved_package_urls={
                 PackageURL(
-                    type='hex',
-                    name='coherence',
-                    version='0.5.2',
-                    ),
-                },
+                    type="hex",
+                    name="coherence",
+                    version="0.5.2",
+                ),
+            },
             vuln_references=[
-                Reference(
-                    url="https://github.com/smpallen99/coherence/issues/270"
-                )
+                Reference(url="https://github.com/smpallen99/coherence/issues/270")
             ],
-            cve_id='2018-20301')
+            cve_id="2018-20301",
+        )
 
         found_data = self.data_src.process_file(path)
 
-        assert expected_data == found_data
+        assert expected_data == found_data