Enter a number to determine whether it is a prime number

Requirement: Enter a number to determine whether it is a prime number
Analysis: 3 5 7 11 Such a number is a prime number

code:

import math
def is_prime_number(num:int):
    '''
    判断输入的数是否是质数
    :param num: 输入的数字
    :return: 布尔类型的值，True，False
    '''
    for i in range(2, int(math.sqrt(num)) + 1):
        # 判断是否能被整除
        if num % i == 0:
            return False
    # 从2到num-1都没有被整除
    return True


if __name__ == '__main__':
    num = int(input(('输入一个大于1的整数：')))
    is_prime = is_prime_number(num)
    print(is_prime)