# auth marcuya
'''
作业：编写登陆模块
1.输入用户名和密码
2.认证成功显示欢迎信息
3.输错3次后锁定用户
设计思路：
1.用户信息保存文件有5个字段：序号，用户名，密码，密码输入错误次数，是否锁定
2.密码输入错误次数大于2，锁定字段为TRUE，用户不能登陆'''

import openpyxl

​

while True:

    #获取用户输入的用户名和密码
   print()

    _username = input("please input username:")

    _password = input("please input password:")

​

    # 打开文件
   userInfoFile = openpyxl.load_workbook("homeWork1CountInfo.xlsx")

​

    userInfoSheet = userInfoFile.active

​

    #errorcount = userInfoSheet.cell(row=2, column=4)
   #print(errorcount.value)
   #errorcount.value = 3
   #userInfoFile.save("homeWork1CountInfo.xlsx")
   #逐行轮询
   for rows in userInfoSheet.iter_rows():

        no = rows[0]

        username = rows[1]

        password = rows[2]

        errorCount = rows[3]

        lock = rows[4]

        #print("{} {} {} {} {}".format(no.value, username.value, password.value, errorCount.value, lock.value))
       if _username == username.value:

            #账户被锁定
           if lock.value == True:

                print("too many error count to input, the user {user} is locked，please contact administrator!".format(user=username.value))

                break
           #密码正确，此处需要把值统一为字符类型对比
           elif _password == str(password.value):

                print("welcome user {user} login !".format(user = username.value))

​

                #将输入错误次数置0
               userInfoSheet.cell(row=(no.value + 1), column=4).value = 0
               userInfoFile.save("homeWork1CountInfo.xlsx")

                break
           else:

                #密码错误，密码输入错误次数+1
               errorCount.value = errorCount.value + 1
               userInfoSheet.cell(row=(no.value+1), column=4).value = errorCount.value

                userInfoFile.save("homeWork1CountInfo.xlsx")

                print("invalid username or password")

​

            #密码输入错误次数大于等于3，锁定账户
           if errorCount.value > 2:

                lock.value = True
               userInfoSheet.cell(row=(no.value+1), column=5).value = True
               userInfoFile.save("homeWork1CountInfo.xlsx")

                break
           else:

                break