more-python/exercises/ex12/rm

27 lines
583 B
Python
Executable File

#!/usr/bin/env python
# Exercise 12b
# rm command
import sys
import shutil
import argparse
from pathlib import Path
parser = argparse.ArgumentParser()
parser.add_argument("paths", nargs="+", type=Path)
parser.add_argument("-r", "--recursive", action="store_true")
args = parser.parse_args()
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)