写一个判素数的函数，在主函数输入一个整数，输出是否为素数的信息。

你可以这样写:

def is_prime(num):
    if num <= 1:
        return False
    for i in range(2, num):
        if num % i == 0:
            return False
    return True

def main():
    num = int(input("请输入一个整数: "))
    if is_prime(num):
        print(num, "是素数")
    else:
        print(num, "不是素数")

if __name__ == '__main__':
    main()