def __init__(self, **kwargs):
logging.getLogger(__name__).debug(f'Connection params: {kwargs}')
super().__init__(
**{k: v
for k, v in kwargs.items() if k != 'secrets_keeper'})
logging.getLogger(__name__).debug(
f'Init: {self.client_id} - {self.client_secret}')
self.authorization_url = (
f'https://login.microsoftonline.com/{self.tenant}/oauth2/v2.0/authorize'
)
self.token_url = f'https://login.microsoftonline.com/{self.tenant}/oauth2/v2.0/token'
# we use __dict__ so that pydantic does not complain about the _oauth2_connector field
self._oauth2_connector = OAuth2Connector(
auth_flow_id=self.auth_flow_id,
authorization_url=self.authorization_url,
scope=self.scope,
token_url=self.token_url,
redirect_uri=self.redirect_uri,
config=OAuth2ConnectorConfig(
client_id=self.client_id,
client_secret=self.client_secret,
),
secrets_keeper=kwargs['secrets_keeper'],
)
def oauth2_connector(secrets_keeper):
return OAuth2Connector(
auth_flow_id='test',
authorization_url=FAKE_AUTHORIZATION_URL,
scope=SCOPE,
config=OAuth2ConnectorConfig(client_id='', client_secret=''),
redirect_uri='',
token_url=FAKE_TOKEN_URL,
secrets_keeper=secrets_keeper,
)
def __init__(self, **kwargs):
super().__init__(
**{k: v for k, v in kwargs.items() if k not in OAuth2Connector.init_params}
)
# we use __dict__ so that pydantic does not complain about the _oauth2_connector field
self._oauth2_connector = OAuth2Connector(
auth_flow_id=self.auth_flow_id,
authorization_url=AUTHORIZATION_URL,
scope=SCOPE,
token_url=TOKEN_URL,
redirect_uri=kwargs['redirect_uri'],
config=OAuth2ConnectorConfig(
client_id=kwargs['client_id'],
client_secret=kwargs['client_secret'],
),
secrets_keeper=kwargs['secrets_keeper'],
)
def __init__(self, **kwargs):
super().__init__(
**{k: v for k, v in kwargs.items() if k not in OAuth2Connector.init_params}
)
self._oauth2_connector = OAuth2Connector(
auth_flow_id=self.auth_flow_id,
authorization_url=AUTHORIZATION_URL,
scope=SCOPE,
token_url=TOKEN_URL,
secrets_keeper=kwargs['secrets_keeper'],
redirect_uri=kwargs['redirect_uri'],
config=OAuth2ConnectorConfig(
client_id=kwargs['client_id'],
client_secret=kwargs['client_secret'],
),
)
self.provided_token = kwargs.get('provided_token', None)
def __init__(self, **kwargs):
super().__init__(
**{k: v
for k, v in kwargs.items() if k != 'secrets_keeper'})
self.token_url = f'https://{self.account}.snowflakecomputing.com/oauth/token-request'
self.authorization_url = f'https://{self.account}.snowflakecomputing.com/oauth/authorize'
self._oauth2_connector = OAuth2Connector(
auth_flow_id=self.auth_flow_id,
authorization_url=self.authorization_url,
scope=self.scope,
token_url=self.token_url,
redirect_uri=self.redirect_uri,
config=OAuth2ConnectorConfig(
client_id=self.client_id,
client_secret=self.client_secret,
),
secrets_keeper=kwargs['secrets_keeper'],
)
def __init__(self, **kwargs):
super().__init__(
**{
k: v
for k, v in kwargs.items()
if k not in OAuth2Connector.init_params
})
self._oauth2_connector = OAuth2Connector(
auth_flow_id=self.auth_flow_id,
authorization_url=AUTHORIZATION_URL_SANDBOX
if self.type == 'SalesforceSandbox' else AUTHORIZATION_URL_PROD,
scope=SCOPE,
token_url=TOKEN_URL_SANDBOX
if self.type == 'SalesforceSandbox' else TOKEN_URL_PROD,
secrets_keeper=kwargs['secrets_keeper'],
redirect_uri=kwargs['redirect_uri'],
config=OAuth2ConnectorConfig(
client_id=kwargs['client_id'],
client_secret=kwargs['client_secret'],
),
)
def get_connector_secrets_form() -> ConnectorSecretsForm:
return ConnectorSecretsForm(
documentation_md=(Path(os.path.dirname(__file__)) /
'doc.md').read_text(),
secrets_schema=OAuth2ConnectorConfig.schema(),
)
def test_should_return_connector_config_form():
assert (get_connector_secrets_form(GoogleSheets2Connector).secrets_schema
== OAuth2ConnectorConfig.schema())
assert get_connector_secrets_form(MongoConnector) is None
