A wrapper around `rm` command to prevent disasters

safe-rm.py

#!/usr/bin/env python3

"""
A wrapper around `rm` command to prevent disasters.

1. Put this script inside `~/.local/bin` (or anywhere on your system)
2. Enable the executable permission of the script using `$ chmod u+x safe-rm.py`
3. Add `alias rm=~/.local/bin/safe-rm.py` in your `~/.bashrc`
4. When removing 3+ files, confirm prompt with a list of targets to be removed will show up
"""

import subprocess
import sys

import click
import pprint


def main():
    # list of arguments without `-` prefix (not an option)
    args = [arg for arg in sys.argv[1:] if not arg.startswith("-")]

    # confirm prompt when removing more then 3
    if len(args) >= 3:
        print(f"You're about to remove:")
        pprint.pprint(args, indent=1)
        click.confirm("Do you want to continue?", default=True, abort=True)

    # run `rm` with same arguments
    result = subprocess.run(
        ["rm", *sys.argv[1:]],
        stdout=subprocess.PIPE,
        stderr=subprocess.STDOUT,
    )
    print(result.stdout.decode())


if __name__ == "__main__":
    try:
        main()
    except click.Abort:
        print("Aborted")