python 实现随机验证码

import random


def generate_code(n=6, alpha=True):
    s = ''
    for i in range(n):
        num = str(random.randint(0, 9))
        if alpha:
            alpha_upper = chr(random.randint(97, 122))
            alpha_lower = chr(random.randint(65, 90))
            num = random.choice([alpha_lower, alpha_upper, num])
        s += num
    return s


print(generate_code())