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()