登陆注册小程序

def login(username, password):
    """
    用于用户登陆
    :param username: 用户输入的用户名
    :param password: 用户输入的密码
    :return: 如果返回值为:True，表示登陆成功，Flase，表示登陆失败
    """
    f = open("db", 'r')
    for line in f:
        line_list = line.split("|")
        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("请输入注册名：")
        f = open("db", 'r')
        for line in f:
            line_list = line.split("|")

            if user == line_list[0]:
                print("重复的用户名")
                return False
        for i in range(5):
            pwd = input("请输入密码：")
            pwd1 = input("请确认密码:")
            if pwd == pwd1:
                print("注册成功")
                break
            else:
                print("前后输入密码不一致,请重新输入")


        register(user, pwd)


main()