pull_request.py
258 lines
| 9.6 KiB
| text/x-python
|
PythonLexer
r2434 | # -*- coding: utf-8 -*- | |||
""" | ||||
r2608 | rhodecode.model.pull_request | |||
~~~~~~~~~~~~~~~~~~~~~~~~~~~~ | ||||
r2434 | ||||
pull request model for RhodeCode | ||||
:created_on: Jun 6, 2012 | ||||
:author: marcink | ||||
:copyright: (C) 2012-2012 Marcin Kuzminski <marcin@python-works.com> | ||||
:license: GPLv3, see COPYING for more details. | ||||
""" | ||||
# This program is free software: you can redistribute it and/or modify | ||||
# it under the terms of the GNU General Public License as published by | ||||
# the Free Software Foundation, either version 3 of the License, or | ||||
# (at your option) any later version. | ||||
# | ||||
# 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 General Public License | ||||
# along with this program. If not, see <http://www.gnu.org/licenses/>. | ||||
import logging | ||||
r2608 | import datetime | |||
r3010 | import re | |||
r2597 | ||||
r2434 | from pylons.i18n.translation import _ | |||
r2597 | from rhodecode.model.meta import Session | |||
|
r3303 | from rhodecode.lib import helpers as h, unionrepo | ||
r2434 | from rhodecode.model import BaseModel | |||
r3175 | from rhodecode.model.db import PullRequest, PullRequestReviewers, Notification,\ | |||
ChangesetStatus | ||||
r2434 | from rhodecode.model.notification import NotificationModel | |||
from rhodecode.lib.utils2 import safe_unicode | ||||
|
r3303 | from rhodecode.lib.vcs.utils.hgcompat import scmutil | ||
|
r3301 | from rhodecode.lib.vcs.utils import safe_str | ||
r3387 | from rhodecode.lib.vcs.backends.base import EmptyChangeset | |||
r2442 | ||||
r2434 | log = logging.getLogger(__name__) | |||
class PullRequestModel(BaseModel): | ||||
r2522 | cls = PullRequest | |||
r2608 | def __get_pull_request(self, pull_request): | |||
return self._get_instance(PullRequest, pull_request) | ||||
r2440 | def get_all(self, repo): | |||
repo = self._get_repo(repo) | ||||
r3262 | return PullRequest.query()\ | |||
.filter(PullRequest.other_repo == repo)\ | ||||
r3389 | .order_by(PullRequest.created_on.desc())\ | |||
r3262 | .all() | |||
r2440 | ||||
r3175 | def create(self, created_by, org_repo, org_ref, other_repo, other_ref, | |||
revisions, reviewers, title, description=None): | ||||
from rhodecode.model.changeset_status import ChangesetStatusModel | ||||
r2541 | ||||
r2440 | created_by_user = self._get_user(created_by) | |||
r2541 | org_repo = self._get_repo(org_repo) | |||
other_repo = self._get_repo(other_repo) | ||||
r2434 | ||||
new = PullRequest() | ||||
r2541 | new.org_repo = org_repo | |||
r2434 | new.org_ref = org_ref | |||
r2541 | new.other_repo = other_repo | |||
r2434 | new.other_ref = other_ref | |||
new.revisions = revisions | ||||
new.title = title | ||||
new.description = description | ||||
r2440 | new.author = created_by_user | |||
r2434 | self.sa.add(new) | |||
r2541 | Session().flush() | |||
r2434 | #members | |||
r3389 | for member in set(reviewers): | |||
r2434 | _usr = self._get_user(member) | |||
reviewer = PullRequestReviewers(_usr, new) | ||||
self.sa.add(reviewer) | ||||
r3175 | #reset state to under-review | |||
ChangesetStatusModel().set_status( | ||||
repo=org_repo, | ||||
status=ChangesetStatus.STATUS_UNDER_REVIEW, | ||||
user=created_by_user, | ||||
pull_request=new | ||||
) | ||||
r2434 | #notification to reviewers | |||
notif = NotificationModel() | ||||
r2440 | ||||
r2799 | pr_url = h.url('pullrequest_show', repo_name=other_repo.repo_name, | |||
pull_request_id=new.pull_request_id, | ||||
qualified=True, | ||||
) | ||||
r2434 | subject = safe_unicode( | |||
h.link_to( | ||||
|
r3251 | _('%(user)s wants you to review pull request #%(pr_id)s: %(pr_title)s') % \ | ||
r2434 | {'user': created_by_user.username, | |||
|
r3251 | 'pr_title': new.title, | ||
r2434 | 'pr_id': new.pull_request_id}, | |||
r2799 | pr_url | |||
r2434 | ) | |||
) | ||||
body = description | ||||
r2799 | kwargs = { | |||
'pr_title': title, | ||||
'pr_user_created': h.person(created_by_user.email), | ||||
'pr_repo_url': h.url('summary_home', repo_name=other_repo.repo_name, | ||||
qualified=True,), | ||||
'pr_url': pr_url, | ||||
'pr_revisions': revisions | ||||
} | ||||
r3389 | ||||
r2541 | notif.create(created_by=created_by_user, subject=subject, body=body, | |||
r2434 | recipients=reviewers, | |||
r2799 | type_=Notification.TYPE_PULL_REQUEST, email_kwargs=kwargs) | |||
r2434 | return new | |||
r2442 | ||||
r2614 | def update_reviewers(self, pull_request, reviewers_ids): | |||
reviewers_ids = set(reviewers_ids) | ||||
pull_request = self.__get_pull_request(pull_request) | ||||
current_reviewers = PullRequestReviewers.query()\ | ||||
.filter(PullRequestReviewers.pull_request== | ||||
pull_request)\ | ||||
.all() | ||||
current_reviewers_ids = set([x.user.user_id for x in current_reviewers]) | ||||
to_add = reviewers_ids.difference(current_reviewers_ids) | ||||
to_remove = current_reviewers_ids.difference(reviewers_ids) | ||||
log.debug("Adding %s reviewers" % to_add) | ||||
log.debug("Removing %s reviewers" % to_remove) | ||||
for uid in to_add: | ||||
_usr = self._get_user(uid) | ||||
reviewer = PullRequestReviewers(_usr, pull_request) | ||||
self.sa.add(reviewer) | ||||
for uid in to_remove: | ||||
reviewer = PullRequestReviewers.query()\ | ||||
.filter(PullRequestReviewers.user_id==uid, | ||||
PullRequestReviewers.pull_request==pull_request)\ | ||||
.scalar() | ||||
if reviewer: | ||||
self.sa.delete(reviewer) | ||||
r2746 | def delete(self, pull_request): | |||
pull_request = self.__get_pull_request(pull_request) | ||||
Session().delete(pull_request) | ||||
r2608 | def close_pull_request(self, pull_request): | |||
pull_request = self.__get_pull_request(pull_request) | ||||
pull_request.status = PullRequest.STATUS_CLOSED | ||||
pull_request.updated_on = datetime.datetime.now() | ||||
self.sa.add(pull_request) | ||||
|
r3192 | def _get_changesets(self, alias, org_repo, org_ref, other_repo, other_ref): | ||
r2442 | """ | |||
|
r3323 | Returns a list of changesets that can be merged from org_repo@org_ref | ||
to other_repo@other_ref ... and the ancestor that would be used for merge | ||||
r2442 | ||||
:param org_repo: | ||||
:param org_ref: | ||||
:param other_repo: | ||||
:param other_ref: | ||||
:param tmp: | ||||
""" | ||||
r3023 | ||||
|
r3323 | ancestor = None | ||
r3023 | ||||
|
r3300 | if alias == 'hg': | ||
|
r3301 | # lookup up the exact node id | ||
_revset_predicates = { | ||||
'branch': 'branch', | ||||
'book': 'bookmark', | ||||
'tag': 'tag', | ||||
'rev': 'id', | ||||
} | ||||
r3387 | ||||
|
r3301 | org_rev_spec = "%s('%s')" % (_revset_predicates[org_ref[0]], | ||
safe_str(org_ref[1])) | ||||
r3387 | if org_ref[1] == EmptyChangeset().raw_id: | |||
org_rev = org_ref[1] | ||||
else: | ||||
org_rev = org_repo._repo[scmutil.revrange(org_repo._repo, | ||||
[org_rev_spec])[-1]] | ||||
|
r3301 | other_rev_spec = "%s('%s')" % (_revset_predicates[other_ref[0]], | ||
safe_str(other_ref[1])) | ||||
r3387 | if other_ref[1] == EmptyChangeset().raw_id: | |||
other_rev = other_ref[1] | ||||
else: | ||||
other_rev = other_repo._repo[scmutil.revrange(other_repo._repo, | ||||
[other_rev_spec])[-1]] | ||||
r2442 | ||||
|
r3300 | #case two independent repos | ||
if org_repo != other_repo: | ||||
|
r3322 | hgrepo = unionrepo.unionrepository(other_repo.baseui, | ||
other_repo.path, | ||||
org_repo.path) | ||||
# all the changesets we are looking for will be in other_repo, | ||||
# so rev numbers from hgrepo can be used in other_repo | ||||
|
r3300 | |||
r3010 | #no remote compare do it on the same repository | |||
|
r3300 | else: | ||
|
r3322 | hgrepo = other_repo._repo | ||
r3315 | ||||
|
r3322 | revs = ["ancestors(id('%s')) and not ancestors(id('%s'))" % | ||
(other_rev, org_rev)] | ||||
|
r3323 | changesets = [other_repo.get_changeset(cs) | ||
for cs in scmutil.revrange(hgrepo, revs)] | ||||
if org_repo != other_repo: | ||||
ancestors = scmutil.revrange(hgrepo, | ||||
["ancestor(id('%s'), id('%s'))" % (org_rev, other_rev)]) | ||||
if len(ancestors) == 1: | ||||
ancestor = hgrepo[ancestors[0]].hex() | ||||
|
r3300 | |||
elif alias == 'git': | ||||
assert org_repo == other_repo, (org_repo, other_repo) # no git support for different repos | ||||
so, se = org_repo.run_git_command( | ||||
'log --reverse --pretty="format: %%H" -s -p %s..%s' % (org_ref[1], | ||||
other_ref[1]) | ||||
) | ||||
|
r3323 | changesets = [org_repo.get_changeset(cs) | ||
for cs in re.findall(r'[0-9a-fA-F]{40}', so)] | ||||
r2442 | ||||
|
r3323 | return changesets, ancestor | ||
r2442 | ||||
def get_compare_data(self, org_repo, org_ref, other_repo, other_ref): | ||||
""" | ||||
|
r3301 | Returns incoming changesets for mercurial repositories | ||
r2442 | ||||
:param org_repo: | ||||
:param org_ref: | ||||
:param other_repo: | ||||
:param other_ref: | ||||
""" | ||||
if len(org_ref) != 2 or not isinstance(org_ref, (list, tuple)): | ||||
raise Exception('org_ref must be a two element list/tuple') | ||||
if len(other_ref) != 2 or not isinstance(org_ref, (list, tuple)): | ||||
raise Exception('other_ref must be a two element list/tuple') | ||||
r3010 | org_repo_scm = org_repo.scm_instance | |||
other_repo_scm = other_repo.scm_instance | ||||
alias = org_repo.scm_instance.alias | ||||
|
r3323 | cs_ranges, ancestor = self._get_changesets(alias, | ||
org_repo_scm, org_ref, | ||||
other_repo_scm, other_ref) | ||||
return cs_ranges, ancestor | ||||