From 05c973db306f00e7f00ac32e9b52d92275d5b22d Mon Sep 17 00:00:00 2001 From: Thomas Patzke Date: Sun, 5 Jan 2025 02:28:28 +0100 Subject: [PATCH] WIP --- sigma/rule/base.py | 12 +++++++----- 1 file changed, 7 insertions(+), 5 deletions(-) diff --git a/sigma/rule/base.py b/sigma/rule/base.py index 891ab14f..a79ee613 100644 --- a/sigma/rule/base.py +++ b/sigma/rule/base.py @@ -372,7 +372,7 @@ def from_yaml( def to_dict(self) -> Dict[str, Any]: """Convert rule object into dict.""" - d = { + d: Dict[str, Any] = { "title": self.title, } # Convert to string where possible @@ -398,7 +398,7 @@ def to_dict(self) -> Dict[str, Any]: return d - def add_backreference(self, rule: "SigmaRuleBase"): + def add_backreference(self, rule: "SigmaRuleBase") -> None: """Add backreference to another rule.""" self._backreferences.append(rule) @@ -406,7 +406,7 @@ def referenced_by(self, rule: "SigmaRuleBase") -> bool: """Check if rule is referenced by another rule.""" return rule in self._backreferences - def set_conversion_result(self, result: List[Any]): + def set_conversion_result(self, result: List[Any]) -> None: """Set conversion result.""" self._conversion_result = result @@ -415,11 +415,12 @@ def get_conversion_result(self) -> List[Any]: if self._conversion_result is None: raise sigma_exceptions.SigmaConversionError( self, + None, "Conversion result not available", ) return self._conversion_result - def set_conversion_states(self, state: List["sigma.conversion.state.ConversionState"]): + def set_conversion_states(self, state: List["sigma.conversion.state.ConversionState"]) -> None: """Set conversion state.""" self._conversion_states = state @@ -428,11 +429,12 @@ def get_conversion_states(self) -> List["sigma.conversion.state.ConversionState" if self._conversion_states is None: raise sigma_exceptions.SigmaConversionError( self, + None, "Conversion state not available", ) return self._conversion_states - def disable_output(self): + def disable_output(self) -> None: """Disable output of rule.""" self._output = False