@@ -47,16 +47,19 @@ 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 = nipyapi .system .get_system_diagnostics ().system_diagnostics .aggregate_snapshot .version_info .ni_fi_version
5253 task = nipyapi .nifi .models .reporting_task_entity .ReportingTaskEntity (
5354 revision = nipyapi .nifi .models .revision_dto .RevisionDTO (version = 0 ),
5455 disconnected_node_acknowledged = False ,
5556 component = nipyapi .nifi .models .reporting_task_dto .ReportingTaskDTO (
5657 name = name ,
5758 type = "org.apache.nifi.reporting.prometheus.PrometheusReportingTask" ,
5859 bundle = nipyapi .nifi .models .bundle_dto .BundleDTO (
59- group = "org.apache.nifi" , artifact = "nifi-prometheus-nar" , version = version
60+ group = "org.apache.nifi" ,
61+ artifact = "nifi-prometheus-nar" ,
62+ version = nifi_version ,
6063 ),
6164 properties = {
6265 "prometheus-reporting-task-metrics-endpoint-port" : port ,
@@ -158,9 +161,7 @@ def main():
158161 reporting_task = find_reporting_task (name = task_name , port = port )
159162
160163 if reporting_task is None :
161- reporting_task = create_reporting_task (
162- name = task_name , port = port , version = args ["nifi_version" ]
163- )
164+ reporting_task = create_reporting_task (name = task_name , port = port )
164165 print (
165166 get_reporting_task_name (task = reporting_task )
166167 + " [%s] -> CREATED" % reporting_task .id
0 commit comments