#! /usr/bin/python -tt

import yum
import sys

if len(sys.argv) != 3 or sys.argv[1] not in ('list','only', 'enable','disable'):
    if sys.argv:
        progname = "yum-repo-mgr"
    else:
        progname = sys.argv[0]
    msg = """Usage: %s "enable"|"disable" <repo-wildcard> """  % progname
    sys.stderr.write(msg)
    sys.exit(1)

yb = yum.YumBase()

if sys.argv[1] == 'list' and sys.argv[2] == 'enabled':
    arg_repos = yb.repos.listEnabled()
else:
    arg_repos = yb.repos.findRepos(sys.argv[2])

if sys.argv[1] == 'only':
    look_at_repos = yb.repos.findRepos('*')
else:
    look_at_repos = arg_repos

for repo in look_at_repos:
    if sys.argv[1] == 'list':
        if repo.isEnabled():
            print repo.id, "enabled"
        else:
            print repo.id, "disabled"
    elif sys.argv[1] == 'only':
        if repo.isEnabled():
            if repo not in arg_repos:
                repo.disablePersistent()
        else:
            if repo in arg_repos:
                repo.enablePersistent()
    elif sys.argv[1] == 'enable':
        if not repo.isEnabled():
            repo.enablePersistent()
        else:
            print "Repo %s already enabled" % repo.id
    else:
        if repo.isEnabled():
            repo.disablePersistent()
        else:
            print "Repo %s already disabled" % repo.id