File tree Expand file tree Collapse file tree 2 files changed +12
-10
lines changed
sdk/cosmos/azure-cosmos/azure/cosmos Expand file tree Collapse file tree 2 files changed +12
-10
lines changed Original file line number Diff line number Diff line change @@ -205,11 +205,12 @@ def __init__( # pylint: disable=too-many-statements
205205 credentials_policy = None
206206 if self .aad_credentials :
207207 scope_override = os .environ .get (Constants .AAD_SCOPE_OVERRIDE , "" )
208- if scope_override :
209- scope = scope_override
210- else :
211- scope = base .create_scope_from_url (self .url_connection )
212- credentials_policy = CosmosBearerTokenCredentialPolicy (self .aad_credentials , scope )
208+ account_scope = base .create_scope_from_url (self .url_connection )
209+ credentials_policy = CosmosBearerTokenCredentialPolicy (
210+ self .aad_credentials ,
211+ account_scope = account_scope ,
212+ override_scope = scope_override if scope_override else None
213+ )
213214 self ._enable_diagnostics_logging = kwargs .pop ("enable_diagnostics_logging" , False )
214215 policies = [
215216 HeadersPolicy (** kwargs ),
Original file line number Diff line number Diff line change @@ -213,11 +213,12 @@ def __init__( # pylint: disable=too-many-statements
213213 credentials_policy = None
214214 if self .aad_credentials :
215215 scope_override = os .environ .get (Constants .AAD_SCOPE_OVERRIDE , "" )
216- if scope_override :
217- scope = scope_override
218- else :
219- scope = base .create_scope_from_url (self .url_connection )
220- credentials_policy = AsyncCosmosBearerTokenCredentialPolicy (self .aad_credentials , scope )
216+ account_scope = base .create_scope_from_url (self .url_connection )
217+ credentials_policy = AsyncCosmosBearerTokenCredentialPolicy (
218+ self .aad_credentials ,
219+ account_scope ,
220+ scope_override
221+ )
221222 self ._enable_diagnostics_logging = kwargs .pop ("enable_diagnostics_logging" , False )
222223 policies = [
223224 HeadersPolicy (** kwargs ),
You can’t perform that action at this time.
0 commit comments