ExternalVendorCode/onefuzz
1
0
Fork 0
mirror of https://github.com/microsoft/onefuzz.git synced 2025-06-19 04:58:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fixing set_admins.py Script. (#2300)

Browse Source 
* Fixing set_admins.

* Moving set_admins to another directory.

* Fixing imports.
This commit is contained in:
Noah McGregor Harper
2022-08-25 15:00:42 -07:00
committed by GitHub
parent c7e7fa1d27
commit d615fccd6d
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/deployment/deploylib/set_admins.py → src/deployment/set_admins.py
View File

@ -8,7 +8,6 @@ from uuid import UUID
from azure.cosmosdb.table.tableservice import TableService from azure.cosmosdb.table.tableservice import TableService
from azure.identity import AzureCliCredential from azure.identity import AzureCliCredential
from azure.mgmt.resource import SubscriptionClient
from azure.mgmt.storage import StorageManagementClient from azure.mgmt.storage import StorageManagementClient
from deploylib.configuration import ( from deploylib.configuration import (
@ -23,12 +22,13 @@ def main() -> None:
parser = argparse.ArgumentParser(formatter_class=formatter) parser = argparse.ArgumentParser(formatter_class=formatter)
parser.add_argument("resource_group") parser.add_argument("resource_group")
parser.add_argument("storage_account") parser.add_argument("storage_account")
parser.add_argument("subscription_id")
parser.add_argument("--admins", type=UUID, nargs="*") parser.add_argument("--admins", type=UUID, nargs="*")
parser.add_argument("--allowed_aad_tenants", type=UUID, nargs="*") parser.add_argument("--allowed_aad_tenants", type=UUID, nargs="*")
args = parser.parse_args() args = parser.parse_args()
credential = AzureCliCredential() credential = AzureCliCredential()
client = StorageManagementClient(credential) client = StorageManagementClient(credential, args.subscription_id)
storage_keys = client.storage_accounts.list_keys( storage_keys = client.storage_accounts.list_keys(
args.resource_group, args.storage_account args.resource_group, args.storage_account
) )