Use os.walk() to find files to delete.

This commit is contained in:
Tim Peters 2003-04-26 00:53:24 +00:00
parent 2f36b3e205
commit 8b7beb631b
1 changed files with 14 additions and 13 deletions

View File

@ -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")