~abentley/bzrtools/bzrtools.dev

def import_version(output_dir, version, fancy=True, fast=False, verbose=False, dry_run=False, max_count=None):

    Cleaning up

        for result in iter_import_version(output_dir, version, tempdir,

                fast=fast, verbose=verbose, dry_run=dry_run, max_count=max_count):

        print 'Cleaning up'

def iter_import_version(output_dir, version, tempdir, fast=False, verbose=False, dry_run=False, max_count=None):

    if max_count:

        ancestors = ancestors[:max_count]

 

    # Not sure if I want this output. basically it tells you ahead of time

    # what it is going to do, but then later it tells you as it is doing it.

    # what probably would be best would be to collapse it into ranges, so that

    # this gives the simple view, and then later it gives the blow by blow.

    #if verbose:

    #    print 'Adding the following revisions:'

    #    for a in ancestors:

    #        print '\t%s' % a

 

    previous_version=None

        if verbose:

            version = str(revision.version)

            if version != previous_version:

                clear_progress_bar()

                print '\rOn version: %s' % version

            yield Progress(str(revision.patchlevel), i, len(ancestors))

            previous_version = version

        else:

            yield Progress("revisions", i, len(ancestors))

    if dry_run:

        print '**Dry run, not modifying output_dir'

        return

def main(args):

    """Just the main() function for this script.

 

    By separating it into a function, this can be called as a child from some other

    script.

 

    :param args: The arguments to this script. Essentially sys.argv[1:]

    """

    import optparse

    parser = optparse.OptionParser(usage='%prog [options] VERSION OUTDIR'

        '\n  VERSION is the arch version to import.'

        '\n  OUTDIR can be an existing directory to be updated'

        '\n         or a new directory which will be created from scratch.')

    parser.add_option('--verbose', action='store_true'

        , help='Get chatty')

 

    g = optparse.OptionGroup(parser, 'Progress options', 'Control how progress is given')

    g.add_option('--not-fancy', dest='fancy', action='store_false')

    g.add_option('--fancy', dest='fancy', action='store_true', default=True

        , help='Fancy or simple progress bar. (default: fancy)')

    parser.add_option_group(g)

 

    g = optparse.OptionGroup(parser, 'Test options', 'Options useful while testing process.')

    g.add_option('--test', action='store_true'

        , help='Run the self-tests and exit.')

    g.add_option('--dry-run', action='store_true'

        , help='Do the update, but don\'t copy the result to OUTDIR')

    g.add_option('--max-count', type='int', metavar='COUNT', default=None

        , help='At most, add COUNT patches.')

    g.add_option('--safe', action='store_false', dest='fast')

    g.add_option('--fast', action='store_true', default=False

        , help='By default the .bzr control directory will be copied, so that an error'

        ' does not modify the original. --fast allows the directory to be renamed instead.')

    parser.add_option_group(g)

 

    (opts, args) = parser.parse_args(args)

 

    if opts.test:

        print "Running tests"

        import doctest

        nfail, ntests = doctest.testmod(verbose=opts.verbose)

        if nfail > 0:

            return 1

        else:

            return 0

 

    if len(args) != 2:

        print 'Invalid number of arguments, try --help for more info'

        return 1

    else:

        try:

            output_dir = os.path.realpath(args[1])

            import_version(output_dir, pybaz.Version(args[0]),

                verbose=opts.verbose, fast=opts.fast,

                fancy=opts.fancy, dry_run=opts.dry_run,

                max_count=opts.max_count)

            return 0

        except UserError, e:

            print e

            return 1

        except KeyboardInterrupt:

            print "Aborted."

            return 1

 

 

if __name__ == '__main__':

    sys.exit(main(sys.argv[1:]))