判断是否为素数

def func(n):
    if 0<n<=2:
        print("{}是一个素数".format(n))
    i = 2
    while n>2:
        if n % i == 0:
            print("{}不是素数".format(n))
            break
        else:
            print("{}是一个素数".format(n))
            break
x = int(input("请输入需要判断的数字："))
func(x)

请输入需要判断的数字：7
7是一个素数

def demo(y):
    def inner(n):
        if 0<n<=2:
            result = "是"
        i = 2
        while n>2:
            if n % i == 0:
                result = "不是"
            break
        else:
            result = "不是"
        return y(result)
    return inner
@demo
def func(x):
    print("{}{}一个素数".format(n,x))
n = int(input("请输入一个数字:"))
func(n)

请输入一个数字:8
8不是一个素数