Skip to content

Commit 2985d25

Browse files
committed
IMPORTER_REGISTRY -> IMPORTERS_REGISTRY
Same for improvers IMPROVER_REGISTRY -> IMPROVERS_REGISTRY Signed-off-by: Hritik Vijay <hritikxx8@gmail.com>
1 parent 76730ca commit 2985d25

5 files changed

Lines changed: 14 additions & 14 deletions

File tree

vulnerabilities/importers/__init__.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -21,6 +21,6 @@
2121
# Visit https://github.com/nexB/vulnerablecode/ for support and download.
2222
from vulnerabilities.importers import nginx
2323

24-
IMPORTERS = [nginx.NginxImporter]
24+
IMPORTERS_REGISTRY = [nginx.NginxImporter]
2525

26-
IMPORTER_REGISTRY = {x.qualified_name: x for x in IMPORTERS}
26+
IMPORTERS_REGISTRY = {x.qualified_name: x for x in IMPORTERS_REGISTRY}
Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
from vulnerabilities.improvers import default
22
from vulnerabilities import importers
33

4-
IMPROVERS = [default.DefaultImprover, importers.nginx.NginxBasicImprover]
4+
IMPROVERS_REGISTRY = [default.DefaultImprover, importers.nginx.NginxBasicImprover]
55

6-
IMPROVER_REGISTRY = {x.qualified_name: x for x in IMPROVERS}
6+
IMPROVERS_REGISTRY = {x.qualified_name: x for x in IMPROVERS_REGISTRY}

vulnerabilities/management/commands/import.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -26,7 +26,7 @@
2626
from django.core.management.base import BaseCommand
2727
from django.core.management.base import CommandError
2828

29-
from vulnerabilities.importers import IMPORTER_REGISTRY
29+
from vulnerabilities.importers import IMPORTERS_REGISTRY
3030
from vulnerabilities.import_runner import ImportRunner
3131

3232

@@ -49,7 +49,7 @@ def handle(self, *args, **options):
4949
return
5050

5151
if options["all"]:
52-
self.import_data(IMPORTER_REGISTRY.values())
52+
self.import_data(IMPORTERS_REGISTRY.values())
5353
return
5454

5555
sources = options["sources"]
@@ -59,7 +59,7 @@ def handle(self, *args, **options):
5959
self.import_data(validate_importers(sources))
6060

6161
def list_sources(self):
62-
importers = list(IMPORTER_REGISTRY)
62+
importers = list(IMPORTERS_REGISTRY)
6363
self.stdout.write("Vulnerability data can be imported from the following importers:")
6464
self.stdout.write("\n".join(importers))
6565

@@ -97,7 +97,7 @@ def validate_importers(sources):
9797
unknown_sources = []
9898
for source in sources:
9999
try:
100-
importers.append(IMPORTER_REGISTRY[source])
100+
importers.append(IMPORTERS_REGISTRY[source])
101101
except KeyError:
102102
unknown_sources.append(source)
103103
if unknown_sources:

vulnerabilities/management/commands/improve.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -27,7 +27,7 @@
2727
from django.core.management.base import CommandError
2828

2929
from vulnerabilities.import_runner import ImportRunner
30-
from vulnerabilities.improvers import IMPROVER_REGISTRY
30+
from vulnerabilities.improvers import IMPROVERS_REGISTRY
3131
from vulnerabilities.improve_runner import ImproveRunner
3232

3333

@@ -51,7 +51,7 @@ def handle(self, *args, **options):
5151
return
5252

5353
if options["all"]:
54-
self.improve_data(IMPROVER_REGISTRY.values())
54+
self.improve_data(IMPROVERS_REGISTRY.values())
5555
return
5656

5757
sources = options["sources"]
@@ -61,7 +61,7 @@ def handle(self, *args, **options):
6161
self.improve_data(validate_improvers(sources))
6262

6363
def list_sources(self):
64-
improvers = list(IMPROVER_REGISTRY)
64+
improvers = list(IMPROVERS_REGISTRY)
6565
self.stdout.write("Vulnerability data can be processed by these available improvers:\n")
6666
self.stdout.write("\n".join(improvers))
6767

@@ -95,7 +95,7 @@ def validate_improvers(sources):
9595
unknown_sources = []
9696
for source in sources:
9797
try:
98-
improvers.append(IMPROVER_REGISTRY[source])
98+
improvers.append(IMPROVERS_REGISTRY[source])
9999
except KeyError:
100100
unknown_sources.append(source)
101101
if unknown_sources:

vulnerabilities/tests/test_upstream.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@
55

66
from vulnerabilities import importers
77
from vulnerabilities.importer import Advisory
8-
from vulnerabilities.importer_yielder import IMPORTER_REGISTRY
8+
from vulnerabilities.importer_yielder import IMPORTERS_REGISTRY
99

1010
MAX_ADVISORIES = 1
1111

@@ -18,7 +18,7 @@ class MaxAdvisoriesCreatedInterrupt(BaseException):
1818
@pytest.mark.webtest
1919
@pytest.mark.parametrize(
2020
("data_source", "config"),
21-
((data["data_source"], data["data_source_cfg"]) for data in IMPORTER_REGISTRY),
21+
((data["data_source"], data["data_source_cfg"]) for data in IMPORTERS_REGISTRY),
2222
)
2323
def test_updated_advisories(data_source, config):
2424
if not data_source == "GitHubAPIImporter":

0 commit comments

Comments
 (0)