# encoding: utf-8

# Creator：耿亚月   Creation time：2017-1-1

# Modifier：耿亚月    Modification time：2017-1-2

#filename：CheckLogin

import xlrd

import os

import getpass

#查看账号是否注册过

def CheckRegister(username,allusers):

    for user in allusers:

        user = user.strip().split(',')

        if username == user[0]:

            time = 4

            register = 'yes'

            break

        else:

            register = 'no'

    if register == 'yes':

        return 'yes'

    else:

        return 'no'

#查看账号是否被锁定过

def CheckLocked(username,lockedusers):

    for user in lockedusers:

        user = user.strip().split(',')

        if username == user[0]:

            locked = 'yes'

            break

        else:

            locked = 'no'

    if locked == 'yes':

        return 'yes'

    else:

        return 'no'

#查看是否可以登陆

def CheckLogin(username,allusers):

    for user in allusers:

        user = user.strip().split(',')

        if username == user[0] and password == user[1]:

            time = 4

            login = 'yes'

            break

        else:

            login = 'no'

    if  login == 'yes':

        return 'yes'

    else:

        return 'no'

# path = os.getcwd()

path = 'E:\\code\\exercise\\Day1\\CheckLogin'

AllUsers = open(path+'\\AllUsers.txt')

LockedUsers = open(path+'\\LockedUsers.txt')

allusers = AllUsers.readlines()

lockedusers = LockedUsers.readlines()

AllUsers.close()

LockedUsers.close()

time = 0

pas = ''

while time < 3:

    username = input('username:')

    password = input('password:')

    # password=getpass.getpass('password:')

#查看账号是否注册过

    if CheckRegister(username,allusers) == 'no':

        print('there is no this username, please register in first')

        break

    else:

    #查看账号是否被锁定过

        if CheckLocked(username,lockedusers) == 'yes':

                print('this ID has been locked')

                break

        else:

        #查看是否可以登陆

                if CheckLogin(username,allusers) == 'yes':

                    print('welcome user {name} login...'.format(name=username))

                    break

                else:

                    time += 1

                    if time < 3:

                        print('Invalid username or password!')

                    else:

                        print('you put username or password over 3 time,this ID has been locked')

                    #将输错超过3次的用户名写入到ockedLUser文件里

                        LockedUsers = open(path+'\\LockedUsers.txt','a')

                        LockedUsers.write('\n'+username + ',' + password )

                        LockedUsers.close()