python隨機驗證碼（數字和字母組合）

import random

# 生成隨機驗證碼
# 數字驗證碼

check_code = ''

for i in range(4):
    current = random.randint(1, 9)
    check_code += str(current)

print(check_code)

# 隨機字母+數字驗證碼

check_code = ''

for i in range(4):
    current = random.randrange(0, 4)
    if current == i:
        tmp = chr(random.randint(65, 90))
    else:
        tmp = random.randint(0, 9)
    check_code += str(tmp)
print(check_code)