Refactor `input()`

[maryjess]

Feasible option(s):

• Encapsulate input() in a function
• Add input validation
• Typecast
• Default input value(s)
• Using decorators

[maryjess]

• Using decorators

def input_validation(validation_func):
    def decorator(func):
        def wrapper(*args, **kwargs):
            while True:
                value = func(*args, **kwargs)
                if validation_func(value):
                    return value
                else:
                    print("Invalid input, please try again.")
        return wrapper
    return decorator

@input_validation(lambda x: x.isdigit())
def get_input(prompt):
    return input(prompt)

age = get_input("Enter your age: ")

— decorators.py