Login module of information management system (implemented in python)

#登录功能
import sys
def login():
    count = 0       #定义输入次数
    user_name = 'XXXX'
    user_code = '666666'
    while count < 3:        #count=0,1,2时运行循环
        input_name = input("请输入用户名：")
        input_code = input("请输入密码：")
        if input_name == user_name and input_code == user_code:
            print("登录成功！")
            break
        else:
            print("登录失败！")
            count += 1
            if count == 3:
                print("对不起，您的账号已被锁定！")
                sys.exit(0)     #登录失败主动结束进程