python 模拟用户登录代码

需求：输入用户名，判断用户是否被锁定，锁定则退出，否则进入密码验证，输入三次错误密码之后此用户被锁定。

userlist.txt里，用":"将用户名、密码、状态码分开：

[root@o2-test ~]# cat userlist.txt
张三:123:0
李四:456:0
admin:123:0
qqq:123:0

代码：

 #!/usr/bin/env python
 # -*- coding=utf-8 -*-
 __author__ = 'gmk'

 import sys
 import fileinput
 import getpass

 #打开用户列表, 只读
 fileinfo = file('userlist.txt','r')
 user_file = fileinfo.readlines()
 fileinfo.close()

 #定义字典 {'张三':['123','1'],...}
 userdict = {}
 for line in user_file:
     # userinfo = ['张三','123','0']
     userinfo = line.strip().split(":")
     # userdict['张三'] = []
     userdict[userinfo[0]] = [userinfo[1],userinfo[2]]

 #验证用户名
 def verify_user(inputname):
     #用户名为空
     if inputname == '':
         inputname = raw_input('用户名不能为空,请输入用户名:').strip()
         verify_user(inputname)
     #用户名不存在
     elif inputname not in userdict:
         inputname = raw_input('用户不存在,请重新输入用户名:').strip()
         verify_user(inputname)
     else:
         #用户名正确，判断是否锁定
         if userdict[inputname][1] == '':
             print '此用户已锁定,请联系管理员！'
             sys.exit()
         else:
         #用户未锁定，进入密码判断
             print '用户名正确！'
             verify_pwd(inputname)

 #验证密码
 def verify_pwd(okname):
     pwdinfo = getpass.getpass('请输入密码:').strip()
     i = 2
     while True:
         if pwdinfo == userdict[okname][0]:
             print '%s 欢迎回来！' % okname
             sys.exit()
         elif pwdinfo == '':
             pwdinfo = getpass.getpass('密码不能为空，请输入密码:').strip()
         else:
             if i > 0:
                 pwdinfo = getpass.getpass("密码错误，您还有 %d 次机会，请重新输入密码:" %i)
                 i -= 1
             else:
                 update_file(okname)
                 print '密码输入超过三次，用户被锁定！'
                 sys.exit()

 #更新文件内容，锁定用户
 def update_file(okname):
     #旧的字符串   admin:123456:0
     old_str = ':'.join([okname,userdict[okname][0],str(userdict[okname][1])])
     #设置锁定标识
     userdict[okname][1] = 1
     #新的字符串   admin:123456:1
     new_str = ':'.join([okname,userdict[okname][0],str(userdict[okname][1])])

     #替换用户状态信息
     for line in fileinput.input('userlist.txt',inplace=1):
         print line.strip().replace(old_str,new_str)
     fileinput.close()

 user_input = raw_input('欢迎使用员工后台系统！请输入用户名:').strip()
 verify_user(user_input)

运行结果：

 [root@o2-test ~]# python login.py
 欢迎使用员工后台系统！请输入用户名:
 用户名不能为空,请输入用户名:qweqwerqwer
 用户不存在,请重新输入用户名:qqq
 用户名正确！
 请输入密码:
 密码不能为空，请输入密码:
 密码错误，您还有 2 次机会，请重新输入密码:
 密码不能为空，请输入密码:
 密码不能为空，请输入密码:
 密码错误，您还有 1 次机会，请重新输入密码:
 密码输入超过三次，用户被锁定！
 [root@o2-test ~]# python login.py
 欢迎使用员工后台系统！请输入用户名:qqq
 此用户已锁定,请联系管理员！
 [root@o2-test ~]# python login.py
 欢迎使用员工后台系统！请输入用户名:admin
 用户名正确！
 请输入密码:
 admin 欢迎回来！

userlist.txt结果：

 [root@o2-test ~]# cat userlist.txt
 张三::
 李四::
 admin::
 qqq::