view tests/list-tree.py @ 46035:6c960b708ac4

upgrade: display the list of processed revlog before proceeding

This help to make sure we don't wrongly skip some in the test and to make sure
the user is aware of the amount of processing they is signing up for.

Differential Revision: https://phab.mercurial-scm.org/D9469

from __future__ import (
    absolute_import,
    print_function,
)

import argparse
import os

ap = argparse.ArgumentParser()
ap.add_argument('path', nargs='+')
opts = ap.parse_args()


def gather():
    for p in opts.path:
        if not os.path.exists(p):
            return
        if os.path.isdir(p):
            yield p + os.path.sep
            for dirpath, dirs, files in os.walk(p):
                for d in dirs:
                    yield os.path.join(dirpath, d) + os.path.sep
                for f in files:
                    yield os.path.join(dirpath, f)
        else:
            yield p


print('\n'.join(sorted(gather(), key=lambda x: x.replace(os.path.sep, '/'))))