ExternalVendorCode/onefuzz
1
0
Fork 0
mirror of https://github.com/microsoft/onefuzz.git synced 2025-06-16 20:08:09 +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 typing import List, Tuple, cast
from azure.identity import DefaultAzureCredential
from azure.mgmt.storage import StorageManagementClient
from memoization import cached
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):
@ -25,7 +24,7 @@ class StorageType(Enum):
@cached
def get_mgmt_client() -> StorageManagementClient:
return StorageManagementClient(
credential=DefaultAzureCredential(), subscription_id=get_subscription()
credential=get_identity(), subscription_id=get_subscription()
)