diff --git a/Python/cube_ssh_batch_restart_service.py b/Python/cube_ssh_batch_restart_service.py new file mode 100644 index 0000000..feb91c9 --- /dev/null +++ b/Python/cube_ssh_batch_restart_service.py @@ -0,0 +1,190 @@ +import csv +import paramiko +import time +from cube_activate_ssh import activate_ssh +from dotenv import load_dotenv +import os +import re +from azure.iot.hub import IoTHubRegistryManager +from azure.iot.hub.models import Twin, TwinProperties + +load_dotenv(override=True) + +ip_address_prefix = "10.162.48." +ssh_command = "hostname" + +csv_filename = "lacabana01.csv" +SITE_NAME = "LACABANA" + +ssh_username = os.getenv("DEFAULT_CUBE_LINUX_ADMIN_USER") +ssh_password = os.getenv("DEFAULT_CUBE_LINUX_ADMIN_PASSWORD") +CONNECTION_STRING = str(os.getenv("CONNECTION_STRING_INOX_PROD")) + +def execute_ssh_command(ip, command): + client = paramiko.SSHClient() + client.set_missing_host_key_policy(paramiko.AutoAddPolicy()) + try: + client.connect(ip, port=11022, username=ssh_username, password=ssh_password, allow_agent=False, look_for_keys=False) + stdin, stdout, stderr = client.exec_command(command) + result = stdout.read().decode().lower().strip() + return result + except Exception as e: + print(f"SSH Error: {str(e)}") + raise + finally: + client.close() + +def update_cloud_config(ip, new_content): + client = paramiko.SSHClient() + client.set_missing_host_key_policy(paramiko.AutoAddPolicy()) + try: + client.connect(ip, port=11022, username=ssh_username, password=ssh_password, allow_agent=False, look_for_keys=False) + stdin, stdout, stderr = client.exec_command(f'sudo -S bash -c \'cat > /etc/cube/config-azure.properties << EOF\n{new_content}\nEOF\'\n') + stdin.write(ssh_password + "\n") + stdin.flush() + stdoutput = [line for line in stdout] + stderroutput = [line for line in stderr] + for output in stdoutput: + print(output.strip()) + except Exception as e: + print(f"SSH Error: {str(e)}") + raise + finally: + client.close() + +def restart_cloudagent(ip): + client = paramiko.SSHClient() + client.set_missing_host_key_policy(paramiko.AutoAddPolicy()) + try: + client.connect(ip, port=11022, username=ssh_username, password=ssh_password, allow_agent=False, look_for_keys=False) + stdin, stdout, stderr = client.exec_command(f'sudo -S bash -c \'systemctl restart cube-web-cloudagent << EOF\n\nEOF\'\n') + stdin.write(ssh_password + "\n") + stdin.flush() + stdoutput = [line for line in stdout] + stderroutput = [line for line in stderr] + for output in stdoutput: + print(output.strip()) + except Exception as e: + print(f"SSH Error: {str(e)}") + raise + finally: + client.close() + +def main(): + + print("Starting...") + + with open(csv_filename, mode="w", newline="") as file: + writer = csv.writer(file) + writer.writerow(["Number", "IP address", "Cube ID", "Environment", "Correct configuration"]) + + registry_manager = IoTHubRegistryManager.from_connection_string(CONNECTION_STRING) + + results = [] + + numbers = range(133, 157) #133 to 156 + for i in numbers: + ip_address = f"{ip_address_prefix}{i}" + print(f"Activating SSH for {ip_address}:", end=" ") + + try: + activate_ssh(ip_address) + except Exception as e: + print("Failed!") + writer.writerow([i, ip_address, "UNREACHABLE", "NA", "NA"]) + file.flush() + continue + + print("Activated!") + + print(f"Executing {ssh_command} for {ip_address}:", end=" ") + try: + cube_id = execute_ssh_command(ip_address, ssh_command) + except Exception as e: + print("Failed!") + writer.writerow([i, ip_address, "UNREACHABLE", "NA", "NA"]) + file.flush() + continue + print(cube_id) + + print(f"Getting configured Connection String") + try: + connection_string = execute_ssh_command(ip_address, "grep \"connection-string\" /etc/cube/config-azure.properties") + if connection_string == "": + raise Exception("No Connection String extracted!") + + iothub_match = re.search(r"hostname\\=(.*?);", connection_string, re.IGNORECASE) + iothub = iothub_match.group(1) if iothub_match else None + if iothub.lower() == "IotHub-CUBE-PROD.azure-devices.net".lower(): + migration = "SAFT" + elif iothub.lower() == "iot-ingest-ess-prod.azure-devices.net".lower(): + migration = "INOX" + else: + migration = "NONE" + + device_id_match = re.search(r"deviceid\\=(.*?);", connection_string, re.IGNORECASE) + cloud_cube_id = device_id_match.group(1) if device_id_match else None + if cloud_cube_id.lower() == cube_id.lower(): + status = "CORRECT" + else: + status = "INCORRECT" + + except Exception as e: + print(e) + migration = "NONE" + status = "INCORRECT" + + # if migration == "SAFT" or migration == "NONE": + # print("SAFT device, migrating to INOX...") + + # print("Creating device on INOX...") + # try: + # registry_manager.create_device_with_sas( + # cube_id, + # primary_key="", secondary_key="", + # status="enabled", + # iot_edge=False + # ) + # except Exception as iot_e: + # print("Error creating new device!") + # print(iot_e) + # continue + + # print("Adding tags to new device...") + # try: + # twin = registry_manager.get_twin(cube_id) + # twin_patch = Twin(properties=TwinProperties(desired={}), tags={ + # "site": SITE_NAME, + # "number": i + # }) + # registry_manager.update_twin(cube_id, twin_patch, twin.etag) + # except Exception as iot_e: + # print("Error assigning tags to new device!") + # print(iot_e) + + # print("Requesting primary key...") + # try: + # device_info = registry_manager.get_device(cube_id) + # primary_key = device_info.authentication.symmetric_key.primary_key + # new_connection_string = f"HostName\\={CONNECTION_STRING.split(';')[0].split('=')[1]};DeviceId\\={cube_id};SharedAccessKey\\={primary_key}" + # new_content = f'light-telemetry=false\ncompression-enabled=true\ntelemetry-on=true\nremote-update-on=true\nconnection-string={new_connection_string}' + # except Exception as iot_e: + # print("Error getting new Connection String!") + # print(iot_e) + # continue + # print("Setting new Connection String...") + # try: + # update_cloud_config(ip_address, new_content) + # except Exception as ssh_e: + # print("Error when setting the new Connection String!") + # print(ssh_e) + # continue + # print("Done!") + + restart_cloudagent(ip_address) + + writer.writerow([i, ip_address, cube_id, migration, status]) + file.flush() + +if __name__ == "__main__": + main()