I have to create a new csv file after removing first column from incoming CSV file and this has to be implemented using Python script. I am not much aware of python scripting and tried to write python code after following some sites. The python code is mentioned below: ***************************************************** from org.apache.commons.io import IOUtils from java.nio.charset import StandardCharsets from org.apache.nifi.processor.io import StreamCallback import csv # Define a subclass of StreamCallback for use in session.write() class PyStreamCallback(StreamCallback): def __init__(self): pass def process(self, inputStream, outputStream): cols_to_remove = [0] # Column indexes to be removed (starts at 0) cols_to_remove = sorted(cols_to_remove, reverse=True) # Reverse so we remove from the end first row_count = 0 # Current amount of rows processed newText ='' Text = IOUtils.toString(inputStream, StandardCharsets.UTF_8) reader = csv.reader(Text,delimiter=",") with open(filePath, "w", newline='') as result: writer = csv.writer(result) for row in reader: row_count += 1 print('\r{0}'.format(row_count), end='') # Print rows processed for col_index in cols_to_remove: del row[col_index] writer.writerow(row) outputStream.write(newText.encode('utf-8')) flowFile = session.get() if(flowFile != None): flowFile = session.write(flowFile, PyStreamCallback()) session.transfer(flowFile, REL_SUCCESS) ******************************************************** However, i am not getting the result as expected. Could someone look into the code and let me know what is to be modified.
... View more
