X-Git-Url: https://git.phdru.name/?a=blobdiff_plain;f=scripts%2Frm.py;h=4f24b7200ab39a4ef92d57ca4a881633736097c5;hb=45a6d6e1d9551954e827efe7eaaf119315048248;hp=4278fef5beba8752cabcccb60f7649c762db7bed;hpb=f8dcc3f8ab3d1487c0e9042e26e437db77c4e311;p=ppu.git

diff --git a/scripts/rm.py b/scripts/rm.py
index 4278fef..4f24b72 100755
--- 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 files/directories')
+    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)