Jack Jansen | f1148f0 | 2002-08-09 10:17:28 +0000 | [diff] [blame] | 1 | #!/usr/local/bin/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:]: |
| 12 | if os.name == 'mac': |
| 13 | import EasyDialogs |
| 14 | dir = EasyDialogs.AskFolder(message='Directory to zap pyc files in') |
| 15 | if not dir: |
| 16 | sys.exit(0) |
| 17 | zappyc(dir) |
| 18 | else: |
| 19 | print 'Usage: zappyc dir ...' |
| 20 | sys.exit(1) |
| 21 | for dir in sys.argv[1:]: |
| 22 | zappyc(dir) |
Jack Jansen | dfebb27 | 2001-01-23 22:45:52 +0000 | [diff] [blame] | 23 | |
| 24 | def zappyc(dir): |
Tim Peters | 182b5ac | 2004-07-18 06:16:08 +0000 | [diff] [blame] | 25 | os.path.walk(dir, walker, None) |
| 26 | |
Jack Jansen | dfebb27 | 2001-01-23 22:45:52 +0000 | [diff] [blame] | 27 | def walker(dummy, top, names): |
Tim Peters | 182b5ac | 2004-07-18 06:16:08 +0000 | [diff] [blame] | 28 | for name in names: |
| 29 | if name[-4:] in ('.pyc', '.pyo'): |
| 30 | path = os.path.join(top, name) |
| 31 | print 'Zapping', path |
| 32 | if doit: |
| 33 | os.unlink(path) |
| 34 | |
Jack Jansen | dfebb27 | 2001-01-23 22:45:52 +0000 | [diff] [blame] | 35 | if __name__ == '__main__': |
Tim Peters | 182b5ac | 2004-07-18 06:16:08 +0000 | [diff] [blame] | 36 | main() |