4 from fnmatch import fnmatch
7 parser = argparse.ArgumentParser(description='Generate sitemap')
8 parser.add_argument('-x', '--exclude', action='append',
9 help='exclude directories (pattern)')
10 parser.add_argument('root_dir', help='Root dicrectory')
11 args = parser.parse_args()
14 for pat in args.exclude:
16 if not pat.endswith('/*'):
17 exclude.append(pat + '/*')
19 os.chdir(args.root_dir)
20 fullpath = os.getcwd()
21 fp_len = len(fullpath)
23 for dirpath, dirs, files in sorted(os.walk(fullpath)):
25 dirpath = dirpath[fp_len:]
29 if fnmatch(dirpath, pat):
31 parts = dirpath.split('/')
32 level = len(parts) - 2
33 print ' ' * level + parts[-1]