view tests/list-tree.py @ 45174:f93a4e3d35ab

osutil: fix excessive decref on tuple creation failure in listdir()

The stat object would be freed on error for the same reason as the previous
patch. makestat() can be inlined, but this patch doesn't change it.

https://github.com/python/cpython/blob/2.7/Python/modsupport.c#L292

The __APPLE__ code is untested.

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, '/'))))