Class KaggleKernelCredentials

patches/kaggle_gcp.py:48–78

Custom Credentials used to authenticate using the Kernel's connected OAuth account. Example usage: client = bigquery.Client(project='ANOTHER_PROJECT', credentials=KaggleKernelCredentials())

            GcpTarget.AUTOML in self.integrations

class KaggleKernelCredentials(credentials.Credentials):
    """Custom Credentials used to authenticate using the Kernel's connected OAuth account.
    Example usage:
    client = bigquery.Client(project='ANOTHER_PROJECT',
                                credentials=KaggleKernelCredentials())
    """
    def __init__(self, target=GcpTarget.BIGQUERY):
        super().__init__()
        self.target = target

    def refresh(self, request):
        try:
            client = UserSecretsClient()
            if self.target == GcpTarget.BIGQUERY:
                self.token, self.expiry = client.get_bigquery_access_token()
            elif self.target == GcpTarget.GCS:
                self.token, self.expiry = client._get_gcs_access_token()
            elif self.target == GcpTarget.CLOUDAI:
                self.token, self.expiry = client._get_cloudai_access_token()
        except ConnectionError as e:
            logging.error(f"Connection error trying to refresh access token: {e}")
            print("There was a connection error trying to fetch the access token. "
                  f"Please ensure internet is on in order to use the {self.target.service} Integration.")
            raise RefreshError('Unable to refresh access token due to connection error.') from e
        except Exception as e:
            logging.error(f"Error trying to refresh access token: {e}")
            if (not get_integrations().has_integration(self.target)):
                logging.error(f"No {self.target.service} integration found.")
                print(
                   f"Please ensure you have selected a {self.target.service} account in the Notebook Add-ons menu.")
            raise RefreshError('Unable to refresh access token.') from e

class KaggleKernelWithProjetCredentials(KaggleKernelCredentials):
    """ Wrapper Kaggle Credentials with quota_project_id.