# 基础需求：

# 让用户输入用户名密码

# 认证成功后显示欢迎信息

# 输错三次后退出程序

username = 'pandaboy'

password = ''

def Login(username,password):

    i=0

    while i<3:

        U = input('Please input username>>>')

        P = input('Please input password>>>')

        if U == username and P == password:

            print('Welcome\n'+username)

            break

        else:

            i+=1

            if 3-i !=0:

                print("Sorry,Don't U forget your account? Please try again,you still have\t" +str(3-i) +"\ttimes.")

            else:

                print("Sorry! I am shutting down!")

                break

if __name__ == '__main__':

    Login(username,password)

# 升级需求：

# 可以支持多个用户登录 (提示，通过列表存多个账户信息){pandaboy的疑问：用列表存储应该更好?}

# 用户3次认证失败后，退出程序，再次启动程序尝试登录时，还是锁定状态（提示:需把用户锁定的状态存到文件里）

'''

1.用户可以输入3次不同的账户，都输错也不会锁定

2.用户全部输入三次相同用户名错误才会锁定

3.创建一个初始化状态的文件

参考了博客文件

'''

import os.path

import pickle

username= ['pandaboy','zombie','plant']

password = ['','','']

def Login(username,password):

    i=0

    while i<3:

        if not os.path.isfile('lock_file'):

            f = open('lock_file', 'wb')

            msg = {'pandaboy': 0, 'zombie': 0, 'plant': 0}

            pickle.dump(msg, f)

            f.close()

        f = open('lock_file', 'rb+')

        msg = pickle.load(f)

        U = input('Please input username>>>')

        if U not in msg :

            print('No such\t' +U+ '\taccount! Try again!')

            f.close()

            continue

        if msg[U] == 3:

            print('your account has been locked,please contact the administrator!')

            f.close()

            break

        P = input('Please input password>>>')

        if U  in username and P == password[username.index(U)]:

            print('Welcome', U)

            msg[U] = 0

            f.seek(0)

            pickle.dump(msg, f)

            f.close()

            break

        else:

            print('Wrong password!')

            msg[U] += 1

            print('Find %s errors' % msg[U])

            f.seek(0)

            pickle.dump(msg, f)

            f.close()

if __name__ == '__main__':

    Login(username,password)