Python 文件读写，条件循环（三次登录锁定账号实例）

需求一

"""
需求描述：
1.输入正确账号，密码，退出程序
2.登录失败，重新输入账号密码
3.同一账号连续登录失败超过三次，则锁定该账号
4.登录已锁定账号，提示账号已锁定
"""

流程图一

实例代码一

#!/usr/bin/python
#coding=utf-8

"""
需求描述：
1.输入正确账号，密码，退出程序
2.登录失败，重新输入账号密码
3.同一账号连续登录失败超过三次，则锁定该账号
4.登录已锁定账号，提示账号已锁定
user.txt:
Milton Loveyp 0
Cherish Loveypgfc 0
yangp Loveypgfc 0
"""

def modify_user(name, mode="add"):
"""
修改用户名单中的登录错误次数
:param name: 用户名称
:param mode:add，登录错误次数自增1；del，登录错误次数重置为0
:return:
"""

# 读取用户文件
with open("user.txt", "r") as read_file:
read_data = read_file.readlines()

# 写入用户文件
with open("user.txt", "w") as write_file:
for line in read_data:
new_line = line.split()
if new_line[0] == name:
if mode == "add":
new_line[2] = bytes(int(new_line[2]) + 1)
else:
new_line[2] = "0"
new_line = " ".join(new_line) + "\n"
write_file.write(new_line)
write_file.flush()
else:
write_file.write(line)

def auth(name, passwd):
"""
用户鉴权
:param name: 登录用户名
:param passwd: 登录密码
:return:
"""
with open("user.txt", "r") as read_file:
for eachLine in read_file:
user_name, password, errors = eachLine.split()
if name == user_name:
if int(errors) < 3:
if password == passwd:
print "Dear [%s],welcome~" % name
modify_user(name, "del")
return True
else:
print "auth fail,please check your account and password!"
modify_user(name, "add")
return False
else:
print "your accout is locked!"
return False
else:
print "your account is not exist!"
return False

def login():
"""
登录
:return:
"""
while True:
name = raw_input("Name:")
passwd = raw_input("PASS:")
if auth(name, passwd):
break

if __name__ == '__main__':
login()

需求二，在一的基础上添加锁定时间控制。

"""
需求点：
1.用户输入账号与密码
2.如果输入正确，则提示欢迎信息
3.如果输入账号不存在，则提示账号不存在
4.如果输入账号与密码不匹配，提示验证失败
5.同一账号，如果连续输入三次错误密码，则锁定该账号2min，2min后重新输入，如果账号密码匹配，则登录成功
6.如果三次登录内成功登录，则清空登录错误次数
7.登录时，如果账号已被锁定，提示账号已锁定信息
"""

流程图二

实例代码二

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

"""
需求点：
1.用户输入账号与密码
2.如果输入正确，则提示欢迎信息
3.如果输入账号不存在，则提示账号不存在
4.如果输入账号与密码不匹配，提示验证失败
5.同一账号，如果连续输入三次错误密码，则锁定该账号2min，2min后重新输入，如果账号密码匹配，则登录成功
6.如果三次登录内成功登录，则清空登录错误次数
7.登录时，如果账号已被锁定，提示账号已锁定信息
user.txt:
Milton Loveyp 0
Cherish Loveypgfc 0
yangp Loveypgfc 0
"""

import time

def modify_user(name, mode="add"):
"""
修改用户名单中的登录错误次数
:param name: 用户名称
:param mode:add，登录错误次数自增1；del，登录错误次数重置为0
:return:
"""
# 读取用户文件
with open("user.txt", "r") as read_file:
read_data = read_file.readlines()

# 写入用户文件
with open("user.txt", "w") as write_file:
for line in read_data:
new_line = line.split()
if new_line[0] == name:
if mode == "add":
new_line[2] = bytes(int(new_line[2]) + 1)
if len(new_line) < 4:
new_line.append(str(time.time()))
else:
new_line[3] = str(time.time())
else:
new_line[2] = "0"
new_line = new_line[0:3]
new_line = " ".join(new_line) + "\n"
write_file.write(new_line)
write_file.flush()
else:
write_file.write(line)

def auth(name, passwd, lock_time):
"""
用户鉴权
:param name: 登录用户名
:param passwd: 登录密码
:return:
"""
with open("user.txt", "r") as read_file:
for eachLine in read_file:
user_list = eachLine.split()
if len(user_list) < 4:
user_list.append(str(time.time() - 100000))
user_name, password, errors, last_time = user_list

if name == user_name:
if password == passwd:
if time.time() - float(last_time) > lock_time or int(errors) < 3:
print "Dear [%s],welcome~" % name
modify_user(name, "del")
return True
else:
print "your accout is locked,please try after [%f]s !" % (
lock_time - (time.time() - float(last_time)))
return False
else:
if int(errors) < 3:
print "auth fail,please check your account and password!"
modify_user(name, "add")
return False
elif time.time() - float(last_time) > lock_time:
print "auth fail,please check your account and password!"
modify_user(name, "del")
modify_user(name, "add")
return False
else:
print "your accout is locked,please try after [%f]s !" % (
lock_time - (time.time() - float(last_time)))
return False
else:
print "your account is not exist!"
return False

def login():
"""
登录
:return:
"""
while True:
user_name = raw_input("Name:")
pwd = raw_input("PASS:")
if auth(user_name, pwd, 120):
break

if __name__ == '__main__':
login()