linter

tarnopolskyi · tarnopolskyi · commit b2ac41836a85 · 2024-05-27T10:46:34.000+02:00
diff --git a/uncoder-core/app/translator/core/custom_types/tokens.py b/uncoder-core/app/translator/core/custom_types/tokens.py
@@ -33,6 +33,12 @@ class GroupType(CustomEnum):
 
 
 STR_SEARCH_OPERATORS = (
-    OperatorType.CONTAINS, OperatorType.NOT_CONTAINS, OperatorType.ENDSWITH, OperatorType.NOT_ENDSWITH,
-    OperatorType.STARTSWITH, OperatorType.NOT_STARTSWITH, OperatorType.REGEX, OperatorType.NOT_REGEX
+    OperatorType.CONTAINS,
+    OperatorType.NOT_CONTAINS,
+    OperatorType.ENDSWITH,
+    OperatorType.NOT_ENDSWITH,
+    OperatorType.STARTSWITH,
+    OperatorType.NOT_STARTSWITH,
+    OperatorType.REGEX,
+    OperatorType.NOT_REGEX,
 )
diff --git a/uncoder-core/app/translator/core/models/field.py b/uncoder-core/app/translator/core/models/field.py
@@ -1,6 +1,6 @@
 from typing import Optional, Union
 
-from app.translator.core.custom_types.tokens import OperatorType, STR_SEARCH_OPERATORS
+from app.translator.core.custom_types.tokens import STR_SEARCH_OPERATORS, OperatorType
 from app.translator.core.mapping import DEFAULT_MAPPING_NAME, SourceMapping
 from app.translator.core.models.identifier import Identifier
 from app.translator.core.str_value_manager import StrValue
@@ -49,7 +49,12 @@ def __add_value(self, value: Optional[Union[int, str, StrValue, list, tuple]]) -
         if value and isinstance(value, (list, tuple)):
             for v in value:
                 self.__add_value(v)
-        elif value and isinstance(value, str) and value.isnumeric() and self.operator.token_type not in STR_SEARCH_OPERATORS:
+        elif (
+            value
+            and isinstance(value, str)
+            and value.isnumeric()
+            and self.operator.token_type not in STR_SEARCH_OPERATORS
+        ):
             self.values.append(int(value))
         elif value is not None and isinstance(value, (int, str)):
             self.values.append(value)
diff --git a/uncoder-core/app/translator/platforms/base/aql/mapping.py b/uncoder-core/app/translator/platforms/base/aql/mapping.py
@@ -28,11 +28,12 @@ def is_suitable(
         device_type_match = set(devicetype).issubset(self.device_types) if devicetype else None
         category_match = set(category).issubset(self.categories) if category else None
         qid_match = set(qid).issubset(self.qids) if qid else None
-        qid_event_category_match = set(qideventcategory).issubset(self.qid_event_categories) if qideventcategory else None
+        qid_event_category_match = (
+            set(qideventcategory).issubset(self.qid_event_categories) if qideventcategory else None
+        )
         return all(
-            condition for condition in (
-                device_type_match, category_match,
-                qid_match, qid_event_category_match)
+            condition
+            for condition in (device_type_match, category_match, qid_match, qid_event_category_match)
             if condition is not None
         )
 
diff --git a/uncoder-core/app/translator/platforms/palo_alto/renders/cortex_xsiam.py b/uncoder-core/app/translator/platforms/palo_alto/renders/cortex_xsiam.py
@@ -20,7 +20,6 @@
 
 from app.translator.const import DEFAULT_VALUE_TYPE
 from app.translator.core.custom_types.values import ValueType
-from app.translator.core.exceptions.render import UnsupportedRenderMethod
 from app.translator.core.models.platform_details import PlatformDetails
 from app.translator.core.render import BaseQueryFieldValue, PlatformQueryRender
 from app.translator.managers import render_manager
@@ -95,7 +94,7 @@ def is_not_none(self, field: str, value: DEFAULT_VALUE_TYPE) -> str:
             return f"({self.or_token.join(self.is_not_none(field=field, value=v) for v in value)})"
         return f"{field} != null"
 
-    def keywords(self, field: str, value: DEFAULT_VALUE_TYPE) -> str:  # noqa: ARG002
+    def keywords(self, field: str, value: DEFAULT_VALUE_TYPE) -> str:
         if isinstance(value, list):
             return f"({self.or_token.join(self.contains_modifier(field=field, value=v) for v in value)})"
         if value.endswith("\\"):
diff --git a/uncoder-core/app/translator/platforms/qradar/renders/qradar.py b/uncoder-core/app/translator/platforms/qradar/renders/qradar.py
@@ -16,7 +16,8 @@
 limitations under the License.
 -----------------------------------------------------------------
 """
-from typing import Callable, List, Optional, Union
+from typing import Callable, Optional, Union
+
 from app.translator.const import DEFAULT_VALUE_TYPE
 from app.translator.core.models.platform_details import PlatformDetails
 from app.translator.core.str_value_manager import StrValue
@@ -29,24 +30,33 @@ class QradarFieldValue(AQLFieldValue):
     details: PlatformDetails = qradar_query_details
 
     def _has_wildcards(self, value: str) -> bool:
-        return any(item in value for item in ('%', '_'))
+        return any(item in value for item in ("%", "_"))
 
-    def _render_qradar_modifiers(self, field: str, values: List[Union[int, str, StrValue]], modifier_func: Callable, prefix: Optional[str] = '', suffix: Optional[str] = '') -> str:
+    def _render_qradar_modifiers(
+        self,
+        field: str,
+        values: list[Union[int, str, StrValue]],
+        modifier_func: Callable,
+        prefix: Optional[str] = "",
+        suffix: Optional[str] = "",
+    ) -> str:
         """
-            Renders values for a list of inputs, applying the appropriate modifier or regex function.
-            
-            Args:
-                field - str: The database field to modify.
-                values - List[Union[int, str, StrValue]]: The list of values to process.
-                modifier_fun - Callable: The function to call for non-regex modifications.
-                prefix - Optional[str]: The prefix to add to each value for regex. Defaults to ''.
-                suffix - Optional[str]: The suffix to add to each value for regex. Defaults to ''.
-            
-            Returns:
-                str: The rendered query part.
+        Renders values for a list of inputs, applying the appropriate modifier or regex function.
+
+        Args:
+            field - str: The database field to modify.
+            values - List[Union[int, str, StrValue]]: The list of values to process.
+            modifier_fun - Callable: The function to call for non-regex modifications.
+            prefix - Optional[str]: The prefix to add to each value for regex. Defaults to ''.
+            suffix - Optional[str]: The suffix to add to each value for regex. Defaults to ''.
+
+        Returns:
+            str: The rendered query part.
         """
         rendered_values = [
-            self.regex_modifier(field, f"{prefix}{v}{suffix}") if self._has_wildcards(str(v)) else modifier_func(field, v)
+            self.regex_modifier(field, f"{prefix}{v}{suffix}")
+            if self._has_wildcards(str(v))
+            else modifier_func(field, v)
             for v in values
         ]
         return f"({self.or_token.join(rendered_values)})"
@@ -75,7 +85,6 @@ def contains_modifier(self, field: str, value: DEFAULT_VALUE_TYPE) -> str:
 
 @render_manager.register
 class QradarQueryRender(AQLQueryRender):
-
     or_token = "OR"
 
     field_value_map = QradarFieldValue(or_token=or_token)
