##// END OF EJS Templates
some fixs
Matthias BUSSONNIER -
Show More
@@ -73,10 +73,14 b" def post_gist(content, description='', filename='file', auth=False):"
73 response_data = json.loads(response.text)
73 response_data = json.loads(response.text)
74 return response_data['html_url']
74 return response_data['html_url']
75
75
76 def get_pull_request(project, num, httpv2=False):
76 def get_pull_request(project, num, github_api=3):
77 if httpv2 :
77 """get pull request info by number
78
79 github_api : version of github api to use
80 """
81 if github_api==2 :
78 url = "http://github.com/api/v2/json/pulls/{project}/{num}".format(project=project, num=num)
82 url = "http://github.com/api/v2/json/pulls/{project}/{num}".format(project=project, num=num)
79 else:
83 elif github_api == 3:
80 url = "https://api.github.com/repos/{project}/pulls/{num}".format(project=project, num=num)
84 url = "https://api.github.com/repos/{project}/pulls/{num}".format(project=project, num=num)
81 response = requests.get(url)
85 response = requests.get(url)
82 response.raise_for_status()
86 response.raise_for_status()
@@ -84,8 +88,12 b' def get_pull_request(project, num, httpv2=False):'
84 return json.loads(response.text)['pull']
88 return json.loads(response.text)['pull']
85 return json.loads(response.text)
89 return json.loads(response.text)
86
90
87 def get_pulls_list(project,httpv2=False):
91 def get_pulls_list(project, github_api=3):
88 if not httpv2 :
92 """get pull request list
93
94 github_api : version of github api to use
95 """
96 if github_api == 3 :
89 url = "https://api.github.com/repos/{project}/pulls".format(project=project)
97 url = "https://api.github.com/repos/{project}/pulls".format(project=project)
90 else :
98 else :
91 url = "http://github.com/api/v2/json/pulls/{project}".format(project=project)
99 url = "http://github.com/api/v2/json/pulls/{project}".format(project=project)
@@ -2,12 +2,13 b''
2 # -*- coding: utf-8 -*-
2 # -*- coding: utf-8 -*-
3 """
3 """
4 Usage:
4 Usage:
5 python test_pr.py 1657
5 python git-mpr.py -m 1657
6 """
6 """
7 from __future__ import print_function
7 from __future__ import print_function
8
8
9 import re
9 import re
10 import requests
10 import requests
11 import argparse
11 from subprocess import call, check_call, check_output, CalledProcessError
12 from subprocess import call, check_call, check_output, CalledProcessError
12 import sys
13 import sys
13
14
@@ -17,7 +18,14 b" ipy_repository = 'git://github.com/ipython/ipython.git'"
17 gh_project="ipython/ipython"
18 gh_project="ipython/ipython"
18 not_merged={}
19 not_merged={}
19
20
20 def get_branch(repo, branch, owner, mergeable):
21
22 def merge_branch(repo, branch, owner, mergeable):
23 """try to merge the givent branch into the current one
24
25 If something does not goes smoothly, merge is aborted
26
27 Returns True if merge sucessfull, False otherwise
28 """
21 merged_branch = "%s-%s" % (owner, branch)
29 merged_branch = "%s-%s" % (owner, branch)
22 # Delete the branch first
30 # Delete the branch first
23 try :
31 try :
@@ -27,20 +35,26 b' def get_branch(repo, branch, owner, mergeable):'
27 return False
35 return False
28 return True
36 return True
29
37
30 def merge_pr(num,httpv2=False):
38
39 def merge_pr(num,github_api=3):
40 """ try to merge the branch of PR `num` into current branch
41
42 github_api : use github api v2 (to bypass https and issues with proxy) to find the
43 remote branch that should be merged by it's number
44 """
31 # Get Github authorisation first, so that the user is prompted straight away
45 # Get Github authorisation first, so that the user is prompted straight away
32 # if their login is needed.
46 # if their login is needed.
33
47
34 pr = gh_api.get_pull_request(gh_project, num, httpv2)
48 pr = gh_api.get_pull_request(gh_project, num, github_api)
35 if(httpv2):
49 if github_api == 2:
36 repo = pr['head']['repository']['url']
50 repo = pr['head']['repository']['url']
37 owner=pr['head']['user']['name'],
51 owner = pr['head']['user']['name']
38 else :
52 elif github_api == 2 :
39 repo=pr['head']['repo']['clone_url']
53 repo = pr['head']['repo']['clone_url']
40 owner=pr['head']['repo']['owner']['login'],
54 owner = pr['head']['repo']['owner']['login']
41
55
42 branch=pr['head']['ref']
56 branch = pr['head']['ref']
43 mergeable = get_branch(repo=repo,
57 mergeable = merge_branch(repo=repo,
44 branch=branch,
58 branch=branch,
45 owner=owner,
59 owner=owner,
46 mergeable=pr['mergeable'],
60 mergeable=pr['mergeable'],
@@ -54,9 +68,7 b' def merge_pr(num,httpv2=False):'
54 print("==============================================================================")
68 print("==============================================================================")
55
69
56
70
57
71 def main(*args):
58 if __name__ == '__main__':
59 import argparse
60 parser = argparse.ArgumentParser(
72 parser = argparse.ArgumentParser(
61 description="""
73 description="""
62 Merge (one|many) github pull request by their number.\
74 Merge (one|many) github pull request by their number.\
@@ -65,7 +77,7 b" if __name__ == '__main__':"
65 and continue to the next if any.
77 and continue to the next if any.
66 """
78 """
67 )
79 )
68 parser.add_argument('-v2','--githubapiv2', action='store_const', const=True)
80 parser.add_argument('-v2','--githubapiv2', action='store_const', const=2)
69
81
70 grp = parser.add_mutually_exclusive_group()
82 grp = parser.add_mutually_exclusive_group()
71 grp.add_argument(
83 grp.add_argument(
@@ -87,11 +99,15 b" if __name__ == '__main__':"
87 metavar='pr-number')
99 metavar='pr-number')
88 not_merged = {};
100 not_merged = {};
89 args = parser.parse_args()
101 args = parser.parse_args()
90 ghv2 = args.githubapiv2
102 if args.githubapiv2 == 2 :
103 github_api = 2
104 else :
105 github_api = 3
106
91 if(args.list):
107 if(args.list):
92 pr_list = gh_api.get_pulls_list(gh_project, ghv2)
108 pr_list = gh_api.get_pulls_list(gh_project, github_api)
93 for pr in pr_list :
109 for pr in pr_list :
94 mergeable = gh_api.get_pull_request(gh_project, pr['number'],httpv2=ghv2)['mergeable']
110 mergeable = gh_api.get_pull_request(gh_project, pr['number'],github_api=github_api)['mergeable']
95
111
96 ismgb = u"√" if mergeable else " "
112 ismgb = u"√" if mergeable else " "
97 print(u"* #{number} [{ismgb}]: {title}".format(
113 print(u"* #{number} [{ismgb}]: {title}".format(
@@ -107,7 +123,7 b" if __name__ == '__main__':"
107
123
108 elif args.merge:
124 elif args.merge:
109 for num in args.merge :
125 for num in args.merge :
110 merge_pr(num,httpv2=ghv2)
126 merge_pr(num, github_api=github_api)
111
127
112 if not_merged :
128 if not_merged :
113 print('*************************************************************************************')
129 print('*************************************************************************************')
@@ -115,3 +131,6 b" if __name__ == '__main__':"
115 for num,cmd in not_merged.items() :
131 for num,cmd in not_merged.items() :
116 print( "PR {num}: {cmd}".format(num=num,cmd=cmd))
132 print( "PR {num}: {cmd}".format(num=num,cmd=cmd))
117 print('*************************************************************************************')
133 print('*************************************************************************************')
134
135 if __name__ == '__main__':
136 main()
General Comments 0
You need to be logged in to leave comments. Login now