comment-github-pullrequest

#!/usr/bin/env python
"""
Makes a comment on Pull request on Github
"""

import os
import sys
import argparse
from socket import setdefaulttimeout
from github import Github

from Mu2eCI import config
from Mu2eCI.logger import log
from Mu2eCI.comment_gh_pr import comment_gh_pr

setdefaulttimeout(120)

parser = argparse.ArgumentParser(
    description="Process a pull request given a repository and PR number."
)
parser.add_argument("-p", "--pullrequest", type=int, help="The Pull Request ID.")
parser.add_argument(
    "-r",
    "--repository",
    type=str,
    help="The GitHub repository. Must be in the format e.g. Mu2e/Offline",
    choices=config.main["supported_repos"],
)

parser.add_argument(
    "-R",
    "--report-file",
    type=str,
    help="The report file e.g. gh-report.md",
    default="",
)

args = parser.parse_args()


if __name__ == "__main__":
    gh = Github(login_or_token=os.environ["GITHUBTOKEN"], retry=3)

    try:
        msg = ""
        if len(args.report_file) > 0:
            with open(args.report_file, "r") as f:
                msg = f.read()
        comment_gh_pr(gh, args.repository, args.pullrequest, msg)
    except Exception:
        log.exception("Failed to add comment")
        sys.exit(1)