check_pr_approval.py

# Copyright (c) 2018 PaddlePaddle Authors. All Rights Reserved.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

import json
import sys


def check_approval(count, required_reviewers):
    json_buff = ""
    for line in sys.stdin:
        json_buff = "".join([json_buff, line])
    json_resp = json.loads(json_buff)
    approves = 0
    approved_user_ids = []
    approved_user_logins = set()
    for review in json_resp:
        if review["state"] == "APPROVED":
            approves += 1
            approved_user_ids.append(review["user"]["id"])
            approved_user_logins.add(review["user"]["login"])

    # convert to int
    required_reviewers_int = set()
    required_reviewers_login = set()
    for rr in required_reviewers:
        if rr.isdigit():
            required_reviewers_int.add(int(rr))
        else:
            required_reviewers_login.add(rr)

    if (
        len(set(approved_user_ids) & required_reviewers_int)
        + len(approved_user_logins & required_reviewers_login)
        >= count
    ):
        print("TRUE")
    else:
        print("FALSE")


if __name__ == "__main__":
    if len(sys.argv) > 1 and sys.argv[1].isdigit():
        check_approval(int(sys.argv[1]), sys.argv[2:])
    else:
        print(
            "Usage: python check_pr_approval.py [count] [required reviewer id] ..."
        )