PythonStudy_生成‘随机验证码’代码

# 生成一个随机验证码：由字母和数字随机搭配
def random_code(n=8):
    '''
    生成一个随机验证码，由字母和数字随机搭配，字母区分大小写。
    :param n: 指定随机验证码位数
    :return: 生成的随机验证码
    '''
    import random
    random_code = ""                # 随机验证码
    for i in range(n):              # 按位生成随机码，共8位
        flag = random.randrange(3)
        if flag == 0:               # 0-9的随机数字
            random_code += str(random.randrange(0,10))
        elif flag == 1:             # 小写英文字母
            random_abc = chr(random.randrange(97,122))
            random_code += random_abc
        else:                       # 大写英文字母
            random_ABC = chr(random.randrange(65,91))
            random_code += random_ABC
    return random_code
if __name__ == '__main__':
    print(random_code(6))