python 从NiFi ExecuteScript处理器使用的Python脚本示例，它从传入的流文件中读取第一行。

Posted 2021-05-09

from org.python.core.util.FileUtil import wrap

from org.apache.nifi.processors.script import ExecuteScript


first_line = None
flow_file = session.get()


def read_first_line(input):
    global first_line
    with wrap(input) as f:
        first_line = f.readline()


if flow_file:
    session.read(flow_file, read_first_line)
    if first_line:
        flow_file = session.putAttribute(flow_file, 'from-content', first_line)
        session.transfer(flow_file, ExecuteScript.REL_SUCCESS)