Python初学者：判断是否是素数

attention：可以一次处理多个数字。输入0结束

要求：每个整数输入完毕后

def prime(m):
    s=0
    for i in range(2,m):
        if m%i==0:
            s+=1
        else:
            s+=0
    if s==0:
        return 1
while True:
    n=int(input())
    if n==0:
        break
    if n<2:
        print("Invalid")#1不是素数
    else:
        if prime(n):
            print("Yes")
        else:
            print("No")

，马上输出该整数是否为素数。素数返回True，否则返回False