Update action_mapper.py

Author: pierrerondel

transformers/action_mapper.py

@@ -1,57 +1,38 @@
 """Action mapping transformer.
 
-This module defines a transformer responsible for mapping action values
-between vendor-specific representations and the universal data model.
+This module provides the ActionMapper class, which converts action values
+between vendor-specific and universal representations.
 """
 
-from typing import Any
-from typing import Dict
+from typing import Dict, Any
 
 from .base_transformer import BaseTransformer
 
 
 class ActionMapper(BaseTransformer):
-    """Map action values between vendor and universal models.
-
-    This transformer replaces the ``action`` field of an item using a
-    predefined mapping dictionary. If the action is not found in the
-    mapping, it is left unchanged.
-    """
+    """Maps actions from vendor-specific values to universal values and vice versa."""
 
     def __init__(self, action_map: Dict[str, str]) -> None:
         """Initialize the ActionMapper.
 
         Args:
-            action_map: A dictionary mapping source action values to
-                destination action values.
+            action_map: A mapping from vendor-specific action names
+                to universal action names.
         """
         self.action_map = action_map
 
     def transform(self, item: Dict[str, Any]) -> Dict[str, Any]:
-        """Transform an item's action field using the action mapping.
+        """Transform the action field of an item using the action map.
+
+        If the item's ``action`` value exists in the action map, it is
+        replaced with the mapped universal value.
 
         Args:
-            item: A dictionary representing a single rule or configuration
-                entry containing an ``action`` field.
+            item: A dictionary representing the item to transform.
 
         Returns:
-            The transformed item with its ``action`` field mapped according
-            to the configured action map.
+            The transformed item dictionary.
         """
-        action = item.get("action")
-        if action in self.action_map:
-            item["action"] = self.action_map[action]
-
-        return itemfrom typing import Dict, Any
-from .base_transformer import BaseTransformer
-
-class ActionMapper(BaseTransformer):
-    """Maps actions from vendor to universal and vice versa."""
-
-    def __init__(self, action_map: Dict[str, str]):
-        self.action_map = action_map
-
-    def transform(self, item: Dict[str, Any]) -> Dict[str, Any]:
         action = item.get("action")
         if action in self.action_map:
             item["action"] = self.action_map[action]