X-Git-Url: https://git.phdru.name/?p=ppu.git;a=blobdiff_plain;f=scripts%2Frm.py;h=b7219f6d20268d3fe8f74ba8077f3fc01a12ebfe;hp=4278fef5beba8752cabcccb60f7649c762db7bed;hb=129068b91b9f880672fe17c248c8085d8589fce7;hpb=c9b64e869f9e673b0d213be3f7d098ef00e4fe1c

diff --git a/scripts/rm.py b/scripts/rm.py
index 4278fef..b7219f6 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 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)