view python/hgrc.py @ 352:eeff298cc922

more crap
author Jeff Hammel <jhammel@mozilla.com>
date Thu, 27 Jun 2013 02:13:40 -0700
parents 971e7deca495
children ee8ab3de9c7f
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',
                      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`")
    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)

    # construct ConfigParser objects and
    # ensure that all the files are parseable
    config = {}
    for path in hgrc:
        config['path'] = ConfigParser()
        if isinstance(path, basestring):
            config['path'].read(path)
        
    if options.print:
        # print the chosen hgrc paths and you're done
        print '\n'.join(hgrc)
        parser.exit()

if __name__ == '__main__':
    main()