axmol/tools/jenkins-scripts/master-scripts/job-comment-trigger.py

137 lines
4.1 KiB
Python

import json
import re
import os
import requests
import sys
import traceback
import urllib2
http_proxy = ''
if('HTTP_PROXY' in os.environ):
http_proxy = os.environ['HTTP_PROXY']
proxyDict = {'http': http_proxy, 'https': http_proxy}
def main():
#get payload from os env
payload_str = os.environ['payload']
#parse to json obj
payload = json.loads(payload_str)
print payload
issue = payload['issue']
#get pull number
pr_num = issue['number']
print 'pr_num:' + str(pr_num)
payload_forword = {"number": pr_num}
comment = payload['comment']
#get comment body
comment_body = comment['body']
print comment_body
#will check 'ci' comment
searchCI = re.search("\[ci.*\]", comment_body)
#will check console/console create
searchConsole = re.search('\[console.*\]', comment_body)
if searchCI is None and searchConsole is None:
print 'skip build for pull request #' + str(pr_num)
return(0)
#build for pull request action 'open' and 'synchronize', skip 'close'
action = issue['state']
print 'action: ' + action
payload_forword['action'] = action
pr = issue['pull_request']
url = pr['html_url']
print "url:" + url
payload_forword['html_url'] = url
#get pull request info
req = os.environ['GITHUB_REMOTE'] + str(pr_num)
pr_payload = ''
try:
pr_payload = urllib2.urlopen(req).read()
except:
traceback.print_exc()
repository = json.loads(pr_payload)
#get statuses url
statuses_url = repository['statuses_url']
payload_forword['statuses_url'] = statuses_url
print 'statuses_url: ' + statuses_url
#get comments url
comments_url = repository['comments_url']
payload_forword['comments_url'] = comments_url
print 'comments_url: ' + comments_url
#get pr target branch
branch = repository['base']['ref']
payload_forword['branch'] = branch
print 'branch: ' + branch
#set commit status to pending
target_url = os.environ['JOB_PULL_REQUEST_BUILD_URL']
if(action == 'closed' or action == 'assigned'
or branch == 'v2' or branch == 'v3-doc' or action == 'unassigned'):
print 'pull request #' + str(pr_num) + ' is ' + action + ', no build triggered'
return(0)
data = {"state": "pending", "target_url": target_url, "context": "Jenkins CI",
"description": "Wait available build machine..."}
access_token = os.environ['GITHUB_ACCESS_TOKEN']
Headers = {"Authorization": "token " + access_token}
try:
if searchCI:
ciOper = searchCI.group()
if('rebuild' in ciOper):
requests.post(statuses_url, data=json.dumps(data),
headers=Headers, proxies=proxyDict)
except:
traceback.print_exc()
job_trigger_url = ''
if searchCI:
ciOper = searchCI.group()
if('rebuild' in ciOper):
job_trigger_url = os.environ['JOB_PULL_REQUEST_BUILD_TRIGGER_URL']
if('emptytest' in ciOper):
job_trigger_url = os.environ['JOB_EMPTYTEST_TRIGGER_URL']
if('release' in ciOper):
searchTag = re.search('\[ci release (.*)\]', ciOper)
if searchTag:
ci_tag = searchTag.group(1)
payload_forword['tag'] = ci_tag
job_trigger_url = os.environ['JOB_RELEASE_TRIGGER_URL']
if searchConsole:
consoleOper = searchConsole.group()
job_trigger_url = os.environ['JOB_CONSOLE_TEST_TRIGGER_URL']
payload_forword['console'] = consoleOper
print 'job_trigger_url is: ', job_trigger_url
#send trigger and payload
if('tag' in payload_forword):
post_data = {'tag': ""}
post_data['tag'] = payload_forword['tag']
else:
post_data = {'payload': ""}
post_data['payload'] = json.dumps(payload_forword)
requests.post(job_trigger_url, data=post_data, proxies=proxyDict)
return(0)
# -------------- main --------------
if __name__ == '__main__':
sys_ret = 0
try:
sys_ret = main()
except:
traceback.print_exc()
sys_ret = 1
finally:
sys.exit(sys_ret)