教你用python写出产生指定长度的验证码

说明：
编写一个程序产生指定长度的验证码，默认长度为4位，验证码由大小写字母和数字构成，同时用户输入随机产生的验证码，进行用户验证，模拟登陆时验证码的验证过程。
代码如下：

import random


def generate_code(a = 4):
    all_chars = '0123456789abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ'
    generate_last = len(all_chars) - 1
    code = ''
    for _ in range(a):
        num = random.randint(0, generate_last)
        code += all_chars[num]
    return code

def main():
    s = generate_code()
    print(s)
    print('请输入验证码：')
    c = input()
    if c == s:
        print('验证成功！')
    else:
        print('验证失败！')

if __name__ == '__main__':
    main()