task.mk/src/confirm.py

#% extends "py-script.mk" %#
#% block name %#confirm#% endblock %#
#% block script %#

import sys

##- '$(ansi_py)' -##


def confirm():
    """
    Ask user to enter Y or N (case-insensitive).
    :return: True if the answer is Y.
    :rtype: bool
    """
    answer = ""
    while answer not in ["y", "n"]:
        answer = input(f"""$(2) {a.b_red}[Y/n]{a.end} """).lower()
    return answer == "y"


if confirm():
    sys.exit(0)
else:
    sys.exit(1)
#% endblock %#