Removed unneeded mapping + added typing

SigmaHQ · Aug 30, 2023 · ee91fe2 · ee91fe2
1 parent 1c9f79d
commit ee91fe2
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 7 deletions.
diff --git a/sigma/pipelines/sysmon/__init__.py b/sigma/pipelines/sysmon/__init__.py
@@ -1,7 +1,3 @@
 from typing import Callable, Dict
 from .sysmon import sysmon_pipeline
-from sigma.processing.pipeline import ProcessingPipeline
-
-pipelines : Dict[str, Callable[[], ProcessingPipeline]] = {
-    "sysmon": sysmon_pipeline,
-}
+from sigma.processing.pipeline import ProcessingPipeline
diff --git a/sigma/pipelines/sysmon/sysmon.py b/sigma/pipelines/sysmon/sysmon.py
@@ -1,6 +1,7 @@
 from sigma.processing.transformations import AddConditionTransformation, ChangeLogsourceTransformation
 from sigma.processing.conditions import LogsourceCondition
 from sigma.processing.pipeline import ProcessingItem, ProcessingPipeline
+from sigma.pipelines.base import Pipeline
 
 sysmon_generic_logsource_eventid_mapping = {    # map generic Sigma log sources to Sysmon event ids
     "process_creation": 1,
@@ -29,8 +30,8 @@
     "sysmon_error": 255,
 }
 
-
-def sysmon_pipeline():
+@Pipeline
+def sysmon_pipeline() -> ProcessingPipeline:
     return ProcessingPipeline(
         name="Generic Log Sources to Sysmon Transformation",
         priority=10,