From f31308a903157ec69579c6c3d1dfb82cd9afb62f Mon Sep 17 00:00:00 2001 From: Andrew Brain Date: Thu, 11 Jan 2024 21:26:28 -0600 Subject: [PATCH] Fix collection issue Signed-off-by: Andrew Brain --- augur/application/db/data_parse.py | 2 +- augur/tasks/gitlab/merge_request_task.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/augur/application/db/data_parse.py b/augur/application/db/data_parse.py index 3127086d4..2d5c51a89 100644 --- a/augur/application/db/data_parse.py +++ b/augur/application/db/data_parse.py @@ -733,7 +733,7 @@ def extract_gitlab_issue_event_data(event: dict, issue_id: int, platform_id: int # retrieve only the needed data for pr reviewers from the api response -def extract_needed_pr_reviewer_data(data: List[dict], pull_request_id, repo_id: int, tool_source: str, tool_version: str, data_source: str) -> List[dict]: +def extract_needed_mr_reviewer_data(data: List[dict], pull_request_id, repo_id: int, tool_source: str, tool_version: str, data_source: str) -> List[dict]: if len(data) == 0: return [] diff --git a/augur/tasks/gitlab/merge_request_task.py b/augur/tasks/gitlab/merge_request_task.py index bb27ecdaf..5672a7989 100644 --- a/augur/tasks/gitlab/merge_request_task.py +++ b/augur/tasks/gitlab/merge_request_task.py @@ -4,7 +4,7 @@ from augur.tasks.init.celery_app import AugurCoreRepoCollectionTask from augur.tasks.gitlab.gitlab_api_handler import GitlabApiHandler from augur.tasks.gitlab.gitlab_task_session import GitlabTaskManifest -from augur.application.db.data_parse import extract_needed_pr_data_from_gitlab_merge_request, extract_needed_merge_request_assignee_data, extract_needed_mr_label_data, extract_needed_pr_reviewer_data, extract_needed_mr_commit_data, extract_needed_mr_file_data, extract_needed_mr_metadata, extract_needed_gitlab_mr_message_ref_data, extract_needed_gitlab_message_data +from augur.application.db.data_parse import extract_needed_pr_data_from_gitlab_merge_request, extract_needed_merge_request_assignee_data, extract_needed_mr_label_data, extract_needed_mr_reviewer_data, extract_needed_mr_commit_data, extract_needed_mr_file_data, extract_needed_mr_metadata, extract_needed_gitlab_mr_message_ref_data, extract_needed_gitlab_message_data from augur.tasks.github.util.util import get_owner_repo, add_key_value_pair_to_dicts from augur.application.db.models import PullRequest, PullRequestAssignee, PullRequestLabel, PullRequestReviewer, PullRequestMeta, PullRequestCommit, PullRequestFile, PullRequestMessageRef, Repo, Message from augur.application.db.util import execute_session_query @@ -299,7 +299,7 @@ def process_mr_reviewers(data, task_name, repo_id, logger, augur_db): pull_request_id = mr_number_to_id_map[id] - reviewers = extract_needed_pr_reviewer_data(values, pull_request_id, repo_id, tool_source, tool_version, data_source) + reviewers = extract_needed_mr_reviewer_data(values, pull_request_id, repo_id, tool_source, tool_version, data_source) all_reviewers += reviewers