1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
|
import requests
import config
# export TOKEN=$(curl -k -X POST https://127.0.0.1.nip.io/nifi-api/access/token\
# -H "Content-Type: application/x-www-form-urlencoded" -d 'username=lab08nifiuser&password=tartunifi2023')
def get_access_token():
token_resp = requests.post(
f"{config.NIFI_HOST}/nifi-api/access/token",
headers={"Content-Type": "application/x-www-form-urlencoded"},
data={"username": config.NIFI_USER, "password": config.NIFI_PASS},
verify=False
)
token_resp.raise_for_status()
token = token_resp.text.strip()
return token
# curl -sk -X POST "https://127.0.0.1.nip.io/nifi-api/process-groups/root/process-groups/upload" -H "Authorization: Bearer $TOKEN"\
# -F "file=@pipelines/test_pipeline.json" -F "groupName=MyNewProcessGroup" -F "positionX=0" -F "positionY=0"
# -F "disconnectedNodeAcknowledged=false" -F "clientId=unique-client-id-123"
def upload_nifi_pipeline(token, pipeline_path, processorGroup_name, username=config.NIFI_USER, password=config.NIFI_PASS, nifi_url=config.NIFI_HOST, position_x=0, position_y=0, client_id = "unique-client-id-123"):
with open(pipeline_path, "r") as json_file:
files = {
"file": json_file,
"groupName": (None, processorGroup_name),
"positionX": (None, str(position_x)),
"positionY": (None, str(position_y)),
"disconnectedNodeAcknowledged": (None, "false"),
"clientId": (None, client_id)
}
upload_resp = requests.post(
f"{nifi_url}/nifi-api/process-groups/root/process-groups/upload",
headers={"Authorization": f"Bearer {token}"},
files=files,
verify=False
)
upload_resp.raise_for_status()
print(f"✅ Uploaded process group '{processorGroup_name}' successfully!")
|
