python之编写登录接口

编写登陆接口

• 输入用户名密码
• 认证成功后显示欢迎信息
• 输错三次后锁定

 

 

针对此实例写了有二种类型的脚本，略有不同，具体如下：

帐号文件account.txt内容如下：

---

sam 123 

david 12 
kevin 123
lin 12 
tailen 123 
jack 12

---

 

锁文件account_lock.txt默认为空

 

一、只针对帐号文件里的用户进行判断并锁定，针对用户和密码各有三次错误重试机会。

1、流程图如下：

代码如下：

#!/usr/bin/env python# Auther by wqimport sys, os, getpass#os.system('clear')i = 0while i < 3:  # 只要用户登录异常不超过3次就不断循环    name = input("请输入用户名：")    lock_file = open('D:\python_project\\account_lock.txt', 'r+')  # 当用户输入用户名后，打开LOCK 文件 以检查是否此用户已经LOCK了    lock_list = lock_file.readlines()    for lock_line in lock_list:  # 循环LOCK文件        lock_line = lock_line.strip('\n')  # 去掉换行符        if name == lock_line:  # 如果LOCK了就直接退出            sys.exit('用户 %s 已经被锁定，退出' % name)    user_file = open('D:\python_project\\account.txt', 'r')  # 打开帐号文件    user_list = user_file.readlines()    for user_line in user_list:  # 对帐号文件进行遍历        (user, password) = user_line.strip('\n').split()  # 分别获取帐号和密码信息        if name == user:  # 如用户名正常匹配            j = 0            while j < 3:  # 只要用户密码异常不超过3次就不断循环                passwd = input('请输入密码：')  # 输入隐藏密码（由于pycharm里不支持getpass.getpass因此这里改为input）                if passwd == password:  # 密码正确，提示欢迎登录                    print('欢迎登录管理平台，用户%s' % name)                    sys.exit(0)  # 正常退出                else:                    if j != 2:  # j=2时，是最后一次机会，不用在提示还剩余0次机会了                        print('用户 %s 密码错误，请重新输入，还有 %d 次机会' % (name, 2 - j))                j += 1  # 密码输入错误后，循环值增加1            else:                lock_file.write(name + '\n')  # 密码输入三次错误后，将该用户追加到LOCK文件                sys.exit('用户 %s 达到最大登录次数，将被锁定并退出' % name)        else:            pass  # 当用户没匹配时，跳过并继续循环    else:        if i != 2:  # i=2时，是最后一次机会，不用在提示还剩余0次机会了            print('用户 %s 不存在，请重新输入，还有 %d 次机会' % (name, 2 - i))    i += 1  # 当用户输入错误时，循环值增加1else:    sys.exit('用户 %s 不存在，退出' % name)  # 用户输入三次错误后，异常退出lock_file.close()  # 关闭LOCK文件user_file.close()  # 关闭帐号文件

 

二、针对帐号文件里的不存在的用户也可以进行判断并锁定，针对用户和密码共有三次错误重试机会

代码如下：

#_*_ coding:utf-8 _*_import sys,os,getpassos.system('clear')retry_limit = 3retry_count = 0account_file = 'account.txt'lock_file = 'account_lock.txt'while retry_count < retry_limit:                                         #只要重试不超过3次就不断循环    username = raw_input('\033[31;43mUsername:\033[0m')    username = username.strip()    lock_check = open(lock_file)                                         #当用户输入用户名后，打开LOCK 文件 以检查是否此用户已经LOCK了    for line in lock_check.readlines():                                  #循环LOCK文件         if username == line.strip('\n'):                                  #去掉换行符            sys.exit('\033[35mUser %s is locked!!!\033[0m' % username)   #如果LOCK了就直接退出    password = raw_input('\033[32;41mPassword:\033[0m')                  #输入密码    f = open(account_file,'r')                                           #打开帐号文件     match_flag = False                                                   # 默认为Flase,如果用户match 上了，就设置为 True     for line in f.readlines():                                                   user,passwd = line.strip('\n').split()                           #去掉每行多余的\n并把这一行按空格分成两列，分别赋值为user,passwd两个变量        if username == user and password == passwd:                      #判断用户名和密码是否都相等           print('hello, %s !!' % username)           match_flag = True                                             #相等就把循环外的match_flag变量改为了True           break                                                         #然后就不用继续循环了，直接 跳出，因为已经match上了    f.close()    if match_flag == False:                                              #如果match_flag还为False,代表上面的循环中跟本就没有match上用户名和密码，所以需要继续循环        print('sorry,%s is unmatched' % username)        retry_count += 1                                                 #计数器加1    else:        print('wlecome login my learning system!')        break                                                            #用户成功登录，退出脚本else:    print("you account %s is locked!!!" % username)    g = open(lock_file,'a')    g.write(username)                                                    #被锁用户追加到用户锁文件    g.write('\n')       g.close()