def get_zookeeper_connect_string(): ah = AmbariHelper() hosts = ah.get_host_components() zkHosts = "" for item in hosts["items"]: if item["HostRoles"]["component_name"] == "ZOOKEEPER_SERVER": zkHosts += item["HostRoles"]["host_name"] zkHosts += ZOOKEEPER_PORT zkHosts += "," if len(zkHosts) > 2: return zkHosts[:-1] else: raise Exception("Failed to get Zookeeper information from Ambari!")