用户登录接口

2017-09-14 14:56:36

本博客内多为原创，如需转载，请注明出处，谢谢。

---

 功能介绍：

根据输入的用户名，首先判断用户名是否在锁文件中。

如果不在锁文件中，输入该用户的密码，若用用户名和密码匹配成功，则显示欢迎信息；若匹配不成功的次数超过限定输入次数，则将该用户写入到锁文件中。

首先，我们不急着写程序，首先进行需求分析，写好流程图：

程序中涉及到两个文件：userlist.txt 和 user_locked.txt    分别为 用户列表、用户锁文件，这两个文件和程序需要放在相同文件夹下。

文件格式（因为比较简单，就不截图了）：

　　userlist.txt 为：username|password

　　user_locked.txt 为：username

　　（均是每行一条数据）

开始写程序：

# _*_ coding:utf-8 _*_
# author:Mr.chan
# date:2017-09-14

""" 需求：
        1、输入用户名密码并判断是否匹配
        2、登录成功后显示登录信息
        3、输错三次后锁定该用户

    本程序包含两个文件：
        user_locked.txt  # 被锁的用户放在该文件中
        userlist.txt     # 用户列表，包含用户名和密码
"""

import sys

def login(username,password):
    """用户登录函数，判断用户名和密码是否正确"""
    with open("userlist.txt", 'r') as f:
        for line in f:
            if [username, password] == line.strip().split('|'):
                return True   # 若用户名和密码匹配成功，则返回True

def info():
    """登录成功后，显示欢迎信息"""
    if True:
        print("欢迎%s登录"%username)

def user_islocked(username):
    """判断输入的用户是否在锁文件中，若存在，则退出"""
    with open("user_locked.txt",'r') as f:
        for line in f:
            if username == line.strip():
                sys.exit("%s is locked!"%username)

def user_locked(username):
    """用户锁文件，把用户写入到该文件中，用户就无法登录"""
    with open("user_locked.txt",'a+') as f:
        f.write(username)
        f.write('
')
        sys.exit("您输入的错误次数过多，已被锁定")

def main():
    """主函数，执行各项操作"""
    global retry_limit,retry_count,username  #调用函数外的全局变量
    while retry_limit > retry_count:
        username = input("请输入用户名：")
        user_islocked(username)  # 调用函数，检测输入的用户是否在锁文件中
        password = input("请输入密码：")
        # 调用函数，判断用户名和密码是否正确
        f = login(username, password)
        if f:  # 若f为True，执行以下操作
            info()
            break
        else:
            print("用户名或密码错误
")
            retry_count += 1
    else:
        user_locked(username)  # 调用用户锁文件，如果超出错误次数，则执行该函数锁定用户

retry_limit = 3  # 输入的限制次数，若超过则锁定
retry_count = 0  # 用户输入次数的计数器

if __name__ == "__main__":
    main()

 

　　　　大功告成！