Skip to content

Commit 15a8492

Browse files
committed
Support export pipeline in pipeline schedules
Signed-off-by: Keshav Priyadarshi <git@keshav.space>
1 parent c60eeaf commit 15a8492

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
@@ -2344,13 +2344,14 @@ def save(self, *args, **kwargs):
23442344
@property
23452345
def pipeline_class(self):
23462346
"""Return the pipeline class."""
2347+
23472348
from vulnerabilities.importers import IMPORTERS_REGISTRY
23482349
from vulnerabilities.improvers import IMPROVERS_REGISTRY
2350+
from vulnerabilities.pipelines.exporters import EXPORTERS_REGISTRY
2351+
2352+
pipeline_registry = IMPORTERS_REGISTRY | IMPROVERS_REGISTRY | EXPORTERS_REGISTRY
23492353

2350-
if self.pipeline_id in IMPROVERS_REGISTRY:
2351-
return IMPROVERS_REGISTRY.get(self.pipeline_id)
2352-
if self.pipeline_id in IMPORTERS_REGISTRY:
2353-
return IMPORTERS_REGISTRY.get(self.pipeline_id)
2354+
return pipeline_registry[self.pipeline_id]
23542355

23552356
@property
23562357
def description(self):

0 commit comments

Comments
 (0)