Python json练习_读写文件函数

需求：

1、用户的注册信息写在json文件中

2、将读写文件的重复操作提取为函数，简洁代码

实现：

import json
def op_data(filename,dic=None):
    if dic: #如果有内容，就写文件
        with open(filename,'w',encoding='utf-8') as fw:
            json.dump(dic,fw,ensure_ascii=False,indent=4)
    else: #没内容，就读文件
        with open(filename,encoding='utf-8') as fr:
            return  json.load(fr) #将json文件转为字典
FILENAME = 'user_info.json'
all_user = op_data(FILENAME)
for i in range(3):
    choice = input('输入选择：1、注册，2、删除')
    if choice == '1': #注册
        username = input('输入注册用户名').strip()
        passwd = input('输入注册密码').strip()
        if username in all_user:
            print('用户已存在')
        else:
            all_user[username] = passwd
            op_data(FILENAME,all_user)
    elif choice == '2': #删除
        username = input('输入删除用户名').strip()
        all_user.pop(username)
        op_data(FILENAME, all_user)