Ejercicio 12b

exercises/ex12/rm

@@ -1,12 +1,27 @@
 #!/usr/bin/env python
-# Exercise 12a
-# ls command
+# Exercise 12b
+# rm command
 
+import sys
+import shutil
 import argparse
 from pathlib import Path
 
 parser = argparse.ArgumentParser()
-parser.add_argument("paths", default=Path().cwd(), nargs="*", type=Path)
+parser.add_argument("paths", nargs="+", type=Path)
 parser.add_argument("-r", "--recursive", action="store_true")
 args = parser.parse_args()
 result = ""
+error = False
+
+for path in args.paths:
+    if path.is_dir() and args.recursive is False:
+        print(f"rm: no se puede borrar '{path}': Es un directorio")
+        error = True
+    elif path.is_file():
+        path.unlink()
+    else:
+        shutil.rmtree(path)
+
+if error:
+    sys.exit(1)