gerrit-to-github-bot/gerrit_to_github_issues/engine.py
Ian H Pittwood 21c88fef51 Prepare project for packaging
Adds setuptools configuration and CLI entry point

Fixes labels list retrieval to use strings
2020-02-27 11:01:02 -06:00

83 lines
4.0 KiB
Python

# Licensed under the Apache License, Version 2.0 (the 'License');
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an 'AS IS' BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
import logging
import github
from github.Repository import Repository
from gerrit_to_github_issues import gerrit
from gerrit_to_github_issues import github_issues
LOG = logging.getLogger(__name__)
def update(gerrit_url: str, gerrit_project_name: str, github_project_name: str, github_user: str, github_password: str,
github_token: str):
repo = github_issues.get_repo(github_project_name, github_user, github_password, github_token)
change_list = gerrit.get_changes(gerrit_url, gerrit_project_name)
for change in change_list['data']:
if 'commitMessage' in change:
process_change(change, repo, gerrit_url)
def process_change(change: dict, repo: Repository, gerrit_url: str):
issue_numbers_dict = github_issues.parse_issue_number(change['commitMessage'])
if not issue_numbers_dict:
LOG.warning(f'No issue tag found for change #{change["number"]}')
return
for key, issues_list in issue_numbers_dict.items():
for issue_number in issues_list:
try:
issue = repo.get_issue(issue_number)
except github.GithubException:
LOG.warning(f'Issue #{issue_number} not found for project')
return
comment_msg = ''
change_url = gerrit.make_gerrit_url(gerrit_url, change['number'])
link_exists = github_issues.check_issue_for_matching_comments(issue, change_url)
if issue.state == 'closed' and not link_exists:
LOG.debug(f'Issue #{issue_number} was closed, reopening...')
#issue.edit(state='open')
#issue.create_comment('Issue reopened due to new activity on Gerrit.\n\n')
labels = [str(l.name) for l in list(issue.get_labels())]
if 'WIP' in change['commitMessage'] or 'DNM' in change['commitMessage']:
if 'wip' not in labels:
LOG.debug(f'add `wip` to #{issue_number}')
#issue.add_to_labels('wip')
if 'ready for review' in labels:
try:
LOG.debug(f'rm `ready for review` to #{issue_number}')
#issue.remove_from_labels('ready for review')
except github.GithubException:
LOG.debug(f'`ready for review` tag does not exist on issue #{issue_number}')
else:
if 'ready for review' not in labels:
LOG.debug(f'add `ready for review` to #{issue_number}')
#issue.add_to_labels('ready for review')
if 'wip' in labels:
try:
LOG.debug(f'rm `wip` to #{issue_number}')
#issue.remove_from_labels('wip')
except github.GithubException:
LOG.debug(f'`wip` tag does not exist on issue #{issue_number}')
if not link_exists:
comment_msg += '### New Related Change\n\n' \
f'**Link:** {change_url}\n' \
f'**Subject:** {change["subject"]}\n' \
f'**Authored By:** {change["owner"]["name"]} ({change["owner"]["email"]})'
if key == 'closes':
comment_msg += '\n\nThis change will close this issue when merged.'
if comment_msg:
LOG.debug(f'Comment to post on #{issue_number}: {comment_msg}')
#issue.create_comment(comment_msg)
LOG.info(f'Comment posted to issue #{issue_number}')