python--函数式登录程序

# Author:Tim Gu
def login(username,password):
    """这个login函数用于用户登录的验证
    :param username:用户输入的用户名
    :param password:用户输入的密码
    :return:True:表示登陆成功，False：表示登陆失败
    """
    f = open('db.txt','r')
    for line in f:
        line_list = line.strip('
').split('|')
        print(line_list)
        if line_list[0] == username and line_list[1] == password:
            return True
    return False


def register(username,password):
    """
    :param username: 用户输入的用户名
    :param password: 用户输入的密码
    :return: 无返回
    """
    f = open('db.txt','a')
    temp = '%s|%s'%(username,password)
    f.write('
%s'%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':
        print('注册')
        username = input('请输入用户名：')
        password = input('请输入密码：')
        register(username,password)
        print('注册完成')

main()