chore: Update KEYVAULT_URL in dotnet and python examples (#573)

* Update KEYVAULT_URL in dotnet and python examples * chore: update example code
Azure · Sep 28, 2022 · a00ef76 · a00ef76
1 parent 83fbae3
commit a00ef76
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 7 deletions.
diff --git a/examples/msal-net/akvdotnet/Program.cs b/examples/msal-net/akvdotnet/Program.cs
@@ -11,11 +11,13 @@ public class Program
         static void Main(string[] args)
         {
             Program P = new Program();
-            string keyvaultName = Environment.GetEnvironmentVariable("KEYVAULT_NAME");
-            string secretName = Environment.GetEnvironmentVariable("SECRET_NAME");
+            string keyvaultURL = Environment.GetEnvironmentVariable("KEYVAULT_URL");
+            if (string.IsNullOrEmpty(keyvaultURL)) {
+                string keyvaultName = Environment.GetEnvironmentVariable("KEYVAULT_NAME");
+                keyvaultURL = "https://" + keyvaultName + ".vault.azure.net/";
+            }
 
-            // keyvault URL
-            string keyvaultURL = "https://" + keyvaultName + ".vault.azure.net/";
+            string secretName = Environment.GetEnvironmentVariable("SECRET_NAME");
 
             SecretClient client = new SecretClient(
                 new Uri(keyvaultURL),
@@ -25,7 +27,7 @@ static void Main(string[] args)
             {
                 Console.WriteLine($"{Environment.NewLine}START {DateTime.UtcNow} ({Environment.MachineName})");
 
-                // <getsecret>                
+                // <getsecret>
                 var keyvaultSecret = client.GetSecret(secretName).Value;
                 Console.WriteLine("Your secret is " + keyvaultSecret.Value);
 

diff --git a/examples/msal-python/main.py b/examples/msal-python/main.py
@@ -14,11 +14,14 @@ def main():
     # create a token credential object, which has a get_token method that returns a token
     token_credential = MyClientAssertionCredential(azure_client_id, azure_tenant_id, azure_authority_host, azure_federated_token_file)
 
-    keyvault_name = os.getenv('KEYVAULT_NAME', '')
+    keyvault_url = os.getenv('KEYVAULT_URL', '')
+    if not keyvault_url:
+        keyvault_name = os.getenv('KEYVAULT_NAME', '')
+        keyvault_url='https://{}.vault.azure.net'.format(keyvault_name)
     secret_name = os.getenv('SECRET_NAME', '')
 
     # create a secret client with the token credential
-    keyvault = SecretClient(vault_url='https://{}.vault.azure.net'.format(keyvault_name), credential=token_credential)
+    keyvault = SecretClient(vault_url=keyvault_url, credential=token_credential)
     secret = keyvault.get_secret(secret_name)
     print('successfully got secret, secret={}'.format(secret.value))