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 )