49 lines
2.4 KiB
Python
49 lines
2.4 KiB
Python
from azure.iot.hub import IoTHubRegistryManager
|
|
from azure.iot.hub.protocol.models import QuerySpecification
|
|
from azure.iot.hub.models import CloudToDeviceMethod, CloudToDeviceMethodResult
|
|
|
|
import json
|
|
|
|
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 = "HostName=IotHub-CUBE-PROD.azure-devices.net;SharedAccessKeyName=iothubowner;SharedAccessKey=...="
|
|
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 tags.site = 'MYRTLE' AND tags.number != '0' AND capabilities.iotEdge = true")
|
|
|
|
query_result = registry_manager.query_iot_hub(query_spec)
|
|
devices = []
|
|
for item in query_result.items:
|
|
devices.append([int(item.tags['number']), item.tags['deviceId'], item.tags['site']])
|
|
|
|
ordered_devices = sorted(devices, key = lambda x: (x[2], x[0]))
|
|
|
|
for i in ordered_devices:
|
|
current_device_modules = registry_manager.get_modules(i[1])
|
|
for module in current_device_modules:
|
|
if module.module_id == module_id:
|
|
thingspro_module = module
|
|
if thingspro_module:
|
|
#print("Found thingspro-agent for " + i[1] + " (" + i[2] + ")")
|
|
try:
|
|
direct_method = CloudToDeviceMethod(method_name=method_name, payload=json.loads(payload))
|
|
response = registry_manager.invoke_device_module_method(device_id=i[1], module_id=module_id, direct_method_request=direct_method)
|
|
#print(response.payload)
|
|
if str(i[1]) == str(response.payload['data']['hostName']):
|
|
print(str(i[2]), str(i[0]), str(i[1]), response.payload['data']['description'], response.payload['data']['firmwareVersion'], sep=";")
|
|
else:
|
|
print(str(i[2]), str(i[0]), str(i[1]), response.payload['data']['description'], response.payload['data']['hostName'], response.payload['data']['firmwareVersion'], sep=";")
|
|
except:
|
|
print(str(i[2]), str(i[0]), str(i[1]), "UNREACHABLE",sep=";")
|
|
else:
|
|
print("No thingspro-agent available for " + i[1] + " (" + i[2] + ")") |