add_quick_script.py

#!/usr/bin/python3
"""
Quickly add a script to `~/.local/bin` as a symbolic link 
"""
import os
import stat
from pathlib import Path

YES = ("Y", "y")
RENAME = ("R", "r")

BIN = Path("~/.local/bin").expanduser()


def add_script_to_path(script_path: str) -> int:
    proper_path = Path(script_path).expanduser().resolve()
    final_path = BIN / Path(script_path).stem
    if not proper_path.exists():
        raise Exception(f"{proper_path} is not a valid path!")
    if final_path.exists():
        raise Exception(f"{final_path} already exists!")
    if not os.access(proper_path, os.X_OK):
        # Give the file the proper access
        proper_path.chmod(proper_path.stat().st_mode | stat.S_IEXEC)

    return io_helper(proper_path, final_path)


def io_helper(proper_path: Path, final_path: Path):
    # Finally, create symlink
    print("Creating the following Symlink:")
    print(f"{proper_path} -> {final_path}")
    print("[Y/y]?")
    print("[R/r] rename?")
    inpt = input()
    if inpt in YES:
        os.symlink(proper_path, final_path)
        return 0
    if inpt not in RENAME:
        print("Goodbye")
        return 1
    return io_helper(
        proper_path,
        BIN / input("Rename the final bit to:\n"),
    )


if __name__ == "__main__":
    import sys
    from argparse import ArgumentParser

    parser = ArgumentParser()
    parser.add_argument("script", help="File to symlink")

    sys.exit(add_script_to_path(parser.parse_args().script))