@@ -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 ,
@@ -125,9 +128,6 @@ def main():
125128 all_args .add_argument (
126129 "-p" , "--password" , required = True , help = "Password for the user."
127130 )
128- all_args .add_argument (
129- "-v" , "--nifi_version" , required = True , help = "The NiFi product version."
130- )
131131 all_args .add_argument (
132132 "-c" ,
133133 "--cert" ,
@@ -158,9 +158,7 @@ def main():
158158 reporting_task = find_reporting_task (name = task_name , port = port )
159159
160160 if reporting_task is None :
161- reporting_task = create_reporting_task (
162- name = task_name , port = port , version = args ["nifi_version" ]
163- )
161+ reporting_task = create_reporting_task (name = task_name , port = port )
164162 print (
165163 get_reporting_task_name (task = reporting_task )
166164 + " [%s] -> CREATED" % reporting_task .id
0 commit comments