mk_group_repos.py

#!/usr/bin/env python3

import sys
import boatswain_env as benv
import argparse
import csv
from github import Github

from add_collaborator import do_add_collaborator
from mk_repo import do_mk_repo, fmt_hyphen

CMD_NAME = 'mk_group_repos'
DESC = 'Make repos for groups'

def mk_group_repos_deco(parser):
    parser.add_argument('org',
                        type=str,
                        help='repo owner organization',
                        metavar='<org>',
    )

    parser.add_argument('prefix',
                        type=str,
                        help='prefix prepended to each repo',
                        metavar='<prefix>',
    )

    parser.add_argument('groups',
                        type=argparse.FileType('r'),
                        help='path of list of groups (group,[member,])',
                        metavar='<groups.csv>',
    )

    parser.add_argument('permission',
                        type=str,
                        help='permissions of each user to be added',
                        metavar='push|pull|admin|none',
    )

    parser.add_argument('-c', '--create',
                        default=False,
                        action='store_true',
                        help='create repo first',
    )


def mk_group_repos(opt):
    g = Github(opt.githubToken())
    org = g.get_organization(opt.org)

    if not opt.promptYes(('Are you sure you would like to create group repos '
                            'for users in {} under org {} with name {}')
                            .format(opt.groups.name, opt.org,
                                fmt_hyphen(opt.prefix, '<group>')),
                        True):
        opt.warn('Aborting')
        return

    opt.info('Creating repos under {} for groups in {}, with name {}'
            .format(opt.groups.name, opt.org,
                fmt_hyphen(opt.prefix, '<group>')))

    for g in csv.reader(opt.groups):
        group, members = g[0], [m.strip() for m in g[1:] if m != '']

        repo_name = fmt_hyphen(opt.prefix, group)

        if opt.create:
            action = 'creating'
        else:
            action = 'looking up'
        opt.info('{} {}/{} and adding members {}'
                .format(action, opt.org, repo_name, members))

        if opt.create:
            repo = do_mk_repo(org, repo_name, opt)
        else:
            repo = org.get_repo(repo_name)
            opt.info('Looked up repo {}/{}'.format(org.name, repo_name))

        if opt.permission == 'none':
            opt.info('Not adding {} to {}'.format(members, repo_name))
        else:
            opt.info('Adding {} to {}'.format(members, repo_name))
            for member in members:
                do_add_collaborator(repo, member, opt.permission, opt)


def main(args=None, config_path=None):
    if args is None:
        args = sys.argv[1:]

    opt = benv.ParseOption(args, section=CMD_NAME, config_path=config_path,
            desc=DESC, parse_deco=mk_group_repos_deco, req_github=True)

    mk_group_repos(opt)


if __name__ == '__main__':
    main()