本次将学习如何编写一个登录接口小程序。
具体的需求
实现一个登录接口,用户输入账户密码即可看到欢迎语句,当三次输入密码错误时,则该帐号被锁定。
锁定的意义是,下次使用该账户登录时,将无法成功登录。
思路
因为需求有次数限制,所以需要有一个计数变量,同时可能还需要用到while和if语句。
需求提到被锁定的帐号在程序下一次启动时仍然记住,众所周知,python有内存回收机制,程序在结束之后,所有列表、变量会从内存中抹去,这就需要有一个文件来存放这些被锁定的帐号。
我们需要使用python的open方法来写入文件,可以将锁定帐号写入文件,可以命名为blacklist.txt之类的。
blacklist=open("blacklist.txt","w")
此外,需求还需要实现多用户的登录操作,这就需要我们建立一个用户密码的文件,可以命名为admin.txt,每次登录需要从里面校验用户名密码是否正确。
程序逻辑
我们接下来通过流程图来把整个过程先梳理一遍:
login_port.png
实现
因为涉及到用户密码读取,用户名和用户密码的一个对应。
拉入黑名单需要涉及文件写入的操作,有点复杂。
可以先用list列表把问题简单化。
所以我先用列表代替文件,密码先只有一个,写死的。
初步实现
运行这段代码,可以看到:
然后开始考虑实现文件的读取,写入操作。
最终实现代码:
改进
当然这样还不完美,还有一个小漏洞,就是这种判断锁定的是第三次用户输错时的账户,如果有人连续输错两次密码,再第三次输入别人的账户,故意输错,就会导致别人的账户被锁定,存在恶意攻击的漏洞,请看如下演示:
修正了一下之后,代码是这样的,加了一个while判断,同时把输入密码的操作放在第二个循环里,这样就避免了恶意锁定别人的账户:
基本部分完成之后,我们再对程序优化下,目前输入密码是明文的,我们可以用getpass模块来解决这个问题,优化的代码示例如下:
到目前为止,一个用列表和while,if语句判断的登陆小程序就算完成了。
现在看来用列表来装这些用户名密码什么的还是挺复杂的,要是用字典就更好了,另外写入黑名单的操作可以做成def函数,这样看起来更清晰,还可以重复调用,效率更高。
字典
将源程序中的列表做成字典的形式,用户名对应密码,结构更清晰。可以利用字典 dict[username]=passwd 把用户名的密码存放,用dict[username]调用对应用户的密码,同时加入了一个old_name的变量,用来判断上一次输入的username是否和这一次的一样,不一样则计数变量count重新开始计算。
实现代码:
实现效果:
好了,大功告成了,这一阶段的学习就到这里啦~