Exercise on www.practicepython.org

check_primality_func.py

def get_user_int(msg):
    '''
    Return integer from user.
    :param msg:
    :return user_input:
    '''
    user_input = input(msg)
    try:
        user_input = int(user_input)
    except:
        print("Something wrong...")
    return user_input

num = get_user_int("Input a number: ")

rng = range(1,num + 1)

new_rng = [i for i in rng if num % i == 0]

if len(new_rng) == 2:
    print("Entered number is prime.")
else:
    print("Entered number is not prime.")