Scripts: Linting.

Author: vladd-bit

nifi/user_python_extensions/convert_json_record_schema.py

@@ -293,7 +293,7 @@ def map_record(self, record: dict, json_mapper_schema: dict) -> dict:
                     parts: list[str] = []
                     for sub_field in old_field:
                         rendered = self._value_for_composite(record, sub_field)
-                        if rendered not in (None, ""):
+                        if rendered is not None and rendered != "":
                             parts.append(rendered)
                     self._set_field(new_record, new_field, "\n".join(parts) if parts else None)
 

nifi/user_python_extensions/record_decompress_cerner_blob.py

@@ -232,7 +232,7 @@ def process(self, context: ProcessContext, flowFile: JavaObject) -> FlowFileTran
         try:
 
             # attempt to see if not a proper compressed blob 
-            output_bytes_decompressed = None
+            output_bytes_decompressed: bytes | bytearray | None = None
             try:
                 if full_compressed_blob.find(b"%PDF") != -1:
                     pdf_start = full_compressed_blob.find(b"%PDF-")
@@ -256,7 +256,10 @@ def process(self, context: ProcessContext, flowFile: JavaObject) -> FlowFileTran
                 decompress_blob.decompress(full_compressed_blob)
                 output_bytes_decompressed = bytes(decompress_blob.output_stream)
 
-            output_merged_record[self.binary_field_name] = output_bytes_decompressed
+            if output_bytes_decompressed is None:
+                raise ValueError("Could not locate or decompress blob payload")
+
+            output_merged_record[self.binary_field_name] = bytes(output_bytes_decompressed)
 
         except Exception as exception:
             raise RuntimeError("Error decompressing Cerner LZW blob") from exception

nifi/user_python_extensions/sample_processor.py

@@ -145,11 +145,10 @@ def process(self, context: ProcessContext, flowFile: JavaObject) -> FlowFileTran
        output_byte_buffer.seek(0)
 
        # add properties to flowfile attributes
-       attributes: dict = {k: str(v) for k, v in flowFile.getAttributes().items()}
        attributes["sample_property_one"] = str(self.sample_property_one)
        attributes["sample_property_two"] = str(self.sample_property_two)
        attributes["sample_property_three"] = str(self.sample_property_three)
 
        return FlowFileTransformResult(relationship="success",
                                       attributes=attributes,
-                                      contents=json.dumps(output_contents))
+                                      contents=json.dumps(output_contents).encode("utf-8"))

typings/nifiapi/__init__.pyi

@@ -14,9 +14,9 @@ import nifiapi.relationship  # noqa: F401
 from .flowfiletransform import (
     FlowFileTransform,
     FlowFileTransformResult,
-    ProcessContext,
 )
 from .properties import (
+    ProcessContext,
     PropertyDescriptor,
     StandardValidators,
 )