Python登录注册功能

def login(username, password):
    """
    用于用户登录
    :param username:用户输入的用户名 
    :param password: 用户输入的密码
    :return: true表示登录成功，false登录失败
    """
    # 打开文件
    f = open('db', 'r')
    # 一行一行遍历
    for line in f:
        line_list = line.strip().split('|')  # 去行（\n）拆分
        if line_list[0] == username and line_list[1] == password:
            return True
    return False


def register(username, password):
    """
    用户注册功能
    :param username: 用户名
    :param password: 密码
    :return: 默认返回null
    """

    f = open('db', 'a')  # 追加模式
    temp = '\n' + username + '|' + password
    f.write(temp)  # 写入到文件
    f.close()


# 主函数
def main():
    t = input('1:登录  2：注册')
    if t == '1':
        user = input('请输入用户名：')
        pwd = input('请输入密码：')
        r = login(user, pwd)
        if r:
            print('登录成功')
        else:
            print('登录失败')
    elif t == '2':
        user = input('请输入用户名：')
        pwd = input('请输入密码：')
        print('注册成功。')
        register(user, pwd)


# 当程序运行时，调用主函数
main()