python随机生成6位验证码

isnum用来控制是纯数字还是数字加字母组合

import random
def generate_verification_code(isnum):
    ''' 随机生成6位的验证码 '''
    code_list = []
    if isnum==True:
        for i in range(10): # 0-9数字
            code_list.append(str(i))
    else:
        for i in range(10): # 0-9数字
            code_list.append(str(i))
        for i in range(65, 91): # A-Z
            code_list.append(chr(i))
        for i in range(97, 123): # a-z
            code_list.append(chr(i))
    myslice = random.sample(code_list, 6)  # 从list中随机获取6个元素，作为一个片断返回
    verification_code = ''.join(myslice) # list to string
    return verification_code