view python/setup_repo.py @ 404:fb304dcd1e64

factor to its own method
author Jeff Hammel <jhammel@mozilla.com>
date Tue, 30 Jul 2013 10:35:47 -0700
parents 62eb670a408f
children 815eb5c796e9
line wrap: on
line source

#!/usr/bin/env python

"""
setup a repository

Parameters:
- host URL
- location on host disk
"""

import os
import subprocess
import sys
import urlparse

from optparse import OptionParser


# global variables
HOST='k0s.org'
HGRC="""[paths]
default = http://%(host)s/%(repo)s/%(name)s
default-push = ssh://%(host)s/%(repo)s/%(name)s
"""

call = subprocess.check_output

def init_repo(directory):
    """setup repository"""
    call(['hg', 'init', directory])
    call(['hg', 'add', '-R', directory])
    call(['hg', 'commit', '-m', options.message, '-R', directory])


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

    # parse command line arguments
    parser = OptionParser('%prog [options] directory')
    parser.add_option('-m', '--message', dest='message',
                      default='initial commit',
                      help='commit message [Default: %default]')
    parser.add_option('-u', '--remote-url', dest='remote_url',
                      default="http://k0s.org/hg/",
                      help="URL of host hg repository collection [Default: %default]")
    parser.add_option('-p', '--remote-path', dest='remote_path',
                      help="remote path of hg repository links; otherwise taken from --remote-url, if specified")
    parser.add_option
    options, args = parser.parse_args(args)
    if len(args) != 1:
        parser.print_usage()
        parser.exit()
    directory = args[0]

    # initialize repository
    init_repo(directory)

    # setup remote, if specified
    name = os.path.basename(directory)
    if options.remote_url:

        # parse remote URL
        host, netloc, path, query, anchor = urlparse.urlsplit(options.remote_url)
        if options.remote_path:
            remote_host = host
            if ':' in remote_path:
                remote_host, remote_path = remote_path.split(':', 1)
        else:
            remote_path = path

        # setup remote repository
        remote_dir = '%s/%s' % (path, name)
        call(['ssh', host, "mkdir -p %s && cd %s && hg init" % (remote_dir, remote_dir)])

        # write local .hgrc file
        # TODO: use ConfigParser
        with file(os.path.join(directory, '.hg', 'hgrc'), 'w') as f:
            print >> f, HGRC % { 'host': host, 'repo': repo, 'name': name}

        # push changes
        call(['hg', 'push', '-R', directory])

if __name__ == '__main__':
  main()