linter

Author: tarnopolskyi

uncoder-core/app/translator/core/str_value_manager.py

@@ -16,6 +16,7 @@
 limitations under the License.
 -----------------------------------------------------------------
 """
+
 from typing import ClassVar, Optional, TypeVar, Union
 
 from app.translator.core.custom_types.values import ValueType

uncoder-core/app/translator/platforms/logrhythm_axon/renders/logrhythm_axon_query.py

@@ -16,9 +16,11 @@
 limitations under the License.
 -----------------------------------------------------------------
 """
+
 from typing import Union
 
 from app.translator.const import DEFAULT_VALUE_TYPE
+from app.translator.core.context_vars import return_only_first_query_ctx_var
 from app.translator.core.custom_types.tokens import LogicalOperatorType
 from app.translator.core.custom_types.values import ValueType
 from app.translator.core.exceptions.core import StrictPlatformException
@@ -262,6 +264,8 @@ def _generate_from_tokenized_query_container(self, query_container: TokenizedQue
                 meta_info=query_container.meta_info,
                 source_mapping=source_mapping,
             )
+            if return_only_first_query_ctx_var.get() is True:
+                return finalized_query
             queries_map[source_mapping.source_id] = finalized_query
 
         return self.finalize(queries_map)

uncoder-core/app/translator/platforms/palo_alto/renders/cortex_xsiam.py

@@ -51,12 +51,12 @@ def _get_value_type(field_name: str, value: Union[int, str, StrValue], value_typ
 
     @staticmethod
     def _wrap_str_value(value: str) -> str:
-        if value == 'SetValue':
-            return f'"REGISTRY_SET_VALUE"'
-        if value == 'DeleteValue':
-            return f'"REGISTRY_DELETE_VALUE"'
-        if value == 'CreateKey':
-            return f'"REGISTRY_CREATE_KEY"'
+        if value == "SetValue":
+            return '"REGISTRY_SET_VALUE"'
+        if value == "DeleteValue":
+            return '"REGISTRY_DELETE_VALUE"'
+        if value == "CreateKey":
+            return '"REGISTRY_CREATE_KEY"'
         return f'"{value}"'
 
     def equal_modifier(self, field: str, value: DEFAULT_VALUE_TYPE) -> str: