Python 判断质数

a = raw_input() #输入数字
a = int(a) #强制转换成int
b=True #一个标记
for i in range(2,a): #从2开始循环本身
    if a%i==0: #如果除了本身和1以外还能被整除
        b=False #标记改成False
        break #循环结束
if b:
    print 'YES'
else:
    print 'NO'