随机验证码——python

'''

    功能：随机验证码
    作者：曾朝阳
    日期：01/22/2019
    注意：
        randrange()返回一个递增集合的随机数，使用它必须导入random包
        randint()返回一个随机数
        chr()返回一个字符，以整数为参数


'''
import random
def check_code():
    check_code = ''
    for i in range(4):
        current = random.randrange(0, 4)
        if current != i:
            temp = chr(random.randint(65, 90))
        else:
            temp = random.randint(0, 9)
        check_code += str(temp)
    return check_code


def main():
    y_or_n = input("是否退出程序（y or n）:")
    while y_or_n == 'n':
        code = check_code()
        print ("验证码>>>" + code)
        get_input = input(">>>")
        if get_input.upper() == code:
            print ("Successfully!")
        y_or_n = input("是否退出程序（y or n）")


if __name__ == '__main__':
    main()