如何将循环添加到用户名和密码登录(Python)？

def login():
    max_attempts = 3
    attempts = 0

    while attempts < max_attempts:
        username = input("请输入用户名：")
        password = input("请输入密码：")

        # 进行用户名和密码的验证
        if username == "admin" and password == "123456":
            print("登录成功！")
            break
        else:
            print("用户名或密码错误，请重新输入。")
            attempts += 1

    if attempts == max_attempts:
        print("登录失败次数过多，账号已锁定。")

login()