Initial commit

Azure/create-moxa-list.py

@@ -0,0 +1,84 @@
+import pandas as pd
+import subprocess
+import argparse
+import sys
+import json
+
+# This Python script will read an Excel file containing a list of devices for a site (using --serials *file name*)
+# It will then create a IoT Edge device for each of the line in the file
+# The user need to precise which IoT Hub is used (using --env *PROD or DEV*)
+# The user need provide the site name, which will be added as a tag (using --site *site name*)
+# The user need to precise which configuration of the modules should be used, according to the running firmware on the devices (using --version *version number*)
+# Example:
+# python create_devices_list.py --serials DANISH.xlsx --env PROD --site DANISH --version 1.5_patch
+# will create all the IoT Edge devices using the serial number found in the DANISH.xlsx file
+#   on the PROD IoT Hub
+#   tag each new device with a site = DANISH
+#   configure the modules for each devices with the corresponding version found in moxa_ac_template_1.5_patch.json
+#   each device will have also be given a number = i where i is incremented automatically according to the number of device in the Excel file
+#   the provided Excel file will be modified, adding a column connection_string for each of the devices created with their corresponding connection string
+#   the output is therefore SENSITIVE
+
+def generate_commands(serials_file, env, site, version):
+    # Read serial numbers from Excel file
+    df = pd.read_excel(serials_file)
+    df = df[df['device_name'].notnull()]
+    serials = df['device_name'].tolist()
+
+    # Initialize number
+    number = 1
+
+    # List to store connection strings
+    connection_strings = []
+
+    # Generate commands for each serial number
+    for serial in serials:
+        # Replace placeholders with actual values
+        print(serial, end=" ")
+        device_id = f"DIGIT-{serial}"
+        print(device_id, end=" ")
+        tags = f'{{"deviceId":"{device_id}","site":"{site}","number":"{number}"}}'
+        content = f"moxa_ac_template_{version}.json"
+        # Construct command strings
+        create_command = f"az iot hub device-identity create --device-id {device_id} --hub-name IotHub-CUBE-{env} --edge-enabled"
+        twin_update_command = f"az iot hub device-twin update --device-id {device_id} --hub-name IotHub-CUBE-{env} --set tags='{tags}'"
+        set_modules_command = f"az iot edge set-modules --device-id {device_id} --hub-name IotHub-CUBE-{env} --content {content}"
+
+        # Execute create command and get primary key
+        create_output = subprocess.check_output(create_command, shell=True)
+        create_output_json = json.loads(create_output.decode('utf-8'))
+        primary_key = create_output_json['authentication']['symmetricKey']['primaryKey']
+        print(primary_key, end=" ")
+        
+        # Generate connection string
+        connection_string = f"HostName=IotHub-CUBE-{env}.azure-devices.net;DeviceId={device_id};SharedAccessKey={primary_key}"
+        print(connection_string)
+        connection_strings.append(connection_string)
+        
+        tags_output = subprocess.run(twin_update_command, shell=True)
+        modules_output = subprocess.run(set_modules_command, shell=True)
+        # Increment number
+        number += 1
+
+    # Add connection strings to DataFrame
+    df['connection_string'] = connection_strings
+
+    # Save DataFrame to Excel file
+    df.to_excel(serials_file, index=False)
+
+if __name__ == "__main__":
+    # Parse command line arguments
+    parser = argparse.ArgumentParser(description='Create devices list')
+    parser.add_argument('--serials', required=True, help='Excel file containing serial numbers')
+    parser.add_argument('--env', required=True, help='Environment (PROD or DEV)')
+    parser.add_argument('--site', required=True, help='Site name')
+    parser.add_argument('--version', required=True, help='Version number')
+    args = parser.parse_args()
+
+    if len(sys.argv) == 1:
+        parser.print_help()
+        sys.exit(1)
+
+    # Generate commands
+    generate_commands(args.serials, args.env, args.site, args.version)
+