diff options
Diffstat (limited to 'PCbuild/rmpyc.py')
-rw-r--r-- | PCbuild/rmpyc.py | 27 |
1 files changed, 14 insertions, 13 deletions
diff --git a/PCbuild/rmpyc.py b/PCbuild/rmpyc.py index 019c69b..43c8576 100644 --- a/PCbuild/rmpyc.py +++ b/PCbuild/rmpyc.py @@ -1,23 +1,24 @@ # Remove all the .pyc and .pyo files under ../Lib. + def deltree(root): import os - def rm(path): - os.unlink(path) + from os.path import join + npyc = npyo = 0 - dirs = [root] - while dirs: - dir = dirs.pop() - for short in os.listdir(dir): - full = os.path.join(dir, short) - if os.path.isdir(full): - dirs.append(full) - elif short.endswith(".pyc"): + for root, dirs, files in os.walk(root): + for name in files: + delete = False + if name.endswith('.pyc'): + delete = True npyc += 1 - rm(full) - elif short.endswith(".pyo"): + elif name.endswith('.pyo'): + delete = True npyo += 1 - rm(full) + + if delete: + os.remove(join(root, name)) + return npyc, npyo npyc, npyo = deltree("../Lib") |