Add option -r for rm.py

[ppu.git] / scripts / rm.py

diff --git a/scripts/rm.py b/scripts/rm.py
index 4278fef5beba8752cabcccb60f7649c762db7bed..b7219f6d20268d3fe8f74ba8077f3fc01a12ebfe 100755 (executable)
--- a/scripts/rm.py
+++ b/scripts/rm.py
@@ -1,7 +1,22 @@
 #! /usr/bin/env python
 
+import argparse
 import os
-import sys
+import shutil
 
-for filename in sys.argv[1:]:
-    os.unlink(filename)
+if __name__ == '__main__':
+    parser = argparse.ArgumentParser(description='Remove old files')
+    parser.add_argument('-r', '--recursive', action='store_true',
+                        help='remove directories recursively')
+    parser.add_argument('names', nargs='+',
+                        help='files/directories names to remove')
+    args = parser.parse_args()
+
+    for name in args.names:
+        if os.path.isdir(name):
+            if args.recursive:
+                shutil.rmtree(name)
+            else:
+                os.rmdir(name)
+        else:
+            os.unlink(name)