New tools for restarting services and fixes

This commit is contained in:
Quentin WEPHRE
2025-06-27 08:40:36 +02:00
parent 856d7ecac3
commit d8df4365c3
7 changed files with 222 additions and 18 deletions

View File

@@ -25,7 +25,7 @@ if CONNECTION_STRING == "":
registry_manager = IoTHubRegistryManager.from_connection_string(CONNECTION_STRING)
query_spec = QuerySpecification(query="SELECT * FROM devices WHERE IS_DEFINED(tags.site) AND capabilities.iotEdge = true ")
query_spec = QuerySpecification(query="SELECT * FROM devices WHERE IS_DEFINED(tags.site) AND tags.site = 'DANISH' AND capabilities.iotEdge = true ")
query_result = registry_manager.query_iot_hub(query_spec)

View File

@@ -0,0 +1,51 @@
from azure.iot.hub import IoTHubRegistryManager
from azure.iot.hub.protocol.models import QuerySpecification, Module
from azure.iot.hub.models import CloudToDeviceMethod, CloudToDeviceMethodResult
from dotenv import load_dotenv
from isight_device import iSightDevice
import json
import os
load_dotenv()
module_id = "$edgeAgent"
method_name = "RestartModule"
module = "edgeHub"
payload = '{"schemaVersion":"1.0", "id":"' + module + '"}'
# Install the Azure IoT Hub SDK:
# pip install azure-iot-hub
# Authenticate to your Azure account
# CONNECTION_STRING = str(os.getenv("CONNECTION_STRING_SAFT_PROD"))
CONNECTION_STRING = str(os.getenv("CONNECTION_STRING_INOX_PROD"))
if CONNECTION_STRING == "":
print("Provide a connection string for the Iot Hub before running the script!")
exit(13)
registry_manager = IoTHubRegistryManager.from_connection_string(CONNECTION_STRING)
query_spec = QuerySpecification(query="SELECT * FROM devices WHERE IS_DEFINED(tags.site) AND tags.site = 'DANISH' AND capabilities.iotEdge = true ")
query_result = registry_manager.query_iot_hub(query_spec)
devices = []
for item in query_result.items:
# currentDevice = iSightDevice(str(item.device_id), str(item.tags['site']), int(item.tags['number']), str(item.tags['version']))
devices.append(iSightDevice(str(item.device_id), str(item.tags['site']), int(item.tags['number']), str(item.tags['version'])))
devices.sort(key = lambda d: (d.site, d.number))
for device in devices:
print(device, end="\t")
current_device_modules = registry_manager.get_modules(device.deviceId)
for module in current_device_modules:
if (module.module_id == "thingspro-agent"):
device.setModule(module)
try:
direct_method = CloudToDeviceMethod(method_name=method_name, payload=json.loads(payload))
response = registry_manager.invoke_device_module_method(device_id=device.deviceId, module_id=module_id, direct_method_request=direct_method)
print(response)
except:
print("ERROR")

View File

@@ -6,6 +6,7 @@ import os
import urllib3
from dotenv import load_dotenv
import io
import time
load_dotenv(override=True)
@@ -31,8 +32,7 @@ def authenticate(base_url):
}
try:
print(f"Authenticating as {username}...")
response = requests.post(auth_url, files=files, verify=False)
response = requests.post(auth_url, files=files, verify=False, timeout=10)
response.raise_for_status() # Raise exception for 4XX/5XX responses
# Extract token from response
@@ -40,9 +40,9 @@ def authenticate(base_url):
token = auth_data.get("token")
if not token:
print("Error: No token received in authentication response")
print("Authentication failure!")
print("Authentication successful.")
print("Authentication success!", end = " ")
return token
except requests.exceptions.RequestException as e:
@@ -66,16 +66,16 @@ def set_ssh_status(base_url, token):
payload = { "currentStatus": True }
try:
print(f"Sending request to enable SSH...")
response = requests.post(ssh_url, headers=headers, json=payload, verify=False)
response = requests.post(ssh_url, headers=headers, json=payload, verify=False, timeout=10)
response.raise_for_status()
print(f"SSH enabled successfully!")
print(f"SSH activation success!")
return True
except requests.exceptions.RequestException as e:
print(f"SSH activation failed: {e}")
print("SSH activation failure!")
print(f"Exception: {e}")
if hasattr(e, 'response') and e.response:
print(f"Response: {e.response.text}")
return False
@@ -88,10 +88,8 @@ def activate_ssh(ip_address):
# Convert http:// to https:// or add https:// if no protocol specified
if url.startswith("http://"):
url = "https://" + url[7:]
print(f"Converting to HTTPS: {url}")
else:
url = "https://" + url
print(f"Adding HTTPS protocol: {url}")
if not url.endswith(":9080"):
url = url + ":9080"
@@ -102,5 +100,5 @@ def activate_ssh(ip_address):
token = authenticate(url)
if not token:
return
time.sleep(3)
set_ssh_status(url, token)

View File

@@ -13,7 +13,7 @@ load_dotenv(override=True)
ip_address_prefix = "10.188.11."
ssh_command = "hostname"
csv_filename = "hoohana6.csv"
csv_filename = "hoohana7.csv"
SITE_NAME = "HOOHANA"
ssh_username = os.getenv("DEFAULT_CUBE_LINUX_ADMIN_USER")
@@ -52,6 +52,24 @@ def update_cloud_config(ip, new_content):
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...")
@@ -64,7 +82,8 @@ def main():
results = []
for i in range (54, 55):
numbers = [16, 18, 34, 35]
for i in numbers:
ip_address = f"{ip_address_prefix}{i}"
print(f"Activating SSH for {ip_address}:", end=" ")
@@ -162,6 +181,8 @@ def main():
continue
print("Done!")
restart_cloudagent(ip_address)
writer.writerow([i, ip_address, cube_id, migration, status])
file.flush()

View File

@@ -0,0 +1,131 @@
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()

View File

@@ -1,2 +1,5 @@
requests
dotenv-python
azure-core
azure-iot-hub
python-dotenv
paramiko
requests