gh-teams.py

#!/usr/bin/env python3
from __future__ import print_function
from copy import deepcopy
from github import Github
from os.path import expanduser, exists
from argparse import ArgumentParser
from sys import exit
from socket import setdefaulttimeout
from github_utils import api_rate_limits, github_api, add_organization_member
from github_utils import create_team, get_pending_members, get_gh_token
from github_utils import get_delete_pending_members, get_failed_pending_members
from categories import CMSSW_ORP, CMSSW_L2, CMS_SDT

setdefaulttimeout(120)

CMS_OWNERS = ["davidlange6", "smuzaffar", "cmsbuild"] + CMSSW_ORP[:]
CMS_ORGANIZATIONS = ["cms-data", "cms-externals", "cms-sw"]

REPO_OWNERS = {}
REPO_TEAMS = {}
for org in CMS_ORGANIZATIONS:
    REPO_OWNERS[org] = CMS_OWNERS[:]
    REPO_TEAMS[org] = {}

#################################
# Set Extra owners for repos     #
#################################
REPO_OWNERS["cms-data"] += []
REPO_OWNERS["cms-externals"] += []
REPO_OWNERS["cms-sw"] += ["sextonkennedy"]

#################################
# Set Teams for organizations    #
#################################
# Teams for cms-data
REPO_TEAMS["cms-data"]["Developers"] = {"members": CMS_SDT[:], "repositories": {"*": "push"}}

# Teams for cms-externals
REPO_TEAMS["cms-externals"]["Developers"] = deepcopy(REPO_TEAMS["cms-data"]["Developers"])
REPO_TEAMS["cms-externals"]["boost-developers"] = {
    "members": ["fwyzard"],
    "repositories": {"boost": "push"},
}
REPO_TEAMS["cms-externals"]["Developers"]["members"].append("gartung")
REPO_TEAMS["cms-externals"]["Developers"]["members"].append("fwyzard")
REPO_TEAMS["cms-externals"]["Developers"]["members"].append("valsdav")

# Teams for cms-sw
REPO_TEAMS["cms-sw"]["core-team"] = {
    "members": ["dan131riley", "alexstrel", "makortel", "Dr15Jones", "wddgit", "gartung"],
    "repositories": {"framework-team": "push"},
}
REPO_TEAMS["cms-sw"]["RecoLuminosity-LumiDB-admins"] = {
    "members": [],
    "repositories": {"RecoLuminosity-LumiDB": "admin"},
}
REPO_TEAMS["cms-sw"]["generators-l2"] = {
    "members": [],
    "repositories": {"genproductions": "admin", "xsecdb": "admin"},
}
REPO_TEAMS["cms-sw"]["Dqm-Integration-developers"] = {
    "members": ["rovere", "deguio"],
    "repositories": {"DQM-Integration": "push"},
}
REPO_TEAMS["cms-sw"]["configdb-owners"] = {
    "members": [
        "fwyzard",
        "Martin-Grunewald",
        "Sam-Harper",
        "silviodonato",
        "mmusich",
        "missirol",
    ],
    "repositories": {"hlt-confdb": "admin", "web-confdb": "admin"},
}
REPO_TEAMS["cms-sw"]["cmsdist-writers"] = {
    "members": ["h4d4", "muhammadimranfarooqi", "arooshap"] + CMS_SDT[:],
    "repositories": {"cmsdist": "push"},
}
REPO_TEAMS["cms-sw"]["cmssw-l2"] = {"members": ["*"], "repositories": {"cmssw": "pull"}}
REPO_TEAMS["cms-sw"]["cmssw-developers"] = {"members": ["*"], "repositories": {"cmssw": "pull"}}
REPO_TEAMS["cms-sw"]["cms-sw-writers"] = {
    "members": CMS_SDT[:],
    "repositories": {"*": "push", "!cmssw": "pull", "!cmsdist": "pull"},
}
REPO_TEAMS["cms-sw"]["cms-sw-admins"] = {
    "members": CMS_SDT[:],
    "repositories": {"cmssdt-wiki": "admin", "cms-sw.github.io": "admin"},
}

REPO_TEAMS["cms-sw"]["all-l2"] = {
    "members": CMSSW_ORP[:],
}

for user in CMSSW_L2:
    REPO_TEAMS["cms-sw"]["all-l2"]["members"].append(user)
    for cat in CMSSW_L2[user]:
        cat = "%s-l2" % cat
        if not cat in REPO_TEAMS["cms-sw"]:
            REPO_TEAMS["cms-sw"][cat] = {"members": []}
        REPO_TEAMS["cms-sw"][cat]["members"].append(user)

#################################
parser = ArgumentParser()
parser.add_argument(
    "-o",
    "--organization",
    dest="organization",
    help="Github Organization name e.g. cms-sw. Default is * i.e. all cms organizations",
    type=str,
    default="*",
)
parser.add_argument("-n", "-dry-run", dest="dryRun", default=False, action="store_true")
args = parser.parse_args()
gh = Github(login_or_token=get_gh_token(token_file=expanduser("~/.github-token")))
cache = {"users": {}}
total_changes = 0
err_code = 0
for org_name in CMS_ORGANIZATIONS:
    if args.organization != "*" and org_name != args.organization:
        continue
    print("Wroking on Organization ", org_name)
    for inv in get_failed_pending_members(org_name):
        if ("failed_reason" in inv) and ("Invitation expired" in inv["failed_reason"]):
            print("  =>Deleting pending invitation ", inv["id"], inv["login"])
            if not args.dryRun:
                get_delete_pending_members(org_name, inv["id"])
                api_rate_limits(gh, msg=False)
    pending_members = []
    for user in get_pending_members(org_name):
        user = user["login"].encode("ascii", "ignore").decode()
        pending_members.append(user)
    print("Pending Invitations: %s" % ",".join(["@%s" % u for u in pending_members]))
    api_rate_limits(gh)
    org = gh.get_organization(org_name)
    ok_mems = []
    print("  Looking for owners:", REPO_OWNERS[org_name])
    chg_flag = 0
    for mem in org.get_members(role="admin"):
        login = mem.login.encode("ascii", "ignore").decode()
        if not login in cache["users"]:
            cache["users"][login] = mem
        if not login in REPO_OWNERS[org_name]:
            print("    =>Removing owner:", login)
            if not args.dryRun:
                try:
                    add_organization_member(org_name, login, role="member")
                    chg_flag += 1
                except Exception as ex:
                    print("  =>", ex)
                    err_code = 1
        else:
            ok_mems.append(login)
    for login in [l for l in REPO_OWNERS[org_name] if not l in ok_mems]:
        print("    =>Adding owner:", login)
        if not args.dryRun:
            add_organization_member(org_name, login, role="admin")
        chg_flag += 1
    total_changes += chg_flag
    if not chg_flag:
        print("    OK Owners")
    print("  Looking for teams:", list(REPO_TEAMS[org_name].keys()))
    org_repos = [repo for repo in org.get_repos()]
    teams = org.get_teams()
    chg_flag = 0
    for team in REPO_TEAMS[org_name]:
        flag = False
        for xteam in teams:
            if xteam.name == team:
                flag = True
                break
        if flag:
            continue
        print("    => Creating team", team)
        if not args.dryRun:
            create_team(org_name, team, "cmssw team for " + team)
            chg_flag += 1
    total_changes += chg_flag
    org_members = [mem.login.encode("ascii", "ignore").decode() for mem in org.get_members()]
    print("  All members: ", org_members)
    if chg_flag:
        teams = org.get_teams()
    for team in teams:
        xfile = "%s-%s.done" % (org_name, team.name)
        if exists(xfile):
            continue
        print("    Checking team:", team.name)
        api_rate_limits(gh, msg=False)
        team_info = {}
        try:
            team_info = REPO_TEAMS[org_name][team.name]
        except:
            print("    WARNING: New team found on Github:", team.name)
            err_code = 1
            continue
        members = team_info["members"]
        tm_members = [mem for mem in team.get_members()]
        tm_members_login = [mem.login.encode("ascii", "ignore").decode() for mem in tm_members]
        print("      Valid Members:", members)
        print("      Existing Members:", tm_members_login)
        ok_mems = ["*"]
        chg_flag = 0
        if not "*" in members:
            for mem in tm_members:
                api_rate_limits(gh, msg=False)
                login = mem.login.encode("ascii", "ignore").decode()
                if not login in cache["users"]:
                    cache["users"][login] = mem
                if login in members:
                    ok_mems.append(login)
                else:
                    if not args.dryRun:
                        team.remove_from_members(mem)
                    print("      =>Removed member:", login)
                    chg_flag += 1
            for login in [l for l in members if not l in ok_mems]:
                api_rate_limits(gh, msg=False)
                if login in pending_members:
                    print("    => Can not add member, pending invitation: %s" % login)
                    continue
                if login not in org_members:
                    print("      =>Inviting member:", login)
                    if not args.dryRun:
                        try:
                            add_organization_member(org_name, login, role="member")
                            chg_flag += 1
                        except Exception as ex:
                            print("  =>", ex)
                            err_code = 1
                    continue
                if not login in cache["users"]:
                    cache["users"][login] = gh.get_user(login)
                if not args.dryRun:
                    try:
                        team.add_to_members(cache["users"][login])
                    except Exception as e:
                        print(e)
                        err_code = 1
                print("      =>Added member:", login)
                chg_flag += 1
        total_changes += chg_flag
        if not chg_flag:
            print("      OK Team members")
        if not "repositories" in team_info:
            ref = open(xfile, "w")
            ref.close()
            continue
        team_repos = [repo for repo in team.get_repos()]
        team_repos_name = [repo.name.encode("ascii", "ignore").decode() for repo in team_repos]
        print("      Checking team repositories")
        print("        Valid Repos:", list(team_info["repositories"].keys()))
        print("        Existing Repos:", team_repos_name)
        repo_to_check = team_repos_name[:]
        for repo in team_info["repositories"]:
            if repo == "*":
                repo_to_check += [r.name.encode("ascii", "ignore").decode() for r in org_repos]
            elif repo.startswith("!"):
                repo_to_check += [repo[1:]]
            else:
                repo_to_check += [repo]
        chg_flag = 0
        for repo_name in set(repo_to_check):
            api_rate_limits(gh, msg=False)
            inv_repo = "!" + repo_name
            repo = [
                r for r in team_repos if r.name.encode("ascii", "ignore").decode() == repo_name
            ]
            if (repo_name in team_info["repositories"]) or (
                "*" in team_info["repositories"] and (not inv_repo in team_info["repositories"])
            ):
                prem = "pull"
                if repo_name in team_info["repositories"]:
                    prem = team_info["repositories"][repo_name]
                elif "*" in team_info["repositories"]:
                    prem = team_info["repositories"]["*"]
                if not repo_name in team_repos_name:
                    if not args.dryRun:
                        if not repo:
                            repo.append(gh.get_repo(org_name + "/" + repo_name))
                        team.set_repo_permission(repo[0], prem)
                    print("        =>Added repo:", repo_name, prem)
                    chg_flag += 1
                else:
                    curperm = repo[0].permissions
                    set_perm = False
                    curperm_name = "pull"
                    if curperm.admin:
                        curperm_name = "admin"
                        if not prem == "admin":
                            set_perm = True
                    elif curperm.push:
                        curperm_name = "push"
                        if not prem == "push":
                            set_perm = True
                    elif curperm.pull:
                        if not prem == "pull":
                            set_perm = True
                    if set_perm:
                        if not args.dryRun:
                            if not repo:
                                repo.append(gh.get_repo(org_name + "/" + repo_name))
                            team.set_repo_permission(repo[0], prem)
                        print("        =>Set Permission:", repo_name, curperm_name, "=>", prem)
                        chg_flag += 1
            elif repo_name in team_repos_name:
                if not args.dryRun:
                    if not repo:
                        repo.append(gh.get_repo(org_name + "/" + repo_name))
                    team.remove_from_repos(repo[0])
                print("        =>Removed repository:", repo_name)
                chg_flag += 1
        if not chg_flag:
            print("        OK Team Repositories")
        total_changes += chg_flag
        ref = open(xfile, "w")
        ref.close()

print("Total Updates:", total_changes)
exit(err_code)