From 94282e6c82774055b4906915c8a52897c0096053 Mon Sep 17 00:00:00 2001
From: bretmulvey <51333626+bretmulvey@users.noreply.github.com>
Date: Wed, 10 Nov 2021 15:17:56 -0800
Subject: [PATCH] Updated to Python sample for IoT Central 1.0.0b1

Updated the Python SDK sample to work with last IoT Central management SDK. Included the same functionality as the previous sample, and added examples of new functionality including

- enabling system-assigned identity
- regional instead of geographic scope
- S1 SKU unavailable
---
 python/run.py | 148 +++++++++++++++++++++++++++++++++++++++-----------
 1 file changed, 116 insertions(+), 32 deletions(-)

diff --git a/python/run.py b/python/run.py
index ba68ebd..52449c1 100644
--- a/python/run.py
+++ b/python/run.py
@@ -1,49 +1,133 @@
-import os, uuid, sys
+
+import os
+import random
+
+from azure.identity._credentials.browser import InteractiveBrowserCredential
 from azure.mgmt.iotcentral import IotCentralClient
-from azure.mgmt.iotcentral.models import App, AppSkuInfo, AppPatch
-from msrestazure.azure_active_directory import MSIAuthentication
-from azure.common.credentials import UserPassCredentials, get_azure_cli_credentials
-
-# login with az login
-creds = get_azure_cli_credentials()
-subId = "FILL IN SUB ID"
-appName = "iot-central-app-tocreate"
+from azure.mgmt.iotcentral.models import App, AppSkuInfo, SystemAssignedServiceIdentityType
+
+region = "westus"
 resourceGroup = "myResourceGroup"
+skuName = "ST2"
+
+
+# Get subscription info from environment.
+
+tenantId = os.environ["AZURE_TENANT_ID"]
+subscriptionId = os.environ["AZURE_SUBSCRIPTION_ID"]
+
+if (tenantId is None) or (subscriptionId is None):
+    raise Exception("Expected environment variables.")
+
+
+# Make IOTC client.
+
+credential = InteractiveBrowserCredential(tenant_id = tenantId)
+iotc = IotCentralClient(credential, subscriptionId)
 
-client = IotCentralClient(creds[0], subId)
 
-result = client.apps.check_name_availability(appName)
-print(result)
+# Choose app name.
 
-app = App(location="unitedstates", sku=AppSkuInfo(name="ST2"))
+appName = f"pysdk-{random.randint(100000, 999999)}-{random.randint(100000, 999999)}"
+
+
+# Define the app configuration.
+
+app = App(location = region, sku = AppSkuInfo(name = skuName))
 app.subdomain = appName
 app.display_name = appName
+app.identity = { "type": "SystemAssigned" }
 
-client.apps.create_or_update(resourceGroup, appName, app)
 
-getResult = client.apps.get(resourceGroup, appName)
-print(getResult)
+# Create the app.
 
-updateApp = AppPatch()
-updateApp.display_name = appName + "-new-name"
+print(f"\nCreating {appName}. Check browser window for login.")
+poller = iotc.apps.begin_create_or_update(resourceGroup, appName, app)
+result = poller.result()
 
-client.apps.update(resourceGroup, appName, updateApp)
+if result.state != "created":
+    raise Exception("Expected 'created' state.")
 
-appsInGroup = client.apps.list_by_resource_group(resourceGroup)
-appsInGroup.next()
-for item in appsInGroup.current_page:
-    print(item)
+print("OK")
+
+
+# Make sure it's idempotent.
+
+print(f"\nUpdating {appName}.")
+poller = iotc.apps.begin_create_or_update(resourceGroup, appName, app)
+result = poller.result()
+
+if result.state != "created":
+    raise Exception("Expected 'created' state.")
+
+print("OK")
 
-operations = client.operations.list()
-operations.next()
-for item in operations.current_page:
-    print(item)
 
-appTemplates = client.apps.list_templates()
-for item in appTemplates:
+# List all the apps in the resource group.
+
+print(f"\nListing IoT Central apps in '{resourceGroup}'")
+appsInGroup = iotc.apps.list_by_resource_group(resourceGroup)
+
+for item in appsInGroup:
     print(item)
 
-# deleteResult = client.apps.delete(resourceGroup, appName)
-# print(deleteResult)
 
-print("done")
+# Update the app tags.
+
+print(f"\nUpdating {appName} tags.")
+tag = "mytag"
+value = "myvalue"
+app.tags = { tag: value }
+poller = iotc.apps.begin_create_or_update(resourceGroup, appName, app)
+result = poller.result()
+
+if result.tags[tag] != value:
+    raise Exception("Expected updated tag.")
+
+print("OK")
+
+
+# Delete the app.
+
+print(f"\nDeleting {appName}")
+poller = iotc.apps.begin_delete(resourceGroup, appName)
+result = poller.result()
+
+if result:
+    print(result)
+    raise Exception("Expected 'None'.")
+
+print("OK")
+
+
+# Verify that we can't create in geography as we could before.
+
+print("\nMake sure we can't use geography.")
+app.location = "unitedstates"
+appName = "us-" + appName
+app.subdomain = app.display_name = appName
+
+try:
+    poller = iotc.apps.begin_create_or_update(resourceGroup, appName, app)
+    result = poller.result()
+    print("It worked but it shouldn't have!")
+
+except:
+    print("OK")
+
+
+# Verify that S1 SKU is no longer allowed.
+
+print(f"\nMake sure we can't use S1 SKU.")
+appNameS1 = "s1-" + appName
+app = App(location = region, sku = AppSkuInfo(name = "S1"))
+app.subdomain = appNameS1
+app.display_name = appNameS1
+
+try:
+    poller = iotc.apps.begin_create_or_update(resourceGroup, appNameS1, app)
+    result = poller.result()
+    print("It worked but it shouldn't have!")
+
+except:
+    print("OK")