# -*- coding:utf-8 -*-
def login(username, password):
    """
    用于用户登陆
    :param username: 用户名
    :param password: 密码
    :return:
    """
    f = open('text', 'r')
    for line in f:
        line_list = line.strip().split('  ')
        if line_list[0] == username and line_list[1] == password:
            return True
        elif line_list[0] != username or line_list[1] != password:
            return 0
    return False


def register():
    """
    用于用户注册
    :return:
    """
    user = input("请输入您的用户名:")
    pwd = input("请输入您的密码:")
    pwd_2 = input("请再次输入您的密码:")
    if pwd != pwd_2:
        print("第二次输入的密码和第一次不同,请重新注册!\n")
        register()
    f = open('text', 'a')
    temp = '\n' + user + '  ' + pwd
    f.write(temp)
    f.close()
    print("注册成功!")


def main():
    choice = input("1、登陆  2、注册\n")
    if choice == '1':
        user = input("请输入您的用户名:")
        pwd = input("请输入您的密码:")
        result = login(user, pwd)
        if result:
            print("登陆成功!")
        elif result == '0':
            print("用户名或密码错误!")
        else:
            print("用户名和密码均错误!")
    if choice == '2':
        register()

main()