python入门之实例-用户登录、注册

用户密码存储文件db（其中用户和密码之间用$符合隔开）：

admin$123456
root$sdfk9f24
chy$654321

代码如下：

def login(username,password):
    """
    用于验证用户名密码
    验证成功返回True，验证失败返回False
    """
    f = open("db","r",encoding='utf-8')
    for line in f:
        #strip函数去掉两头的空格，换行符
        #split按照指定参数分割成列表
        line = line.strip()
        line_list = line.split("$")
        if username == line_list[0] and password == line_list[1]:
            return True
    return False

def register(username,password):
    """
    用于用户注册
    注册成功返回True
    """
    with open("db",'a',encoding='utf-8') as f:
        temp = "
" + username + "$" + password
        f.write(temp)
    return True

def user_exist(username):
    """
    用于判断用户是否存在
    存在返回True，不存在返回False
    """
    with open("db","r",encoding='utf-8') as f:
        for line in f:
            line = line.strip()
            line_list = line.split("$")
            if line_list[0] == username:
                return True
    return False

def main():
    print("欢迎登录xxx系统")
    inp = input("1.登录；2.注册")
    if inp == "1":
        user = input("请输入用户：")
        pwd = input("请输入密码：")
        is_login = login(user,pwd)
        if is_login:
            print("登录成功")
        else:
            print("登录失败")
    elif inp == "2":
        user = input("请输入用户：")
        pwd = input("请输入密码：")
        is_exist = user_exist(user)
        if is_exist:
            print("用户名已存在，无法注册")
        else:
            register(user,pwd)
            print("注册成功")

main()