随机生成六位验证码函数版(python)

import random
def code(n=6,alpha=True):
    s = ‘‘ # 创建字符串变量,存储生成的验证码
    for i in range(n):  # 通过for循环控制验证码位数
        num = random.randint(0,9)  # 生成随机数字0-9
        if alpha: # 需要字母验证码,不用传参,如果不需要字母的,关键字alpha=False
            upper_alpha = chr(random.randint(65,90))
            lower_alpha = chr(random.randint(97,122))
            num = random.choice([num,upper_alpha,lower_alpha])
        s = s + str(num)
    return s
print(code(6,False))  # 打印6位数字验证码
print(code(6,True))   # 打印6位数字字母混合验证码
print(code(4,False))  # 打印4位数字验证码
print(code(4,True))   # 打印4位数字字母混合验证码