Python实现简单的登录接口

#!/usr/bin/env python
#-*- coding:utf-8 -*-

import getpass                                                              #调用getpass模块，避免用户输入的密码展示在命令行
import time                                                                 #调用time时间模块


def login():                                                                #定义函数login，功能：注册时使用
    print('\033[32m+\033[0m' * 20)
    print('\033[32m欢迎来到注册中心!!\033[0m')
    print('\033[32m+\033[0m' * 20)
    n = []                                                                  #定义空列表
    User=input('User:').strip()                                             #定以变量User，记录用户输入
    Passwd=getpass.getpass('Passwd:').strip()                               #同上
    n.append(User)                                                          #将User、Passwd添加到n列表里
    n.append(Passwd)
    Name=':'.join(n)                                                        #使用join函数将列表转换成字符串以：号分割
    UserFile.write('%s \n' % Name)                                          #写到用户的文件里

while True:                                                                 #定义while循环，只有条件不成立时才会退出循环    
    UserFile=open('/home/haifeng/python_script/day1/user.txt','r+')         
    LockFile=open('/home/haifeng/python_script/day1/lock.txt','r+')         #以读写的方式打开文件
    User={}                                                                 #定义一个空字典
    for Line in UserFile.readlines():                                       #将文件全部一行一行的读取出来，定义为Line
        key,value=Line.split(':')                                           #定义字典的key,value 以：分割
        User[key]= value                                                    #给字典的key赋值，{'alex': '123\n', 'jack': '456\n'}
    LockUser=[]                                                             #定义一个空列表
    for a in LockFile.readlines():                                          #将锁文件里的内容一行一行读进内存
        LockLine=a.strip()                                                  #去掉空格
        LockUser.append(LockLine)                                           #添加到LockUser列表里

    break_flag=False                                                        #定义一个跳出循环的flag，当子循环要求父循环一起跳出循环时，值为True
    UserName=input('请输入用户名:').strip()                                 #记录用户名输入
    if len(UserName)==0:                                                    #判断输入是否为空，空则跳出当前循环，进入下一次循环
        print('\033[31m你没有输入，请输入正确用户名\033[0m')
        continue
    elif UserName not in User:                                                    #如果用户输入用户名不在User里，会提示用户是否注册新用户
        Num=input('\033[31m无效用户!! 是否注册新用户: y or n:\033[0m').strip()    
        if Num == 'y':                                                            #如果用户输入y就是调用上面的login函数注册新用户，n则跳出当前循环
            time.sleep(1)
            login()
            print('\033[32m注册成功!!!\033[0m')
        else:
           print('请重新输入!!!')
           continue
    elif UserName in LockUser:                                                    #如果用户输入用户名在LockUser里直接退出
        print('账户已经被锁定，请联系管理员进行解锁!!!')        
        break
    else:
        Num = 1                                                                   #定义计数器
        for i in range(3):                                                        #for循环三次
           PassWord=getpass.getpass('请输入密码:').strip()                        #提示用户输入密码，为空重新输入
           if len(PassWord)==0:
               print('密码不能为空!!')
               continue
           elif  PassWord in User[UserName]:                                      #匹配用户名密码
              print("\033[32m登录成功!祝你购物愉快!!!\033[0m")
              break_flag=True                                                     #登录成功，退出子循环
              break
           else:
              print('密码错误,三次以后账号锁定,你已使用%s次.' % Num)
           Num += 1                                                               #相反密码输入错误就会自加1,当Num等于4的时候账户锁定
           if Num == 4:
              print('账户锁定')
              LockFile.write('%s \n' % UserName)                                  #将锁定用户写入LockFile里
              break_flag=True                                                     #退出子循环
              break
        if break_flag:                                                            #判断子循环是否退出，退出则一起退出
           break