用Python编写登录接口

登陆接口要求：

1. 输入用户名和密码
2. 认证成功后显示欢迎信息
3. 输错三次后锁定

代码：

#!/usr/bin/env python
# -*- coding:utf-8 -*-
# Author:James Tao

import pickle
import os

#注册
register={'tz':'1234','t':'123','tza':'12345abc'}
locked_account=[]

#也可由用户输入注册账号
#register_username=input('newusername:')
#register_passord=input('newpassword:')
#print('注册成功')
#register[register_username]=register_passord

#用pickle把已注册的账号字典序列化写到文件中
w_register=open('register.txt','wb')
pickle.dump(register,w_register)
w_register.close()

#登陆
count=0
while count<3:
    print('请登录')
    username=input('username:')
    password=input('password:')

    #若被锁定账号文件不为空，读出已锁定的账号
    if os.path.getsize('lockedaccount.txt') > 0:

        #用pickle把被锁定账号从文件中反序列化读出来
        r_lockedaccount=open('lockedaccount.txt', 'rb')
        locked_account=pickle.load(r_lockedaccount)
        r_lockedaccount.close()

    #判断输入的账号是否已被锁定，若账号已被锁定则直接跳出所有循环
    judge_of_locked=False
    for locked_name in locked_account:
        if username==locked_name:
            print('The account has been locked')
            judge_of_locked = True
            break
    if judge_of_locked:
        break

    #用pickle把已注册的账号字典反序列化读出来
    r_register=open('register.txt','rb')
    login=pickle.load(r_register)
    r_register.close()

    #判断输入的账号是否存在密码是否正确
    if username in login and password==login[username]:
        print('Welcome')
        break
    else:
        print('Non-existed username or password error')
    count+=1

#若账号和密码输错三次账号会被锁定
else:
    print('You have tried too many times,The account has been locked')

    #插入被锁定账号的用户名
    locked_account.append(username)

    #用pickle把被锁定账号的列表序列化写到文件中
    w_lockedaccount = open('lockedaccount.txt', 'wb')
    pickle.dump(locked_account,w_lockedaccount)
    w_lockedaccount.close()