python3 用户登录注册(函数，文件操作实现)

#!/usr/bin/env python
# -*- coding:utf-8 -*-

#用函数和文件操作实现用户登录与用户注册

def login(username,password):
    """
    此函数用于用户登录验证
    :param username:用户名
    :param password:密码
    :return:True 表示用户验证成功，False 表示用户验证失败
    """
    with open("login.txt","r",encoding="utf-8") as f:
     for line in f:
        #默认strip无参数:删除字符串左右两边的空格或者换行符
        #有参数：删除两侧指定的值
        line_new=line.strip()
        #指定分隔符对字符串进行切片，最终输出一个列表
        line_list=line_new.split("&")
        if username==line_list[0] and password==line_list[1]:
            return True
     return False

def register(username,password):
    """
    此函数用于用户注册
    :param username: 用户名
    :param password: 密码
    :return: True表示用户注册成功
    """
    with open("login.txt","a",encoding="utf-8") as f:
        temp="\n"+username+"&"+password
        f.write(temp)
    return True


def user_exist(username):
    """
    此函数用于判断用户是否已注册
    :param username: 用户名
    :return: True表示注册成功，False表示注册失败
    """
    with open("login.txt", "r", encoding="utf-8") as f:
        for line in f:
            line_new=line.strip()
            line_list=line_new.split("&")
            if username==line_list[0]:
                return True
    return False

#定义一个主函数，执行操作
def main():
 print("欢迎登录XXX系统")
 inp=input("1,登录，2，注册:")
 user = input('请输入用户名:')
 pwd = input('请输入密码:')
 if inp=="1":
     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:
         result=register(user,pwd)
         if result:
             print("注册成功")
         else:
             print("注册失败")

main()