run precommit

Author: dmirizzi

target_elasticsearch/sinks.py

@@ -32,7 +32,9 @@ def __init__(
     ):
         super().__init__(target, stream_name, schema, key_properties)
         self.client = self._authenticated_client()
-        self.index_schema_fields = self.config.get("index_schema_fields", {}).get(self.stream_name, {})
+        self.index_schema_fields = self.config.get("index_schema_fields", {}).get(
+            self.stream_name, {}
+        )
         self.metadata_fields = self.config.get("metadata_fields", {}).get(self.stream_name, {})
         self.index_mappings = self.config.get("index_mappings", {}).get(self.stream_name, {})
         self.index_name = None
@@ -97,7 +99,9 @@ def _build_fields(
         for k, v in mapping.items():
             match = jsonpath_ng.parse(v).find(record)
             if len(match) == 0:
-                self.logger.warning(f"schema key {k} with json path {v} could not be found in record: {record}")
+                self.logger.warning(
+                    f"schema key {k} with json path {v} could not be found in record: {record}"
+                )
                 schemas[k] = v
             else:
                 if len(match) > 1:
@@ -153,9 +157,13 @@ def create_index(self, index: str) -> None:
                         index=index, fields=list(self.index_mappings.keys())
                     )[index]["mappings"].items()
                 }
-                if not all(self.index_mappings[key]["type"] == value for key, value in mappings.items()):
+                if not all(
+                    self.index_mappings[key]["type"] == value for key, value in mappings.items()
+                ):
                     try:
-                        self.client.indices.put_mapping(index=index, body={"properties": self.index_mappings})
+                        self.client.indices.put_mapping(
+                            index=index, body={"properties": self.index_mappings}
+                        )
                     except elasticsearch.exceptions.BadRequestError as e:
                         if e.message == "illegal_argument_exception":
                             self.logger.warning(
@@ -209,7 +217,9 @@ def process_batch(self, context: dict[str, Any]) -> None:
         Args:
             context: Dictionary containing batch processing context including records.
         """
-        updated_records, distinct_indices = self.build_request_body_and_distinct_indices(context["records"])
+        updated_records, distinct_indices = self.build_request_body_and_distinct_indices(
+            context["records"]
+        )
         for index in distinct_indices:
             self.create_index(index)
         try:

target_elasticsearch/target.py

@@ -154,8 +154,12 @@ def __init__(
             validate_config=validate_config,
             setup_mapper=setup_mapper,
         )
-        assert bool(self.config.get("username") is None) == bool(self.config.get("password") is None)
-        assert bool(self.config.get("api_key_id") is None) == bool(self.config.get("api_key") is None)
+        assert bool(self.config.get("username") is None) == bool(
+            self.config.get("password") is None
+        )
+        assert bool(self.config.get("api_key_id") is None) == bool(
+            self.config.get("api_key") is None
+        )
 
     @property
     def state(self) -> Dict: