使用Python代码调用NiFi API启动处理器遇201认证错误求助

问题概述 #

我编写了一段Python代码，尝试通过用户名和密码认证NiFi服务器并启动指定处理器，但认证请求返回状态码201，无法完成后续操作。代码如下：

from flask import Flask
import nipyapi
import requests
 
app = Flask(__name__)

@app.route('/')
def home():
    nifi_url = "https://mynifiinstance:port/nifi-api/access"
    processor_id = <processorID>
    nifi_general = "https://mynifiinstance/nifi-api"
    start_processor_url = f"{nifi_general}/processors/{processor_id}/run-status"
 
    headers = {'Content-Type': 'application/x-www-form-urlencoded'}
    username = "username"
    password = "password"
    data = { "username": username,
            "password": password
        }
 
 
    response = requests.post(url=nifi_url, data=data,verify=False, headers=headers)
 
    if response.status_code == 200:
        # if successfully authenticated, then start the processor
        start_processor = requests.put(url=start_processor_url, headers=headers)
        print(start_processor.status_code)
        if start_processor.status_code == 200:
            return("Processor started successfully.")
        else:
            return(f"Starting processor failed. Status code: {start_processor.status_code}")
    else:
        return(f"Failed to start processor. Status code: {response.status_code}")
   
if __name__ == '__main__':
    app.run()

我尚未修改nifi.properties文件，请问是否需要修改该文件来解决此问题？

问题分析与解决 #

1. 认证状态码的误解 #

首先明确：NiFi的/nifi-api/access认证接口成功时返回201状态码，不是200。201表示认证成功并创建了新的会话，同时会在响应头中设置Set-Cookie字段，包含会话令牌。你的代码错误地以200作为成功判断条件，这是核心问题之一。

2. 是否需要修改nifi.properties？ #

这取决于NiFi当前的认证模式：

• 如果是单用户模式（默认初始模式）：
  • 检查nifi.properties中的nifi.security.user.authentication是否为single-user
  • 确认nifi.security.user.single.user.credentials设置的用户名密码与代码中一致
    这种情况下通常不需要额外修改，只要凭证正确即可。
• 如果是LDAP认证模式：
  • 需确保nifi.properties中LDAP相关配置（如nifi.security.user.ldap.url、nifi.security.user.ldap.user.search.base等）正确配置，否则会导致认证失败。
• 如果nifi.security.user.authentication设为none：
  • NiFi未启用认证，此时不需要发送/access请求，直接调用处理器接口即可。

3. 修正后的代码示例 #

需要调整状态码判断逻辑，并在后续请求中携带认证成功后的Cookie：

from flask import Flask
import requests
 
app = Flask(__name__)

@app.route('/')
def home():
    nifi_url = "https://mynifiinstance:port/nifi-api/access"
    processor_id = "YOUR_ACTUAL_PROCESSOR_ID"  # 替换为真实处理器ID，注意加引号
    nifi_general = "https://mynifiinstance/nifi-api"
    start_processor_url = f"{nifi_general}/processors/{processor_id}/run-status"
 
    headers = {'Content-Type': 'application/x-www-form-urlencoded'}
    username = "username"
    password = "password"
    data = { "username": username,
            "password": password
        }
 
    # 发送认证请求
    response = requests.post(url=nifi_url, data=data, verify=False, headers=headers)
 
    if response.status_code == 201:  # 201为NiFi认证成功的状态码
        # 提取会话Cookie
        session_cookies = response.cookies
        # 启动处理器时携带认证Cookie
        start_processor = requests.put(
            url=start_processor_url,
            headers=headers,
            cookies=session_cookies,
            verify=False  # 生产环境请移除该参数，配置合法SSL证书
        )
        print(start_processor.status_code)
        if start_processor.status_code == 200:
            return "Processor started successfully."
        else:
            return f"Starting processor failed. Status code: {start_processor.status_code}, Response: {start_processor.text}"
    else:
        return f"Authentication failed. Status code: {response.status_code}, Response: {response.text}"
   
if __name__ == '__main__':
    app.run()

额外注意事项 #

• 原代码中processor_id = <processorID>是占位符，需替换为真实的处理器ID字符串，并添加引号。
• 生产环境中禁止使用verify=False，应配置正确的SSL证书校验，避免安全风险。
• 若使用nipyapi库，可以直接利用其认证方法简化代码，无需手动处理Cookie：

  nipyapi.config.nifi_config.host = "https://mynifiinstance:port/nifi-api"
  nipyapi.security.authenticate(username, password)
  # 之后直接调用nipyapi的处理器启动方法
  nipyapi.canvas.schedule_processor(processor_id, True)
  

内容的提问来源于stack exchange，提问作者Kiran