71 lines
1.8 KiB
Python
71 lines
1.8 KiB
Python
#!/usr/bin/env python
|
|
"""Auth flags for calling BigQuery."""
|
|
|
|
|
|
import os
|
|
from typing import Optional
|
|
|
|
from absl import flags
|
|
|
|
|
|
USE_GCE_SERVICE_ACCOUNT = flags.DEFINE_boolean(
|
|
'use_gce_service_account',
|
|
False,
|
|
'Only for the gcloud wrapper use.'
|
|
)
|
|
CREDENTIAL_FILE = flags.DEFINE_string(
|
|
'credential_file',
|
|
os.path.join(
|
|
os.path.expanduser('~'),
|
|
'.bigquery.v2.token'
|
|
),
|
|
'Only for the gcloud wrapper use.'
|
|
)
|
|
APPLICATION_DEFAULT_CREDENTIAL_FILE: flags.FlagHolder[Optional[str]] = (
|
|
flags.DEFINE_string(
|
|
'application_default_credential_file',
|
|
'',
|
|
'Only for the gcloud wrapper use.'
|
|
)
|
|
)
|
|
SERVICE_ACCOUNT: flags.FlagHolder[Optional[str]] = flags.DEFINE_string(
|
|
'service_account',
|
|
'',
|
|
'Only for the gcloud wrapper use.'
|
|
)
|
|
SERVICE_ACCOUNT_PRIVATE_KEY_FILE: flags.FlagHolder[Optional[str]] = (
|
|
flags.DEFINE_string(
|
|
'service_account_private_key_file',
|
|
'',
|
|
'Only for the gcloud wrapper use.'
|
|
)
|
|
)
|
|
SERVICE_ACCOUNT_PRIVATE_KEY_PASSWORD = flags.DEFINE_string(
|
|
'service_account_private_key_password',
|
|
'notasecret',
|
|
'Only for the gcloud wrapper use.'
|
|
)
|
|
SERVICE_ACCOUNT_CREDENTIAL_FILE = flags.DEFINE_string(
|
|
'service_account_credential_file',
|
|
None,
|
|
'Only for the gcloud wrapper use.'
|
|
)
|
|
OAUTH_ACCESS_TOKEN = flags.DEFINE_string(
|
|
'oauth_access_token',
|
|
'',
|
|
'Only for the gcloud wrapper use.'
|
|
)
|
|
USE_GOOGLE_AUTH = flags.DEFINE_boolean(
|
|
'use_google_auth',
|
|
True,
|
|
'Use new google auth libraries.',
|
|
)
|
|
QUOTA_PROJECT_ID: flags.FlagHolder[Optional[str]] = flags.DEFINE_string(
|
|
'quota_project_id',
|
|
None,
|
|
'ID of a Google Cloud Project as the quota project to be used for billing '
|
|
'and quota limits.',
|
|
allow_hide_cpp=True,
|
|
allow_override=True,
|
|
)
|