Python小程序之用户登陆接口

编写登陆接口

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

程序逻辑图：

代码：

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

import sys

userinfo = r'D:Python_Scriptsday1userinfo.txt'
userlock = r'D:Python_Scriptsday1userlock.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
' % (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('
'):
                return True
        else:
            return False

#锁定用户
def lockuser(user):
    fd = open(userlock,'a+')
    fd.write('%s
' % 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()