view python/hgrc.py @ 402:c00b70509fff

flesh out
author Jeff Hammel <jhammel@mozilla.com>
date Tue, 30 Jul 2013 03:32:31 -0700
parents ee8ab3de9c7f
children 6797477f6a8f
line wrap: on
line source

#!/usr/bin/env python

"""
Script for modifying hgrc files.

Actions:
(TBD)
"""
# imports
import optparse
import os
import sys
from ConfigParser import RawCOnfigParser as ConfigParser

def main(args=sys.argv[1:]):

    # command line parser
    usage = '%prog [options] repository <repository> <...>'
    parser = optparse.OptionParser(usage=usage, description=__doc__)
    parser.add_option('-p', '--print', dest='print_hgrc',
                      action='store_true', default=False,
                      help="print full path to hgrc files and exit")
    parser.add_option('--ssh', dest='default_push_ssh',
                      action='store_true', default=False,
                      help="use `default` entries for `default-push`")
    parser.add_option('--push', '--default-push', dest='default_push',
                      help="set [paths] default-push location")
    options, args = options.parse_args(args)
    if not args:
        parser.print_usage()
        parser.exit()

    # find all hgrc files
    hgrc = []
    missing = []
    not_hg = []
    not_a_directory = []
    errors = {'Missing path': missing,
              'Not a mercurial directory': not_hg,
              'Not a directory': not_a_directory,
              }
    for path in args:
        if not os.path.exists(path):
            missing.append(path)
        path = os.path.abspath(os.path.normpath(path))
        if os.path.isdir(path):
            basename = os.path.basename(path)
            subhgdir = os.path.join(path, '.hg') # hypothetical .hg subdirectory
            if basename == '.hg':
                hgrcpath = os.path.join(path, 'hgrc')
            elif os.path.exists(subhgdir):
                if not os.path.isdir(subhgdir):
                    not_a_directory.append(subhgdir)
                    continue
            else:
                not_hg.append(path)
                continue
        else:
            assert os.path.isfile(path), "%s is not a file, exiting" % path
        hgrc.append(path)

    # raise errors if encountered
    if sum(errors.values()):
        for key, value in errors.items():
            if value:
                print '%s: %s' % (key, ', '.join(value))
        parser.exit(1)

    # construct ConfigParser objects and
    # ensure that all the files are parseable
    config = {}
    for path in hgrc:
        config[path] = ConfigParser()
        if isinstance(path, basestring):
            if os.path.exists(path):
                config[path].read(path)
        else:
            # XXX this code path is untenable
            config[path].

    if options.print_hgrc:
        # print the chosen hgrc paths and you're done
        print '\n'.join(hgrc)
        parser.exit()

if __name__ == '__main__':
    main()