48 lines
1.5 KiB
Python
48 lines
1.5 KiB
Python
|
#!/usr/bin/env python
|
||
|
|
|
||
|
|
"""\
|
||
|
|
py.cleanup [PATH]
|
||
|
|
|
||
|
|
Delete pyc file recursively, starting from PATH (which defaults to the current
|
||
|
|
working directory). Don't follow links and don't recurse into directories with
|
||
|
|
a ".".
|
||
|
|
"""
|
||
|
|
import py
|
||
|
|
|
||
|
|
def main():
|
||
|
parser = py.std.optparse.OptionParser(usage=__doc__)
|
||
|
|
parser.add_option("-e", "--remove", dest="ext", default=".pyc", action="store",
|
||
|
|
help="remove files with the given comma-separated list of extensions"
|
||
|
|
)
|
||
|
|
parser.add_option("-n", "--dryrun", dest="dryrun", default=False,
|
||
|
|
action="store_true",
|
||
|
|
help="display would-be-removed filenames"
|
||
|
|
)
|
||
|
|
parser.add_option("-d", action="store_true", dest="removedir",
|
||
|
|
help="remove empty directories")
|
||
|
|
(options, args) = parser.parse_args()
|
||
|
|
if not args:
|
||
|
|
args = ["."]
|
||
|
|
ext = options.ext.split(",")
|
||
|
|
def shouldremove(p):
|
||
|
|
return p.ext in ext
|
||
|
|
|
||
|
|
for arg in args:
|
||
|
|
path = py.path.local(arg)
|
||
|
py.builtin.print_("cleaning path", path, "of extensions", ext)
|
||
|
|
for x in path.visit(shouldremove, lambda x: x.check(dotfile=0, link=0)):
|
||
|
|
remove(x, options)
|
||
|
|
if options.removedir:
|
||
|
|
for x in path.visit(lambda x: x.check(dir=1),
|
||
|
|
lambda x: x.check(dotfile=0, link=0)):
|
||
|
|
if not x.listdir():
|
||
|
|
remove(x, options)
|
||
|
|
|
||
|
|
def remove(path, options):
|
||
|
|
if options.dryrun:
|
||
|
|
py.builtin.print_("would remove", path)
|
||
|
|
else:
|
||
|
|
py.builtin.print_("removing", path)
|
||
|
|
path.remove()
|
||
|
|
|