AddOrganizationMembers.py

#!/usr/bin/env python3

"""
mass add users to organization

    python AddOrganizationMembers.py my.xlsx ~/.ssh/orgOauth myorg 4

Oauth permissions
https://docs.github.com/en/rest/repos/repos?apiVersion=2022-11-28
https://github.com/settings/tokens/new

* fine-grained: Oauth "Metadata" repository permissions (read)
  -- but needs to be for org. Classic may be better
* classic: Oauth "write:org" and public_repo (or repo for private) permissions
"""

import pandas
import github
from gitbulk import check_api_limit, connect
from pathlib import Path
from argparse import ArgumentParser


def adder(users: pandas.DataFrame, op, sess) -> None:
    members = [m.login for m in op.get_members()]
    invited = [m.login for m in op.invitations()]

    for u in users.values:
        login = u.strip()
        if login in members or login in invited:
            continue

        try:
            user = sess.get_user(login)
        except github.GithubException:
            raise ValueError(f"unknown GitHub username {login}")

        op.add_to_members(user)
        print(f"invited: {login}")


p = ArgumentParser(description="mass add team members to repos, optionally creating new repos")
p.add_argument("fn", help=".xlsx with group info")
p.add_argument("oauth", help="Oauth file")
p.add_argument("orgname", help="Github Organization")
p.add_argument("col", help="column for GitHub Username", type=int)
P = p.parse_args()

fn = Path(P.fn).expanduser()

match fn.suffix:
    case ".xls" | ".xlsx":
        users = pandas.read_excel(fn, usecols=[P.col]).squeeze().dropna()
    case ".csv":
        users = pandas.read_csv(fn, usecols=[P.col]).squeeze().dropna()
    case _:
        raise ValueError(f"Unknown file type {fn}")

if not users.ndim == 1:
    raise ValueError("need to have member names. Check that -col argument matches spreadsheet.")
# %%
op, sess = connect(P.oauth, P.orgname)
check_api_limit(sess)

adder(users, op, sess)