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

use the cached Azure Identity instance for storage operations (#526)

Browse Source
This commit is contained in:
bmc-msft
2021-02-09 12:20:12 -05:00
committed by GitHub
parent 91a3690551
commit 8ee7fae240
1 changed files with 2 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
src/api-service/__app__/onefuzzlib/azure/storage.py
View File

@ -9,12 +9,11 @@ import random
from enum import Enum from enum import Enum
from typing import List, Tuple, cast from typing import List, Tuple, cast
from azure.identity import DefaultAzureCredential
from azure.mgmt.storage import StorageManagementClient from azure.mgmt.storage import StorageManagementClient
from memoization import cached from memoization import cached
from msrestazure.tools import parse_resource_id from msrestazure.tools import parse_resource_id
from .creds import get_base_resource_group, get_subscription from .creds import get_base_resource_group, get_identity, get_subscription
class StorageType(Enum): class StorageType(Enum):
@ -25,7 +24,7 @@ class StorageType(Enum):
@cached @cached
def get_mgmt_client() -> StorageManagementClient: def get_mgmt_client() -> StorageManagementClient:
return StorageManagementClient( return StorageManagementClient(
credential=DefaultAzureCredential(), subscription_id=get_subscription() credential=get_identity(), subscription_id=get_subscription()
) )