需求:
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)
来源: http://www.bubuko.com/infodetail-2572510.html