1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
|
#!/usr/bin/env python
import os
import json
import argparse
from github import Github, Auth as GithubAuth
from github.PullRequest import PullRequest
from gh_status import update_pr_comment_text
def main():
parser = argparse.ArgumentParser()
parser.add_argument("--rewrite", dest="rewrite", action="store_true")
parser.add_argument("--color", dest="color", default="white")
parser.add_argument("--fail", dest="fail", action="store_true")
parser.add_argument("--ok", dest="ok", action="store_true")
parser.add_argument("text", type=argparse.FileType("r"), nargs="?", default="-")
args = parser.parse_args()
color = args.color
if args.ok:
color = 'green'
elif args.fail:
color = 'red'
build_preset = os.environ["BUILD_PRESET"]
gh = Github(auth=GithubAuth.Token(os.environ["GITHUB_TOKEN"]))
with open(os.environ["GITHUB_EVENT_PATH"]) as fp:
event = json.load(fp)
prnum = event.get("pull_request")
if not prnum is None:
pr = gh.create_from_raw_data(PullRequest, prnum)
update_pr_comment_text(pr, build_preset, color, args.text.read().rstrip(), args.rewrite)
if __name__ == "__main__":
main()
|
