项目需求描述:要求用户输入用户名和密码,认证成功后显示欢迎信息,如果连续输错三次密码则锁定该用户。


逻辑流程图:

用Python实现用户登录接口_python


实现代码:

#!/usr/bin/env python
import sys

account_file = 'account.txt'
lock_file = 'lock.txt'

# put accounts in a list
fh_account = open(account_file)
account_list = fh_account.readlines()
fh_account.close()

# every account has 3 chances of attempt at the beginning
retry_count = {}
for line in account_list:
    line = line.split()
    retry_count[line[0]] = 3

while True:
    # put locked accounts in a list
    fh_lock = open(lock_file)
    lock_list = []
    for i in fh_lock.readlines():
        line = i.strip('\n')
        lock_list.append(line)
    fh_lock.close()

    # handle the username and password empty issue
    username = raw_input('Username: ').strip()
    if len(username) == 0:
        print '\033[31;1mUsername should not be empty !\033[0m'
        continue
    password = raw_input('Password: ').strip()
    if len(password) == 0:
        print '\033[31;1mPassword should not be empty !\033[0m'
        continue

    # authentication part
    if username in lock_list:
        print "\033[31;1mSorry, '%s' is locked already !\033[0m" % username
        continue
    if not retry_count.has_key(username):   # inexistent account
        retry_count[username] = 3
    for line in account_list:
        line = line.split()
        if username == line[0] and password == line[1]: # authentication pass
            retry_count[username] = 3       # reset retry times for this account
            sys.exit('\033[32;1mWelcome %s login my system !\033[0m' % username)
    else:                                   # authentication failed
        print '\033[31;1mWrong username or password !\033[0m'
        retry_count[username] -= 1
        if retry_count[username] > 0:
            print '\033[31;1mYou have %d more chances !\033[0m' % retry_count[username]
        else:                               # no more chance of attempt
            fh = open(lock_file, 'a')
            fh.write('%s\n' % username)
            fh.close()
            print "\033[31;1mSorry, '%s' is locked !\033[0m" % username