14
14
# along with this program; if not, write to the Free Software
15
15
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
21
17
from bzrlib import (
21
from bzrlib.lazy_import import lazy_import
22
lazy_import(globals(), """
26
28
from bzrlib.plugins.launchpad import (
31
from lazr.restfulclient import errors as restful_errors
34
35
class ProposeMergeHooks(hooks.Hooks):
55
56
hooks = ProposeMergeHooks()
57
58
def __init__(self, tree, source_branch, target_branch, message, reviews,
59
staging=False, approve=False):
61
62
:param tree: The working tree for the source branch.
65
66
:param reviews: A list of tuples of reviewer, review type.
66
67
:param staging: If True, propose the merge against staging instead of
69
:param approve: If True, mark the new proposal as approved immediately.
70
This is useful when a project permits some things to be approved
71
by the submitter (e.g. merges between release and deployment
71
76
lp_instance = 'staging'
78
lp_instance = 'production'
74
79
service = lp_registration.LaunchpadService(lp_instance=lp_instance)
75
80
self.launchpad = lp_api.login(service)
76
81
self.source_branch = lp_api.LaunchpadBranch.from_bzr(
81
86
self.target_branch = lp_api.LaunchpadBranch.from_bzr(
82
87
self.launchpad, target_branch)
83
88
self.commit_message = message
89
# XXX: this is where bug lp:583638 could be tackled.
85
91
target_reviewer = self.target_branch.lp.reviewer
86
92
if target_reviewer is None:
90
96
self.reviews = [(self.launchpad.people[reviewer], review_type)
91
97
for reviewer, review_type in
99
self.approve = approve
94
101
def get_comment(self, prerequisite_branch):
95
102
"""Determine the initial comment for the merge proposal."""
147
154
if mp.target_branch.self_link == self.target_branch.lp.self_link:
148
155
raise errors.BzrCommandError(
149
156
'There is already a branch merge proposal: %s' %
157
lp_api.canonical_url(mp))
152
159
def _get_prerequisite_branch(self):
153
160
hooks = self.hooks['get_prerequisite']
160
167
'prerequisite_branch': prerequisite_branch})
161
168
return prerequisite_branch
170
def call_webservice(self, call, *args, **kwargs):
171
"""Make a call to the webservice, wrapping failures.
173
:param call: The call to make.
174
:param *args: *args for the call.
175
:param **kwargs: **kwargs for the call.
176
:return: The result of calling call(*args, *kwargs).
178
from lazr.restfulclient import errors as restful_errors
180
return call(*args, **kwargs)
181
except restful_errors.HTTPError, e:
183
for line in e.content.splitlines():
184
if line.startswith('Traceback (most recent call last):'):
186
error_lines.append(line)
187
raise Exception(''.join(error_lines))
163
189
def create_proposal(self):
164
190
"""Perform the submission."""
165
191
prerequisite_branch = self._get_prerequisite_branch()
175
201
review_types.append(review_type)
176
202
reviewers.append(reviewer.self_link)
177
203
initial_comment = self.get_comment(prerequisite_branch)
179
mp = self.source_branch.lp.createMergeProposal(
180
target_branch=self.target_branch.lp,
181
prerequisite_branch=prereq,
182
initial_comment=initial_comment,
183
commit_message=self.commit_message, reviewers=reviewers,
184
review_types=review_types)
185
except restful_errors.HTTPError, e:
187
for line in e.content.splitlines():
188
if line.startswith('Traceback (most recent call last):'):
190
error_lines.append(line)
191
raise Exception(''.join(error_lines))
193
webbrowser.open(canonical_url(mp))
204
mp = self.call_webservice(
205
self.source_branch.lp.createMergeProposal,
206
target_branch=self.target_branch.lp,
207
prerequisite_branch=prereq,
208
initial_comment=initial_comment,
209
commit_message=self.commit_message, reviewers=reviewers,
210
review_types=review_types)
212
self.call_webservice(mp.setStatus, status='Approved')
213
webbrowser.open(lp_api.canonical_url(mp))
196
216
def modified_files(old_tree, new_tree):
200
220
if c and k == 'file':
204
def canonical_url(object):
205
"""Return the canonical URL for a branch."""
206
scheme, netloc, path, params, query, fragment = urlparse.urlparse(
207
str(object.self_link))
208
path = '/'.join(path.split('/')[2:])
209
netloc = netloc.replace('api.', 'code.')
210
return urlparse.urlunparse((scheme, netloc, path, params, query,