Python小程序之用户登陆接口

编写登陆接口

• 输入用户名密码
• 认证成功后显示欢迎信息
• 输错三次后锁定

程序逻辑图：

代码：

 #!/usr/bin/env python
 #_*_ coding:UTF-8 _*_
 #__author__ = lee sir

 import sys

 userinfo = r'D:\Python_Scripts\day1\userinfo.txt'
 userlock = r'D:\Python_Scripts\day1\userlock.txt'

 #用户输入用户名和密码
 def user_input():
     name = input("username:")
     passwd = input("password:")
     return name,passwd

 #判断用户是否是已注册用户
 def judge_register(user):
     with open(userinfo) as fd:
         for info in fd.readlines():
             username,password = info.split()
             if user == username:
                 return True
         else:
             return False

 #注册用户
 def register_user(user,password):
     fd = open(userinfo,'a+')
     fd.write('%s %s\n' % (user,password))
     sys.exit('user register Success!')

 #判断用户名密码正确性
 def judge_password(user,passwd):
     with open(userinfo) as fd:
         for info in fd.readlines():
             username,password = info.split()
             if user == username and passwd == password:
                 return True
         else:
             return False

 #测试用户是否被锁定
 def test_user(user):
     with open(userlock) as fd:
         for info in fd.readlines():
             if user == info.strip('\n'):
                 return True
         else:
             return False

 #锁定用户
 def lockuser(user):
     fd = open(userlock,'a+')
     fd.write('%s\n' % user )
     fd.close()
     sys.exit("Your Account is locking !!")

 #主函数
 def main():
     count = 0
     while count < 3:
         username,password = user_input()    #获取用户输入的用户名和密码
         judge_register_result = judge_register(username)
         if judge_register_result:   #启动登陆流程
             test_user_result = test_user(username)     #判断用户是否被锁定
             if test_user_result:
                 sys.exit('Sorry,Your Account is locked !!')
             else:   #验证用户名密码是否正确
                 judge_password_result = judge_password(username,password)
                 if judge_password_result:
                     sys.exit('Hello,Welcome login!!')
                 else:
                     if count < 2:
                         print('Username or Password is Wrong,You have %s chance' % (2-count))
                 count += 1
         else:   #启动注册流程
             user_choice = input('Are you want to register ? [y/n]')
             if user_choice == 'y' or user_choice == 'Y':
                 register_user(username,password)
                 sys.exit('register successful,Please relogin !')
             elif user_choice == 'n' or user_choice == 'N':
                 sys.exit('Bye Bye')
             else:
                 sys.exit('Input Error,Bye Bye!')
     else:    #三次失败 锁定帐户
         lockuser(username)

 if __name__ == '__main__':
     main()