diff options
Diffstat (limited to 'build/scripts/extract_docs.py')
| -rw-r--r-- | build/scripts/extract_docs.py | 23 |
1 files changed, 15 insertions, 8 deletions
diff --git a/build/scripts/extract_docs.py b/build/scripts/extract_docs.py index cdcf93105f..20e8311346 100644 --- a/build/scripts/extract_docs.py +++ b/build/scripts/extract_docs.py @@ -18,14 +18,21 @@ def main(): prefixes = ['{}{}'.format(os.path.normpath(p), os.path.sep) for p in args.skip_prefixes] - for src in filter(lambda(p): os.path.basename(p) == 'preprocessed.tar.gz', args.docs): - rel_dst = os.path.dirname(os.path.normpath(src)) - for prefix in prefixes: - if src.startswith(prefix): - rel_dst = rel_dst[len(prefix):] - break - assert not os.path.isabs(rel_dst) - dest_dir = os.path.join(args.dest_dir, rel_dst) + def _valid_docslib(path): + base = os.path.basename(path) + return base.endswith(('.docslib', '.docslib.fake')) or base == 'preprocessed.tar.gz' + + for src in [p for p in args.docs if _valid_docslib(p)]: + if src == 'preprocessed.tar.gz': + rel_dst = os.path.dirname(os.path.normpath(src)) + for prefix in prefixes: + if src.startswith(prefix): + rel_dst = rel_dst[len(prefix):] + continue + assert not os.path.isabs(rel_dst) + dest_dir = os.path.join(args.dest_dir, rel_dst) + else: + dest_dir = args.dest_dir if not os.path.exists(dest_dir): os.makedirs(dest_dir) with tarfile.open(src, 'r') as tar_file: |