写一个判断素数的函数，在主函数输入一个整数，输出是否是素数的消息，如果是素数输出prime 如果不是输出not prime。python

可以使用如下代码实现：

def is_prime(n):
    """
    判断一个数是否是素数
    """
    if n < 2:
        return False
    for i in range(2, int(n ** 0.5) + 1):
        if n % i == 0:
            return False
    return True

# 主函数
num = int(input("请输入一个整数："))
if is_prime(num):
    print("prime")
else:
    print("not prime")

首先定义了一个函数 `is_prime`，用于判断一个数是否是素数。在主函数中，首先输入一个整数，然后调用 `is_prime` 函数进行判断，根据返回值输出相应的消息。