python练习：编写函数，输入数字，判断是否是素数

import math

flage = True

num = int(input("请输入一个数字："))

if num == 2:

    print("2是一个素数")

else:

    for i in range(2,int(math.sqrt(num)+1)):

        if num%i == 0:

            flage = False

            break

    if flage == True:

        print("%d是一个素数"%(num))

    else:

        print("%d不是一个素数"%(num))