判断一个数是不是质数

n = int(input("请输入一个数："))
if n == 1:
    print("不是质数")
else:
    count = 2
    while count <= n-1:
        if n % count == 0:
            print("不是质数")
            break
        count += 1
    else:
        print("是质数")