30 lines
838 B
Python
30 lines
838 B
Python
|
import json, os, sys
|
||
|
|
if sys.version.startswith('2'):
|
||
|
|
from urllib2 import urlopen, HTTPError
|
||
|
|
else:
|
||
|
|
from urllib.request import urlopen
|
||
|
|
from urllib.error import HTTPError
|
||
|
|
try:
|
||
|
|
pr_id = int(os.environ.get('TRAVIS_PULL_REQUEST', 'false'))
|
||
|
|
except ValueError:
|
||
|
|
print('Not a pull request')
|
||
|
|
sys.exit(0)
|
||
|
|
print('Pull request %i' % pr_id)
|
||
|
|
res = {}
|
||
|
|
try:
|
||
|
|
res = json.loads(urlopen('https://api.github.com/repos/dfhack/dfhack/pulls/%i' % pr_id).read().decode('utf-8'))
|
||
|
|
except ValueError:
|
||
|
|
pass
|
||
|
|
except HTTPError:
|
||
|
|
print('Failed to retrieve PR information from API')
|
||
|
|
sys.exit(2)
|
||
|
|
if 'base' not in res or 'ref' not in res['base']:
|
||
|
|
print('Invalid JSON returned from API')
|
||
|
|
sys.exit(2)
|
||
|
|
if res['base']['ref'] != 'develop':
|
||
|
|
print('Not based on develop branch')
|
||
|
|
sys.exit(1)
|
||
|
|
else:
|
||
|
|
print('Ok')
|
||
|
|
sys.exit(0)
|