132 lines
4.8 KiB
Python
132 lines
4.8 KiB
Python
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)
|
|
|
|
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 main():
|
|
|
|
ip_address_prefix = "10.84.171."
|
|
start_ip = 1
|
|
end_ip = 5
|
|
SITE_NAME = "MYRTLE"
|
|
|
|
print(f"Site: {SITE_NAME}")
|
|
print(f"From {ip_address_prefix}{str(start_ip)} to {ip_address_prefix}{str(end_ip)}")
|
|
|
|
file_numbering = 0
|
|
|
|
while os.path.exists(SITE_NAME + "_" + str(file_numbering) + ".csv"):
|
|
file_numbering = file_numbering + 1
|
|
|
|
csv_filename = SITE_NAME + "_" + str(file_numbering) + ".csv"
|
|
|
|
print(f"Logging results to {csv_filename}")
|
|
|
|
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 = []
|
|
|
|
for i in range (start_ip, end_ip):
|
|
ip_address = f"{ip_address_prefix}{i}"
|
|
print(f"Activating SSH for {ip_address}:", end=" ")
|
|
|
|
try:
|
|
activate_ssh(ip_address)
|
|
except Exception as e:
|
|
writer.writerow([i, ip_address, "UNREACHABLE", "NA", "NA"])
|
|
file.flush()
|
|
continue
|
|
|
|
ssh_command = "hostname"
|
|
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)
|
|
|
|
ssh_command = "grep \"connection-string\" /etc/cube/config-azure.properties"
|
|
print(f"Getting configured Connection String for {ip_address}:", end=" ")
|
|
try:
|
|
connection_string = execute_ssh_command(ip_address, ssh_command)
|
|
if connection_string == "":
|
|
raise Exception("No Connection String was 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"
|
|
finally:
|
|
print(f"{migration} {status}")
|
|
|
|
writer.writerow([i, ip_address, cube_id, migration, status])
|
|
file.flush()
|
|
|
|
if __name__ == "__main__":
|
|
main()
|