生成随机数字验证码或者数字字母混合验证码

import random
#完整版 函数生成验证码 =========字母或数字字母验证码
def code(n=6,alpha = True):
    s =''
    for i in range(n):
        num = str(random.randint(0,9))
        if alpha:
            alpha_upper = chr(random.randint(65, 90))
            alpha_lower = chr(random.randint(97,122))

            num = random.choice([alpha_upper, alpha_lower, num])
        s += num
    return s
print(code())
print(code(4,False))