42 lines
1.8 KiB
Python
42 lines
1.8 KiB
Python
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()
|
|
|
|
# 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)
|
|
|
|
SITE_NAME = "Bell"
|
|
|
|
|
|
registry_manager = IoTHubRegistryManager.from_connection_string(CONNECTION_STRING)
|
|
#query_spec = QuerySpecification(query="SELECT * FROM devices WHERE IS_DEFINED(tags.site) AND tags.site = '" + SITE_NAME + "' AND capabilities.iotEdge = true")
|
|
query_spec = QuerySpecification(query="SELECT * FROM devices WHERE IS_DEFINED(tags.site) AND tags.site = '" + SITE_NAME + "' AND capabilities.iotEdge = false")
|
|
|
|
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.get('site')) if item.tags.get('site') else None
|
|
number = int(item.tags.get('number')) if item.tags.get('number') else None
|
|
cloud_version = str(item.tags.get('version')) if item.tags.get('version') else None
|
|
devices.append([deviceId, site, number, cloud_version])
|
|
|
|
ordered_devices = sorted(devices, key = lambda x: (x[1], x[2]))
|
|
|
|
for i in ordered_devices:
|
|
device_info = registry_manager.get_device(i[0])
|
|
primary_key = device_info.authentication.symmetric_key.primary_key
|
|
connection_string = f"HostName={CONNECTION_STRING.split(';')[0].split('=')[1]};DeviceId={i[0]};SharedAccessKey={primary_key}"
|
|
print(i[1], i[2], i[0])
|
|
print(connection_string) |