summaryrefslogtreecommitdiffstats
path: root/PCbuild/rmpyc.py
diff options
context:
space:
mode:
Diffstat (limited to 'PCbuild/rmpyc.py')
-rw-r--r--PCbuild/rmpyc.py27
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")