python登陆模块编写

# 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