简单的登录注册函数

'''
1. 编写注册函数
   将用户信息保存到文件内，用户信息可保存为`nick:123|sean:456|tank:789`
2. 编写登录函数
   从文件内读取用户信息进行身份识别
'''
#注册函数
def register():
    print('注册账号')
    n_inp = input('请输入注册名:')
    p_inp = input('请输入注册密码')
    with open('作业用户信息.txt','a') as f:
        if ':' in 'n_inp' or ':' in p_inp:    #不允许用户输入文件中包含的特殊字符
            print('输入错误,不能包含":"')
        elif '|' in 'n_inp' or '|' in p_inp:
            print('输入错误,不能包含"|"')
        else:
            f.write(f'{n_inp}:{p_inp}|')
            print('恭喜,注册成功')
# register()
#登录函数
def login():
    log_name = input('请输入登录名:')
    log_pwd = input('请输入登录密码:')
    dic_info = {}   #定义字典存储互相匹配的账号密码
    with open('作业用户信息.txt','r') as f1:
        data = f1.read()    #读取用户信息文件
    user_info = data.split('|')    #切分用户信息以列表方式赋值给定义的变量
    for i in user_info[0:len(user_info)-1]: #循环遍历第一次切分的列表中的字符串,注意剔除最后会切分出来的空字符
        j = i.split(':')    #将遍历得到的字符串按关键字第二次切分成列表存储
        dic_info[j[0]] = j[1]
    if log_name in dic_info.keys() and log_pwd in dic_info.values() and dic_info[log_name] == log_pwd:  #判断用户名密码是否正确
        print('登录成功')
    else:
        print('登录失败')
login()