def get_auth_manager(self):
if self.auth_type is None:
return AuthManager()
if self.auth_type == BASIC_AUTH:
return BasicAuthManager(
self.username,
self.plaintext_password,
)
if self.auth_type == DIGEST_AUTH:
return DigestAuthManager(
self.username,
self.plaintext_password,
)
if self.auth_type == OAUTH1:
return OAuth1Manager(
client_id=self.client_id,
client_secret=self.plaintext_client_secret,
api_endpoints=self._get_oauth1_api_endpoints(),
connection_settings=self,
)
if self.auth_type == BEARER_AUTH:
return BearerAuthManager(
self.username,
self.plaintext_password,
)
if self.auth_type == OAUTH2_PWD:
return OAuth2PasswordGrantManager(
self.url,
self.username,
self.plaintext_password,
client_id=self.client_id,
client_secret=self.plaintext_client_secret,
api_settings=self._get_oauth2_api_settings(),
connection_settings=self,
)
def get_auth_manager(self):
# Auth types that don't require a username:
if self.auth_type is None:
return AuthManager()
if self.auth_type == OAUTH1:
return OAuth1Manager(
client_id=self.client_id,
client_secret=self.plaintext_client_secret,
api_endpoints=self._get_oauth1_api_endpoints(),
connection_settings=self,
)
if self.auth_type == OAUTH2_CLIENT:
return OAuth2ClientGrantManager(
self.url,
client_id=self.client_id,
client_secret=self.plaintext_client_secret,
token_url=self.token_url,
refresh_url=self.refresh_url,
connection_settings=self,
)
# Auth types that require a username:
if not isinstance(self.username, str):
return AuthManager()
if self.auth_type == BASIC_AUTH:
return BasicAuthManager(
self.username,
self.plaintext_password,
)
if self.auth_type == DIGEST_AUTH:
return DigestAuthManager(
self.username,
self.plaintext_password,
)
if self.auth_type == BEARER_AUTH:
return BearerAuthManager(
self.username,
self.plaintext_password,
)
if self.auth_type == OAUTH2_PWD:
return OAuth2PasswordGrantManager(
self.url,
self.username,
self.plaintext_password,
client_id=self.client_id,
client_secret=self.plaintext_client_secret,
token_url=self.token_url,
refresh_url=self.refresh_url,
pass_credentials_in_header=self.pass_credentials_in_header,
connection_settings=self,
)
raise ValueError(f'Unknown auth type {self.auth_type!r}')
def get_auth_manager(self):
if self.auth_type is None:
return AuthManager()
if self.auth_type == BASIC_AUTH:
return BasicAuthManager(
self.username,
self.plaintext_password,
)
if self.auth_type == DIGEST_AUTH:
return DigestAuthManager(
self.username,
self.plaintext_password,
)
if self.auth_type == OAUTH1:
raise NotImplementedError(
_('OAuth1 authentication workflow not yet supported.'))
if self.auth_type == BEARER_AUTH:
return BearerAuthManager(
self.username,
self.plaintext_password,
)
def test_digest_auth(self):
auth_manager = DigestAuthManager(USERNAME, PASSWORD)
auth = auth_manager.get_auth()
self.assertEqual(auth.__class__.__name__, 'HTTPDigestAuth')