python 手機登陸驗證碼

# -*- coding: utf-8 -*-
"""
Created on Tue Mar 26 11:43:49 2019

@author: Alisonman
"""

import re                                                                   #引用re類
import random                                                          #引用random類

def phone_pat():                                                       #定義手機號碼驗證函數
phone_pat = re.compile('^(13\d|14[5|7]|15\d|166|17[3|6|7]18\d)\d{8}$')
while 1:
phone = input('請輸入您的手機號碼:\n')
res = re.search(phone_pat,phone)
if res:
print('正常手機號碼')
break
else:
print('不是手機號')
return phone

def alpha():                     #生成隨機驗證碼函數
a = ''.join(random.sample(
[str(random.randint(0, 9)) for _ in range(0,5)] +
[chr(random.randint(65, 90)) for _ in range(0,5)] +
[chr(random.randint(97,122)) for _ in range(0,5)] , 6))
print(a)
return a

def bingo(a):                #驗證用戶輸入的驗證碼是否正確（這里的驗證碼是區分大小寫的）
val = input("請輸入驗證碼：\n")
if ''.join(val) == a:
print("登陸成功")
return 1
else:
print("驗證碼錯誤")
return -1

def main():
phone_pat()
a = alpha()
bingo(a)

if __name__ == "__main__":
main()