From d615fccd6dba9690a4016377d83eb0e65f0d3dcb Mon Sep 17 00:00:00 2001
From: Noah McGregor Harper <74685766+nharper285@users.noreply.github.com>
Date: Thu, 25 Aug 2022 15:00:42 -0700
Subject: [PATCH] Fixing set_admins.py Script. (#2300)

* Fixing set_admins.

* Moving set_admins to another directory.

* Fixing imports.
---
 src/deployment/{deploylib => }/set_admins.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)
 rename src/deployment/{deploylib => }/set_admins.py (92%)

diff --git a/src/deployment/deploylib/set_admins.py b/src/deployment/set_admins.py
similarity index 92%
rename from src/deployment/deploylib/set_admins.py
rename to src/deployment/set_admins.py
index 86318a736..e4ed61d47 100644
--- a/src/deployment/deploylib/set_admins.py
+++ b/src/deployment/set_admins.py
@@ -8,7 +8,6 @@ from uuid import UUID
 
 from azure.cosmosdb.table.tableservice import TableService
 from azure.identity import AzureCliCredential
-from azure.mgmt.resource import SubscriptionClient
 from azure.mgmt.storage import StorageManagementClient
 
 from deploylib.configuration import (
@@ -23,12 +22,13 @@ def main() -> None:
     parser = argparse.ArgumentParser(formatter_class=formatter)
     parser.add_argument("resource_group")
     parser.add_argument("storage_account")
+    parser.add_argument("subscription_id")
     parser.add_argument("--admins", type=UUID, nargs="*")
     parser.add_argument("--allowed_aad_tenants", type=UUID, nargs="*")
     args = parser.parse_args()
 
     credential = AzureCliCredential()
-    client = StorageManagementClient(credential)
+    client = StorageManagementClient(credential, args.subscription_id)
     storage_keys = client.storage_accounts.list_keys(
         args.resource_group, args.storage_account
     )