New functions using Azure SDK for Python

Python/azure_iot_hub_list_devices.py

@@ -0,0 +1,59 @@
+from azure.iot.hub import IoTHubRegistryManager
+from azure.iot.hub.protocol.models import QuerySpecification
+from azure.iot.hub.models import CloudToDeviceMethod, CloudToDeviceMethodResult
+from dotenv import load_dotenv
+
+import json
+import os
+
+load_dotenv()
+
+module_id = "thingspro-agent"
+method_name = "thingspro-api-v1"
+payload = '{"method":"GET", "path":"/device/general"}'
+
+# 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 capabilities.iotEdge = true AND tags.site = 'PIERREFONDS'")
+
+query_result = registry_manager.query_iot_hub(query_spec)
+
+devices = []
+for item in query_result.items:
+    deviceId = str(item.device_id)
+    site = str(item.tags['site'])
+    number = int(item.tags['number'])
+    cloud_version = str(item.tags['version'])
+    devices.append([deviceId, site, number, cloud_version])
+
+ordered_devices = sorted(devices, key = lambda x: (x[1], x[2]))
+
+for index in range(len(ordered_devices)):
+    current_device_modules = registry_manager.get_modules(ordered_devices[index][0])
+    for module in current_device_modules:
+        if module.module_id == module_id:
+            thingspro_module = module
+            print(thingspro_module)
+        try:
+            direct_method = CloudToDeviceMethod(method_name=method_name, payload=json.loads(payload))
+            response = registry_manager.invoke_device_module_method(device_id=ordered_devices[index][0], module_id=module_id, direct_method_request=direct_method)
+            # print(str(ordered_devices[index][0]), str(ordered_devices[index][1]), str(ordered_devices[index][2]), response.payload['data']['description'], "device version: " + str(response.payload['data']['firmwareVersion']), "cloud version: " + str(ordered_devices[index][3]), sep=";")
+            device_version = str(response.payload['data']['firmwareVersion'])
+            if device_version != "1.6.0":
+                payload = '{"deleteFileAfterInstallComplete": true, "install": true, "url": "https://files.thingsprocloud.com/package/Upgrade_AIG-301_2.5.0-4404_IMG_1.5_to_1.6.0.yaml"}'
+                direct_method = CloudToDeviceMethod(method_name=method_name, payload=json.loads(payload))
+        except:
+            print(str(ordered_devices[index][0]), str(ordered_devices[index][1]), str(ordered_devices[index][2]), "UNREACHABLE", "device version: UNREACHABLE", "cloud version: " + str(ordered_devices[index][3]), sep=";")
+    else:
+        print("No thingspro-agent available for " + ordered_devices[index][0] + " (" + ordered_devices[index][1] + " " + ordered_devices[index][2] + ")")
+