小程序：登陆和注册

# -*- coding:utf-8 -*-
def login(username, password):
    """
    用于用户登陆
    :param username: 用户名
    :param password: 密码
    :return:
    """
    f = open('text', 'r')
    lines = f.readlines()              # 调用文件的readlines方法，读取文件全部内容
    for line in lines:
        line_list = line.strip().split('  ')
        if line_list[0] == username and line_list[1] == password:
            return True
    return False


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


def main():
    choice = input("1、登陆  2、注册
")
    if choice == '1':
        user = input("请输入您的用户名：")
        pwd = input("请输入您的密码：")
        result = login(user, pwd)
        if result:
            print("登陆成功！")
        else:
            print("登陆失败，用户名或密码错误！")
    if choice == '2':
        register()

main()