n位随机码

设计一个函数：生成指定长度的随机验证码
验证码由数字0-9和大小写26*2个英文字母构成

from random import choice
def generate_code(length):
    num = "qwertyuiopasdfghjklzxcvbnmQWERTYUIOPASDFGHJKLZXCVBNM0123456789"
    code =''
    for i in range(length):
        code += choice(num)
    return code
print(generate_code(6))