Benjamin Peterson | 9cf41d0 | 2010-03-11 22:33:25 +0000 | [diff] [blame] | 1 | #!/usr/bin/env python |
Jack Jansen | 3337ea7 | 2002-09-06 20:23:09 +0000 | [diff] [blame] | 2 | """Recursively zap all .pyc and .pyo files""" |
Jack Jansen | dfebb27 | 2001-01-23 22:45:52 +0000 | [diff] [blame] | 3 | import os |
| 4 | import sys |
| 5 | |
Jack Jansen | f1148f0 | 2002-08-09 10:17:28 +0000 | [diff] [blame] | 6 | # set doit true to actually delete files |
| 7 | # set doit false to just print what would be deleted |
Jack Jansen | dfebb27 | 2001-01-23 22:45:52 +0000 | [diff] [blame] | 8 | doit = 1 |
| 9 | |
| 10 | def main(): |
Tim Peters | 182b5ac | 2004-07-18 06:16:08 +0000 | [diff] [blame] | 11 | if not sys.argv[1:]: |
Ronald Oussoren | 9545a23 | 2010-05-05 19:09:31 +0000 | [diff] [blame] | 12 | print 'Usage: zappyc dir ...' |
| 13 | sys.exit(1) |
Tim Peters | 182b5ac | 2004-07-18 06:16:08 +0000 | [diff] [blame] | 14 | for dir in sys.argv[1:]: |
| 15 | zappyc(dir) |
Jack Jansen | dfebb27 | 2001-01-23 22:45:52 +0000 | [diff] [blame] | 16 | |
| 17 | def zappyc(dir): |
Tim Peters | 182b5ac | 2004-07-18 06:16:08 +0000 | [diff] [blame] | 18 | os.path.walk(dir, walker, None) |
| 19 | |
Jack Jansen | dfebb27 | 2001-01-23 22:45:52 +0000 | [diff] [blame] | 20 | def walker(dummy, top, names): |
Tim Peters | 182b5ac | 2004-07-18 06:16:08 +0000 | [diff] [blame] | 21 | for name in names: |
| 22 | if name[-4:] in ('.pyc', '.pyo'): |
| 23 | path = os.path.join(top, name) |
| 24 | print 'Zapping', path |
| 25 | if doit: |
| 26 | os.unlink(path) |
| 27 | |
Jack Jansen | dfebb27 | 2001-01-23 22:45:52 +0000 | [diff] [blame] | 28 | if __name__ == '__main__': |
Tim Peters | 182b5ac | 2004-07-18 06:16:08 +0000 | [diff] [blame] | 29 | main() |