Skip to content

Commit 64ac12b

Browse files
committed
Support export pipeline in pipeline schedules
Signed-off-by: Keshav Priyadarshi <git@keshav.space>
1 parent 6f6f1f9 commit 64ac12b

File tree

1 file changed

+5
-4
lines changed

1 file changed

+5
-4
lines changed

vulnerabilities/models.py

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -2341,13 +2341,14 @@ def save(self, *args, **kwargs):
23412341
@property
23422342
def pipeline_class(self):
23432343
"""Return the pipeline class."""
2344+
23442345
from vulnerabilities.importers import IMPORTERS_REGISTRY
23452346
from vulnerabilities.improvers import IMPROVERS_REGISTRY
2347+
from vulnerabilities.pipelines.exporters import EXPORTERS_REGISTRY
2348+
2349+
pipeline_registry = IMPORTERS_REGISTRY | IMPROVERS_REGISTRY | EXPORTERS_REGISTRY
23462350

2347-
if self.pipeline_id in IMPROVERS_REGISTRY:
2348-
return IMPROVERS_REGISTRY.get(self.pipeline_id)
2349-
if self.pipeline_id in IMPORTERS_REGISTRY:
2350-
return IMPORTERS_REGISTRY.get(self.pipeline_id)
2351+
return pipeline_registry[self.pipeline_id]
23512352

23522353
@property
23532354
def description(self):

0 commit comments

Comments
 (0)