python编写登陆模块

简介： 通过python编写登陆模块，主要是使用openpyxl对excell表格的操作

# auth marcuya
'''
作业：编写登陆模块
1.输入用户名和密码
2.认证成功显示欢迎信息
3.输错3次后锁定用户

设计思路：
1.用户信息保存文件有4个字段：用户名，密码，密码输入错误次数，是否锁定
2.密码输入错误次数等于3，锁定字段为TRUE，用户不能登陆
3.解除用户锁定，密码输入错误次数 置0，是否锁定修改为 FALSE
'''

import getpass
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