utils.py
111 lines
| 3.8 KiB
| text/x-python
|
PythonLexer
r5088 | # Copyright (C) 2016-2023 RhodeCode GmbH | |||
r1769 | # | |||
# This program is free software: you can redistribute it and/or modify | ||||
# it under the terms of the GNU Affero General Public License, version 3 | ||||
# (only), as published by the Free Software Foundation. | ||||
# | ||||
# This program is distributed in the hope that it will be useful, | ||||
# but WITHOUT ANY WARRANTY; without even the implied warranty of | ||||
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | ||||
# GNU General Public License for more details. | ||||
# | ||||
# You should have received a copy of the GNU Affero General Public License | ||||
# along with this program. If not, see <http://www.gnu.org/licenses/>. | ||||
# | ||||
# This program is dual-licensed. If you wish to learn more about the | ||||
# RhodeCode Enterprise Edition, including its added features, Support services, | ||||
# and proprietary license terms, please see https://rhodecode.com/licenses/ | ||||
r4509 | from rhodecode.lib import helpers as h, rc_cache | |||
r1769 | from rhodecode.lib.utils2 import safe_int | |||
r4346 | from rhodecode.model.pull_request import get_diff_info | |||
r4500 | from rhodecode.model.db import PullRequestReviewers | |||
# V3 - Reviewers, with default rules data | ||||
# v4 - Added observers metadata | ||||
r4559 | # v5 - pr_author/commit_author include/exclude logic | |||
REVIEWER_API_VERSION = 'V5' | ||||
r1769 | ||||
r4500 | def reviewer_as_json(user, reasons=None, role=None, mandatory=False, rules=None, user_group=None): | |||
r1769 | """ | |||
Returns json struct of a reviewer for frontend | ||||
:param user: the reviewer | ||||
:param reasons: list of strings of why they are reviewers | ||||
:param mandatory: bool, to set user as mandatory | ||||
""" | ||||
r4500 | role = role or PullRequestReviewers.ROLE_REVIEWER | |||
if role not in PullRequestReviewers.ROLES: | ||||
raise ValueError('role is not one of %s', PullRequestReviewers.ROLES) | ||||
r1769 | ||||
return { | ||||
'user_id': user.user_id, | ||||
r1789 | 'reasons': reasons or [], | |||
r2484 | 'rules': rules or [], | |||
r4500 | 'role': role, | |||
r1769 | 'mandatory': mandatory, | |||
r2484 | 'user_group': user_group, | |||
r1769 | 'username': user.username, | |||
r1815 | 'first_name': user.first_name, | |||
'last_name': user.last_name, | ||||
r2484 | 'user_link': h.link_to_user(user), | |||
r1769 | 'gravatar_link': h.gravatar_url(user.email, 14), | |||
} | ||||
r4509 | def to_reviewers(e): | |||
if isinstance(e, (tuple, list)): | ||||
r5119 | return list(map(reviewer_as_json, e)) | |||
r4509 | else: | |||
return reviewer_as_json(e) | ||||
def get_default_reviewers_data(current_user, source_repo, source_ref, target_repo, target_ref, | ||||
include_diff_info=True): | ||||
r4346 | """ | |||
Return json for default reviewers of a repository | ||||
""" | ||||
r1769 | ||||
r4509 | diff_info = {} | |||
if include_diff_info: | ||||
diff_info = get_diff_info( | ||||
source_repo, source_ref.commit_id, target_repo, target_ref.commit_id) | ||||
r1769 | ||||
reasons = ['Default reviewer', 'Repository owner'] | ||||
r3230 | json_reviewers = [reviewer_as_json( | |||
r4500 | user=target_repo.user, reasons=reasons, mandatory=False, rules=None, role=None)] | |||
r1769 | ||||
r4509 | compute_key = rc_cache.utils.compute_key_from_params( | |||
current_user.user_id, source_repo.repo_id, source_ref.type, source_ref.name, | ||||
source_ref.commit_id, target_repo.repo_id, target_ref.type, target_ref.name, | ||||
target_ref.commit_id) | ||||
r1769 | return { | |||
r4346 | 'api_ver': REVIEWER_API_VERSION, # define version for later possible schema upgrade | |||
r4509 | 'compute_key': compute_key, | |||
r4346 | 'diff_info': diff_info, | |||
r3230 | 'reviewers': json_reviewers, | |||
r1769 | 'rules': {}, | |||
'rules_data': {}, | ||||
r4559 | 'rules_humanized': [], | |||
r1769 | } | |||
def validate_default_reviewers(review_members, reviewer_rules): | ||||
""" | ||||
Function to validate submitted reviewers against the saved rules | ||||
""" | ||||
reviewers = [] | ||||
reviewer_by_id = {} | ||||
for r in review_members: | ||||
reviewer_user_id = safe_int(r['user_id']) | ||||
r4500 | entry = (reviewer_user_id, r['reasons'], r['mandatory'], r['role'], r['rules']) | |||
r1769 | ||||
reviewer_by_id[reviewer_user_id] = entry | ||||
reviewers.append(entry) | ||||
return reviewers | ||||
r4500 | ||||
r4516 | def validate_observers(observer_members, reviewer_rules): | |||
r4500 | return {} | |||