Skip to content

Commit 159af01

Browse files
committed
fix: determine nifi version in nifi 1 create reporting task script
1 parent 5485644 commit 159af01

File tree

1 file changed

+8
-3
lines changed

1 file changed

+8
-3
lines changed

nifi/python/create_nifi_reporting_task.py

Lines changed: 8 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -47,16 +47,21 @@ def find_reporting_task(name: str, port: str):
4747
return None
4848

4949

50-
def create_reporting_task(name: str, port: str, version: str):
50+
def create_reporting_task(name: str, port: str):
5151
"""Create a ReportingTask"""
52+
nifi_version = (
53+
nipyapi.system.get_system_diagnostics().system_diagnostics.aggregate_snapshot.version_info.ni_fi_version
54+
)
5255
task = nipyapi.nifi.models.reporting_task_entity.ReportingTaskEntity(
5356
revision=nipyapi.nifi.models.revision_dto.RevisionDTO(version=0),
5457
disconnected_node_acknowledged=False,
5558
component=nipyapi.nifi.models.reporting_task_dto.ReportingTaskDTO(
5659
name=name,
5760
type="org.apache.nifi.reporting.prometheus.PrometheusReportingTask",
5861
bundle=nipyapi.nifi.models.bundle_dto.BundleDTO(
59-
group="org.apache.nifi", artifact="nifi-prometheus-nar", version=version
62+
group="org.apache.nifi",
63+
artifact="nifi-prometheus-nar",
64+
version=nifi_version,
6065
),
6166
properties={
6267
"prometheus-reporting-task-metrics-endpoint-port": port,
@@ -159,7 +164,7 @@ def main():
159164

160165
if reporting_task is None:
161166
reporting_task = create_reporting_task(
162-
name=task_name, port=port, version=args["nifi_version"]
167+
name=task_name, port=port
163168
)
164169
print(
165170
get_reporting_task_name(task=reporting_task)

0 commit comments

Comments
 (0)