def scheduled_updates():
"""Endpoint for updating imports."""
task_info = flask.request.get_json(force=True)
if 'absolute_import_name' not in task_info:
return {'error': 'absolute_import_name not found'}
task_configs = task_info.get('configs', {})
config = configs.ExecutorConfig(**task_configs)
executor = import_executor.ImportExecutor(
uploader=file_uploader.GCSFileUploader(
project_id=config.gcs_project_id,
bucket_name=config.storage_prod_bucket_name),
github=github_api.GitHubRepoAPI(
repo_owner_username=config.github_repo_owner_username,
repo_name=config.github_repo_name,
auth_username=config.github_auth_username,
auth_access_token=config.github_auth_access_token),
dashboard=dashboard_api.DashboardAPI(config.dashboard_oauth_client_id),
config=config)
result = executor.execute_imports_on_update(
task_info['absolute_import_name'])
return dataclasses.asdict(result)
def execute_imports():
"""Endpoint for executing imports on GitHub commits.
Logs to the import progress dashboard.
"""
task_info = flask.request.get_json(force=True)
if 'COMMIT_SHA' not in task_info:
return {'error': 'COMMIT_SHA not found'}
commit_sha = task_info['COMMIT_SHA']
repo_name = task_info.get('REPO_NAME')
branch_name = task_info.get('BRANCH_NAME')
pr_number = task_info.get('PR_NUMBER')
task_configs = task_info.get('configs', {})
config = configs.ExecutorConfig(**task_configs)
executor = import_executor.ImportExecutor(
uploader=file_uploader.GCSFileUploader(
project_id=config.gcs_project_id,
bucket_name=config.storage_dev_bucket_name,
path_prefix=config.storage_executor_output_prefix),
github=github_api.GitHubRepoAPI(
repo_owner_username=config.github_repo_owner_username,
repo_name=config.github_repo_name,
auth_username=config.github_auth_username,
auth_access_token=config.github_auth_access_token),
config=config,
dashboard=dashboard_api.DashboardAPI(config.dashboard_oauth_client_id),
notifier=email_notifier.EmailNotifier(config.email_account,
config.email_token),
importer=import_service.ImportServiceClient(
project_id=config.gcs_project_id,
executor_output_prefix=config.storage_executor_output_prefix,
importer_output_prefix=config.storage_importer_output_prefix,
unresolved_mcf_bucket_name=config.storage_dev_bucket_name,
resolved_mcf_bucket_name=config.storage_importer_bucket_name))
result = executor.execute_imports_on_commit(commit_sha=commit_sha,
repo_name=repo_name,
branch_name=branch_name,
pr_number=pr_number)
return dataclasses.asdict(result)
def setUp(self, _):
self.io = file_uploader.GCSFileUploader(project_id='project-id',
bucket_name='bucket-name')