本次将学习如何编写一个登录接口小程序。
具体的需求
实现一个登录接口，用户输入账户密码即可看到欢迎语句，当三次输入密码错误时，则该帐号被锁定。
锁定的意义是，下次使用该账户登录时，将无法成功登录。
思路
因为需求有次数限制，所以需要有一个计数变量，同时可能还需要用到while和if语句。
需求提到被锁定的帐号在程序下一次启动时仍然记住，众所周知，python有内存回收机制，程序在结束之后，所有列表、变量会从内存中抹去，这就需要有一个文件来存放这些被锁定的帐号。
我们需要使用python的open方法来写入文件，可以将锁定帐号写入文件，可以命名为blacklist.txt之类的。
blacklist=open("blacklist.txt","w")
此外，需求还需要实现多用户的登录操作，这就需要我们建立一个用户密码的文件，可以命名为admin.txt，每次登录需要从里面校验用户名密码是否正确。
passwd=open("admin.txt","r")
程序逻辑
我们接下来通过流程图来把整个过程先梳理一遍：
login_port.png
实现
因为涉及到用户密码读取，用户名和用户密码的一个对应。
拉入黑名单需要涉及文件写入的操作，有点复杂。
可以先用list列表把问题简单化。
所以我先用列表代替文件，密码先只有一个，写死的。
初步实现
#owner:houyizhong
#version:demo
admin_name=["houyizhong","test"]
admin_passwd="137010116"
blacklist=["test"]
count = 0
while count < 3:
username=input("Enter your username:")
passwd=input("Enter your password:")
if username not in admin_name:
print("Sorry,your account is not exist!")
continue
elif username in blacklist:
print ("Sorry you are in blacklist!")
exit()
else:
if passwd != admin_passwd:
print("Sorry,enter your password again!")
count += 1
else:
print("Welcome! {0}".format(username))
break
else:
print("Sorry you are in blacklist!")
blacklist.append(username)
exit()
运行这段代码，可以看到：
[root@houyizhong ~]# python login.py
Enter your username:houyizhong
Enter your password:137010116
Welcome! houyizhong
然后开始考虑实现文件的读取，写入操作。
最终实现代码：
# -*- coding:utf-8 -*-
#owner:houyizhong
#version1.0
#创建列表
admin_name=[]
admin_passwd=[]
blacklist=[]
#读取用户文件
passwd_file=open("admin.txt","r")
admin_list=passwd_file.readlines()
passwd_file.close()
#提取出用户名，密码，添加进列表
for i in range(len(admin_list)):
admin_parts=admin_list[i]
admin_passwd.append(admin_parts.strip())
admin_part=admin_parts.split(":")
admin_name.append(admin_part[0])
#读取黑名单文件
blacklist_file=open("blacklist.txt","r")
blacklist_raw=blacklist_file.readlines()
blacklist_file.close()
#提取黑名单用户名
for i in range(len(blacklist_raw)):
blacklist_value=blacklist_raw[i]
blacklist.append(blacklist_value.strip())
count = 0
while count < 3:
username=input("Enter your username:")
passwd=input("Enter your password:")
if username not in admin_name:
print("Sorry,your account is not exist!")
continue
elif username in blacklist:
print ("Sorry you are in blacklist!")
exit()
else:
if ("{0}:{1}".format(username,passwd)) not in admin_passwd:
print("Sorry,enter your password again!")
count += 1
else:
print("Welcome! {0}".format(username))
break
else:
print("Sorry you are in blacklist!")
blacklist_file=open("blacklist.txt","a")
blacklist_file.write("{0}\n".format(username))
blacklist_file.close()
exit()
改进
当然这样还不完美，还有一个小漏洞，就是这种判断锁定的是第三次用户输错时的账户，如果有人连续输错两次密码，再第三次输入别人的账户，故意输错，就会导致别人的账户被锁定，存在恶意攻击的漏洞，请看如下演示：
[root@houyizhong pythonprobe]# python login.py
Enter your username:houyizhong
Enter your password:11111
Sorry,enter your password again!
Enter your username:houyizhong
Enter your password:11111
Sorry,enter your password again!
Enter your username:root
Enter your password:1111
Sorry,enter your password again!
Sorry you are in blacklist!
[root@houyizhong pythonprobe]# cat blacklist.txt
root
修正了一下之后，代码是这样的，加了一个while判断，同时把输入密码的操作放在第二个循环里，这样就避免了恶意锁定别人的账户：
# -*- coding:utf-8 -*-
#owner:houyizhong
#version2.0
#创建列表
admin_name=[]
admin_passwd=[]
blacklist=[]
#读取用户文件
passwd_file=open("admin.txt","r")
admin_list=passwd_file.readlines()
passwd_file.close()
#提取出用户名，密码，添加进列表
for i in range(len(admin_list)):
admin_parts=admin_list[i]
admin_passwd.append(admin_parts.strip())
admin_part=admin_parts.split(":")
admin_name.append(admin_part[0])
#读取黑名单文件
blacklist_file=open("blacklist.txt","r")
blacklist_raw=blacklist_file.readlines()
blacklist_file.close()
#提取黑名单用户名
for i in range(len(blacklist_raw)):
blacklist_value=blacklist_raw[i]
blacklist.append(blacklist_value.strip())
count = 0
while count < 3:
username=input("Enter your username:")
if username not in admin_name:
print("Sorry,your account is not exist!")
continue
elif username in blacklist:
print ("Sorry you are in blacklist!")
else:
pass
while count < 3:
passwd=input("Enter your password:")
if ("{0}:{1}".format(username,passwd)) not in admin_passwd:
print("Sorry,enter your password again!")
count += 1
else:
print("Welcome! {0}".format(username))
exit()
else:
print("Sorry you are in blacklist!")
blacklist_file=open("blacklist.txt","a")
blacklist_file.write("{0}\n".format(username))
blacklist_file.close()
exit()
基本部分完成之后，我们再对程序优化下，目前输入密码是明文的，我们可以用getpass模块来解决这个问题，优化的代码示例如下：
# -*- coding:utf-8 -*-
#owner:houyizhong
#version2.0
import getpass
#创建列表
admin_name=[]
admin_passwd=[]
blacklist=[]
#读取用户文件
passwd_file=open("admin.txt","r")
admin_list=passwd_file.readlines()
passwd_file.close()
#提取出用户名，密码，添加进列表
for i in range(len(admin_list)):
admin_parts=admin_list[i]
admin_passwd.append(admin_parts.strip())
admin_part=admin_parts.split(":")
admin_name.append(admin_part[0])
#读取黑名单文件
blacklist_file=open("blacklist.txt","r")
blacklist_raw=blacklist_file.readlines()
blacklist_file.close()
#提取黑名单用户名
for i in range(len(blacklist_raw)):
blacklist_value=blacklist_raw[i]
blacklist.append(blacklist_value.strip())
count = 0
while count < 3:
username=input("Enter your username:")
if username not in admin_name:
print("Sorry,your account is not exist!")
continue
elif username in blacklist:
print ("Sorry you are in blacklist!")
else:
pass
while count < 3:
passwd=getpass.getpass("Enter your password:")
if ("{0}:{1}".format(username,passwd)) not in admin_passwd:
print("Sorry,enter your password again!")
count += 1
else:
print("Welcome! {0}".format(username))
exit()
else:
print("Sorry you are in blacklist!")
blacklist_file=open("blacklist.txt","a")
blacklist_file.write("{0}\n".format(username))
blacklist_file.close()
exit()
到目前为止，一个用列表和while，if语句判断的登陆小程序就算完成了。
现在看来用列表来装这些用户名密码什么的还是挺复杂的，要是用字典就更好了，另外写入黑名单的操作可以做成def函数，这样看起来更清晰，还可以重复调用，效率更高。
字典
将源程序中的列表做成字典的形式，用户名对应密码，结构更清晰。可以利用字典 dict[username]=passwd 把用户名的密码存放，用dict[username]调用对应用户的密码,同时加入了一个old_name的变量，用来判断上一次输入的username是否和这一次的一样，不一样则计数变量count重新开始计算。
实现代码：
#-*- coding:utf-8 -*-
#version3.0
#owner:houyizhong
#create the list and dict
#admin_name=[]
admin_passwd={}
blacklist=[]
#read admin file
admin_file=open("admin.txt","r")
admin_list=admin_file.readlines()
admin_file.close()
#read admin username and password
for i in range(len(admin_list)):
admin_parts=admin_list[i]
admin_raw=admin_parts.strip()
admin_part=admin_raw.split(':')
name=admin_part[0]
passwd=admin_part[1]
admin_passwd[name]=passwd
print(admin_passwd)
#read blacklist file
blacklist_file=open("blacklist.txt","r")
blacklist_raw=blacklist_file.readlines()
blacklist_file.close()
#read blacklist username
for i in range(len(blacklist_raw)):
blacklist_value=blacklist_raw[i]
blacklist.append(blacklist_value.strip())
count = 0
old_name=""
while count < 3:
username=input("username:")
if username != old_name:
count = 0
passwd=input("passwd:")
old_name=username
if username not in admin_passwd:
print ("Sorry ,you account is not exist!")
continue
elif username in blacklist:
print ("Sorry ,you are in the blacklist!")
exit()
else:
if passwd != admin_passwd[username]:
count += 1
print("Sorry,you password is wrong.")
else:
print ("Welcome!{0}".format(username))
break
else:
print ("You are in blacklist now!")
blacklist_file=open("blacklist.txt","a")
blacklist_file.write("{0}\n".format(username))
blacklist_file.close()
exit()
实现效果：
[houyizhong@localhost py]$ python login.py
username:test
passwd:
Sorry,you password is wrong.
username:test
passwd:
Sorry,you password is wrong.
username:root
passwd:
Sorry,you password is wrong.
username:root
passwd:
Welcome!root
好了，大功告成了，这一阶段的学习就到这里啦~
