/brz/remove-bazaar

To get this branch, use:
bzr branch http://gegoxaren.bato24.eu/bzr/brz/remove-bazaar

« back to all changes in this revision

Viewing changes to breezy/plugins/propose/github.py

  • Committer: Jelmer Vernooij
  • Date: 2020-03-22 20:02:36 UTC
  • mto: (7490.7.7 work)
  • mto: This revision was merged to the branch mainline in revision 7501.
  • Revision ID: jelmer@jelmer.uk-20200322200236-fsbl91ktcn6fcbdd
Fix tests.

Show diffs side-by-side

added added

removed removed

Lines of Context:
breezy/plugins/propose/github.py
16
16
 
17
17
"""Support for GitHub."""
18
18
 
 
19
from __future__ import absolute_import
 
20
 
19
21
import json
20
22
import os
21
23
 
42
44
    version_string as breezy_version,
43
45
    )
44
46
from ...config import AuthenticationConfig, GlobalStack
45
 
from ...errors import (
46
 
    InvalidHttpResponse,
47
 
    PermissionDenied,
48
 
    UnexpectedHttpStatus,
49
 
    )
 
47
from ...errors import InvalidHttpResponse, PermissionDenied
50
48
from ...git.urls import git_url_to_bzr_url
51
49
from ...i18n import gettext
 
50
from ...sixish import PY3
52
51
from ...trace import note
53
52
from ...transport import get_transport
54
53
from ...transport.http import default_user_agent
124
123
    def __repr__(self):
125
124
        return "<%s at %r>" % (type(self).__name__, self.url)
126
125
 
127
 
    name = 'GitHub'
128
 
 
129
126
    @property
130
127
    def url(self):
131
128
        return self._pr['html_url']
138
135
    def get_source_branch_url(self):
139
136
        return self._branch_from_part(self._pr['head'])
140
137
 
141
 
    def get_source_revision(self):
142
 
        """Return the latest revision for the source branch."""
143
 
        from breezy.git.mapping import default_mapping
144
 
        return default_mapping.revision_id_foreign_to_bzr(
145
 
            self._pr['head']['sha'].encode('ascii'))
146
 
 
147
138
    def get_target_branch_url(self):
148
139
        return self._branch_from_part(self._pr['base'])
149
140
 
168
159
        if response.status == 422:
169
160
            raise ValidationFailed(json.loads(response.text))
170
161
        if response.status != 200:
171
 
            raise UnexpectedHttpStatus(self._pr['url'], response.status)
 
162
            raise InvalidHttpResponse(self._pr['url'], response.text)
172
163
        self._pr = json.loads(response.text)
173
164
 
174
165
    def set_description(self, description):
197
188
 
198
189
    def merge(self, commit_message=None):
199
190
        # https://developer.github.com/v3/pulls/#merge-a-pull-request-merge-button
200
 
        data = {}
201
 
        if commit_message:
202
 
            data['commit_message'] = commit_messae
203
 
        response = self._gh._api_request(
204
 
            'PUT', self._pr['url'] + "/merge", body=json.dumps(data).encode('utf-8'))
205
 
        if response.status == 422:
206
 
            raise ValidationFailed(json.loads(response.text))
207
 
        if response.status != 200:
208
 
            raise UnexpectedHttpStatus(self._pr['url'], response.status)
 
191
        self._pr.merge(commit_message=commit_message)
209
192
 
210
193
    def get_merged_by(self):
211
194
        merged_by = self._pr.get('merged_by')
220
203
        import iso8601
221
204
        return iso8601.parse_date(merged_at)
222
205
 
223
 
    def post_comment(self, body):
224
 
        data = {'body': body}
225
 
        response = self._gh._api_request(
226
 
            'POST', self._pr['comments_url'], body=json.dumps(data).encode('utf-8'))
227
 
        if response.status == 422:
228
 
            raise ValidationFailed(json.loads(response.text))
229
 
        if response.status != 201:
230
 
            raise UnexpectedHttpStatus(
231
 
                self._pr['comments_url'], response.status)
232
 
        json.loads(response.text)
233
 
 
234
206
 
235
207
def parse_github_url(url):
236
208
    (scheme, user, password, host, port, path) = urlutils.parse_url(
250
222
 
251
223
 
252
224
def github_url_to_bzr_url(url, branch_name):
 
225
    if not PY3:
 
226
        branch_name = branch_name.encode('utf-8')
253
227
    return git_url_to_bzr_url(url, branch_name)
254
228
 
255
229
 
289
263
            raise NoSuchProject(path)
290
264
        if response.status == 200:
291
265
            return json.loads(response.text)
292
 
        raise UnexpectedHttpStatus(path, response.status)
 
266
        raise InvalidHttpResponse(path, response.text)
293
267
 
294
268
    def _get_repo_pulls(self, path, head=None, state=None):
295
269
        path = path + '?'
305
279
            raise NoSuchProject(path)
306
280
        if response.status == 200:
307
281
            return json.loads(response.text)
308
 
        raise UnexpectedHttpStatus(path, response.status)
 
282
        raise InvalidHttpResponse(path, response.text)
309
283
 
310
284
    def _create_pull(self, path, title, head, base, body=None, labels=None,
311
285
                     assignee=None, draft=False, maintainer_can_modify=False):
328
302
        if response.status == 403:
329
303
            raise PermissionDenied(path, response.text)
330
304
        if response.status != 201:
331
 
            raise UnexpectedHttpStatus(path, response.status)
 
305
            raise InvalidHttpResponse(path, 'req is invalid %d %r: %r' % (response.status, data, response.text))
332
306
        return json.loads(response.text)
333
307
 
334
308
    def _get_user_by_email(self, email):
335
309
        path = 'search/users?q=%s+in:email' % email
336
310
        response = self._api_request('GET', path)
337
311
        if response.status != 200:
338
 
            raise UnexpectedHttpStatus(path, response.status)
 
312
            raise InvalidHttpResponse(path, response.text)
339
313
        ret = json.loads(response.text)
340
314
        if ret['total_count'] == 0:
341
315
            raise KeyError('no user with email %s' % email)
350
324
            path = 'user'
351
325
        response = self._api_request('GET', path)
352
326
        if response.status != 200:
353
 
            raise UnexpectedHttpStatus(path, response.status)
 
327
            raise InvalidHttpResponse(path, response.text)
354
328
        return json.loads(response.text)
355
329
 
356
330
    def _get_organization(self, name):
357
331
        path = 'orgs/%s' % name
358
332
        response = self._api_request('GET', path)
359
333
        if response.status != 200:
360
 
            raise UnexpectedHttpStatus(path, response.status)
 
334
            raise InvalidHttpResponse(path, response.text)
361
335
        return json.loads(response.text)
362
336
 
363
337
    def _list_paged(self, path, parameters=None, per_page=None):
376
350
                ';'.join(['%s=%s' % (k, urlutils.quote(v))
377
351
                          for (k, v) in parameters.items()]))
378
352
            if response.status != 200:
379
 
                raise UnexpectedHttpStatus(path, response.status)
 
353
                raise InvalidHttpResponse(path, response.text)
380
354
            data = json.loads(response.text)
381
355
            for entry in data['items']:
382
356
                i += 1
390
364
        return self._list_paged(path, {'q': query}, per_page=DEFAULT_PER_PAGE)
391
365
 
392
366
    def _create_fork(self, path, owner=None):
393
 
        if owner and owner != self.current_user['login']:
 
367
        if owner and owner != self._current_user['login']:
394
368
            path += '?organization=%s' % owner
395
369
        response = self._api_request('POST', path)
396
370
        if response.status != 202:
397
 
            raise UnexpectedHttpStatus(path, response.status)
 
371
            raise InvalidHttpResponse(path, 'status: %d, %r' % (response.status, response.text))
398
372
        return json.loads(response.text)
399
373
 
400
374
    @property
404
378
    def __init__(self, transport):
405
379
        self._token = retrieve_github_token('https', GITHUB_HOST)
406
380
        self.transport = transport
407
 
        self._current_user = None
408
 
 
409
 
    @property
410
 
    def current_user(self):
411
 
        if self._current_user is None:
412
 
            self._current_user = self._get_user()
413
 
        return self._current_user
 
381
        self._current_user = self._get_user()
414
382
 
415
383
    def publish_derived(self, local_branch, base_branch, name, project=None,
416
384
                        owner=None, revision_id=None, overwrite=False,
418
386
        base_owner, base_project, base_branch_name = parse_github_branch_url(base_branch)
419
387
        base_repo = self._get_repo(base_owner, base_project)
420
388
        if owner is None:
421
 
            owner = self.current_user['login']
 
389
            owner = self._current_user['login']
422
390
        if project is None:
423
391
            project = base_repo['name']
424
392
        try:
454
422
        base_owner, base_project, base_branch_name = parse_github_branch_url(base_branch)
455
423
        base_repo = self._get_repo(base_owner, base_project)
456
424
        if owner is None:
457
 
            owner = self.current_user['login']
 
425
            owner = self._current_user['login']
458
426
        if project is None:
459
427
            project = base_repo['name']
460
428
        try:
529
497
            query.append('is:closed')
530
498
        elif status == 'merged':
531
499
            query.append('is:merged')
532
 
        query.append('author:%s' % self.current_user['login'])
 
500
        query.append('author:%s' % self._current_user['login'])
533
501
        for issue in self._search_issues(query=' '.join(query)):
534
502
            url = issue['pull_request']['url']
535
503
            response = self._api_request('GET', url)
536
504
            if response.status != 200:
537
 
                raise UnexpectedHttpStatus(url, response.status)
 
505
                raise InvalidHttpResponse(url, response.text)
538
506
            yield GitHubMergeProposal(self, json.loads(response.text))
539
507
 
540
508
    def get_proposal_by_url(self, url):
543
511
    def iter_my_forks(self):
544
512
        response = self._api_request('GET', '/user/repos')
545
513
        if response.status != 200:
546
 
            raise UnexpectedHttpStatus(self.transport.user_url, response.status)
 
514
            raise InvalidHttpResponse(url, response.text)
547
515
        for project in json.loads(response.text):
548
516
            if not project['fork']:
549
517
                continue
558
526
            return
559
527
        if response.status == 200:
560
528
            return json.loads(response.text)
561
 
        raise UnexpectedHttpStatus(path, response.status)
562
 
 
563
 
    def get_current_user(self):
564
 
        return self.current_user['login']
565
 
 
566
 
    def get_user_url(self, username):
567
 
        return urlutils.join(self.base_url, username)
 
529
        raise InvalidHttpResponse(path, response.text)
568
530
 
569
531
 
570
532
class GitHubMergeProposalBuilder(MergeProposalBuilder):