'''

有bug未及时调整

登出函数有问题

'''

import json

import time

def init_info():

    user_if = {

    "egon":{"password":"",'status':False,'timeout':0},

    "alex":{"password":"",'status':False,'timeout':0},

    }

    with open('user_info','w',encoding='utf-8') as f:

        json.dump(user_if,f)

def read_info():

    with open('user_info','r+',encoding='utf-8') as f:

        print(json.load(f,encoding = 'utf-8'))

user_name = ''

class User:

    def db(self):

        '''

        db方法：调用时打印当前用户信息和信息存储数据格式

        :return:

        '''

        with open('user_info', 'r+', encoding='utf-8') as f:

            user_info = json.load(f, encoding='utf-8')

            for k, v in user_info.items():

                print('username:%s -' % k, type(k),

                      'password:',[val for val in v.values()][0], type([val for val in v.values()][0]),

                      'status:',[val for val in v.values()][1], type([val for val in v.values()][1]),

                      'timeout:', [val for val in v.values()][2], type([val for val in v.values()][2]),

                      )

    def login(self):

        '''

        登录

        登录成功将状态(status)修改为True

        密码输入错误三次将设置锁定时间(下次登录如果和当前时间比较大于10秒即不允许登录)[个人理解应该是10秒内不能登录，这里的表述有问题]

        :return:

        '''

        count = 0

        while count < 3:

            username = input('请输入用户名>>>').strip()

            password = input('请输入密码>>>').strip()

            with open('user_info', 'r+', encoding='utf-8') as f:

                user_info = json.load(f, encoding='utf-8')

                user_name_list = []

                for k in user_info.keys():

                    user_name_list.append(k)

                if username in user_name_list and password == user_info.get(username)['password']:

                    if user_info.get(username)['timeout'] - time.time()<0:

                        user_info.get(username)['status'] = True

                        print('用户：%s 登陆成功' % username)

                        user_dict = {username:user_info.get(username)}

                        user_name = username

                        with open('user_info','w+',encoding='utf-8') as f:

                            json.dump(user_info, f)

                        return user_dict[username]

                    else:

                        return '用户账号已锁定！'

                elif count == 2:

                    if username in user_name_list:

                        user_info.get(username)['timeout'] = time.time()+10

                        with open('user_info','w+',encoding='utf-8') as f:

                            json.dump(user_info, f)

                        return  '%s账户已锁定'%username

                    else:

                        return '用户名不存在！'

                else:

                    print('账号或密码错误，还有%s次尝试机会'% (3-(count+1)))

                    count += 1

    def logout(self):

        '''

        退出将状态修改为False(退出要判断是否处于登录状态)

        :return:

        '''

        with open('user_info', 'r+', encoding='utf-8') as f:

            user_info = json.load(f, encoding='utf-8')

        user_list = []

        for k in user_info.keys():

            user_list.append(k)

        if user_name in user_list:

            print('当前登录账户%s'%user_name)

            choice = input('请确认是否退出，是：Y  否：N').strip()

            if choice == 'y'.upper() or choice == 'y'.lower():

                user_info.get(user_name)['status'] = False

                with open('user_info', 'w+', encoding='utf-8') as f:

                    json.dump(user_info, f)

            else:

                pass

        else:

            pass

us = User()

# us.db()

u = us.login()