This function gets an input, and if it is a whole number between 1 and 10, it puts the value inuser_balance

01User_Balance.py

"""
Component 1 of Lucky Unicorn
This function gets an input, and if it is a whole number between 1 and 10, it
puts the value in user_balance
Otherwise, it raises errors and tells the user that a invalid input is entered
until a valid input is entered.
Written by Nathan Smith
28/06/21
"""


def check():
    print('How much money would you like to play with?')
    while True:
        user_balance = input('Please enter a whole number from 1-10: ')
        try:
            user_balance = int(user_balance)
        except ValueError:
            pass
        else:
            if 0 < user_balance < 11:
                break
            else:
                pass


check()