scripts/notify.py

# -*- coding: utf-8 -*-

# Copyright (C) 2010-2016  RhodeCode GmbH
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU Affero General Public License, version 3
# (only), as published by the Free Software Foundation.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU Affero General Public License
# along with this program.  If not, see <http://www.gnu.org/licenses/>.
#
# This program is dual-licensed. If you wish to learn more about the
# RhodeCode Enterprise Edition, including its added features, Support services,
# and proprietary license terms, please see https://rhodecode.com/licenses/

import argparse

from rhodecode_api import RhodeCodeAPI


def add_api_pr_args(parser):
    parser.add_argument("--url", help="Rhodecode host url", required=True)
    parser.add_argument("--token", help="Rhodecode user token", required=True)
    parser.add_argument("--repoid", help="Repo id", required=True)
    parser.add_argument("--message", help="Comment to add", required=True)
    parser.add_argument("--status", help="Status to set (approved|rejected)")
    parser.add_argument(
        "--prid",
        help="(Optional) Pull request id")
    parser.add_argument(
        "--commit-id",
        help="(Optional) Expected commit ID for source repo")


def get_pr_head(args):
    rc = RhodeCodeAPI(args.url, args.token)
    pr_info = rc.get_pull_request(
        repoid=args.repoid,
        pullrequestid=args.prid
    )
    if pr_info['error']:
        return None

    return pr_info['result']['source']['reference']['commit_id']


def pr_has_same_head(args):
    head_id = get_pr_head(args)
    return head_id == args.commit_id


if __name__ == "__main__":
    parser = argparse.ArgumentParser()
    add_api_pr_args(parser)
    args = parser.parse_args()
    status = args.status

    rc = RhodeCodeAPI(args.url, args.token)

    if not (args.prid or args.commit_id):
        raise AttributeError('You must specify --prid or --commit-id')

    if args.prid:
        if args.commit_id and status:
            if not pr_has_same_head(args):
                # PR updated, don't set status
                status = None

        resp = rc.comment_pull_request(
            repoid=args.repoid,
            pullrequestid=int(args.prid),
            message=args.message,
            status=status
        )
    elif args.commit_id:
        resp = rc.comment_commit(
            repoid=args.repoid,
            commit_id=args.commit_id,
            message=args.message,
            status=status,
        )

    if resp['error']:
        raise Exception(resp['error'])
    else:
        print resp