函数式登录与注册

def login(username,password):
    """
    用于用户登录
    :param username: 用户于用户名
    :param password: 用户输入密码
    :return: true，表示登录成功;false，登录失败
    """
    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: 默认返回none
    """
    f=open('db','a')
    temp='
'+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('请输入密码：')
        register(user,pwd)
main()