import pickle
import json
#-------------------------------------------pickle模块-----------------------------------
#存入变量, dumps(obj)返回存入的字节
dic = {'age': 23, 'job': 'student'}
byte_data = pickle.dumps(dic)
# out -> b'\x80\x03}q\x00(X\x03\x00\x00\...'
print(byte_data)
#从变量读取数据,数据以字节保存在了byte_data变量中,需要再次使用的时候使用loads函数就行了。
obj = pickle.loads(byte_data)
print(obj)
#存储在文件中
#也可以存在文件中,使得对象持久化。使用的是dump和load函数,注意和上面的区别,少了s。
#由于pickle写入的是二进制数据,所以打开方式需要以wb和rb的模式。
# 序列化
with open('abc.pkl', 'wb') as f:
dic = {'age': 23, 'job': 'student'}
pickle.dump(dic, f)
# 反序列化
with open('abc.pkl', 'rb') as f:
aa = pickle.load(f)
print(aa)
print(type(aa)) # <class 'dict'>
#序列化用户自定义对象
class Person:
def __init__(self, name, age, job):
self.name = name
self.age = age
self.job = job
def work(self):
print(self.name, 'is working...')
# 将实例存储在变量中,当然也能存在文件中
a_person = Person('abc', 22, 'waiter')
person_abc = pickle.dumps(a_person)
p = pickle.loads(person_abc)
p.work()
# 将类本身存储在变量中,loads的时候返回类本身,而非它的一个实例
class_Person = pickle.dumps(Person)
Person = pickle.loads(class_Person)
p = Person('Bob', 23, 'Student')
p.work()
# 下面这个例子演示的就是将类存储在文件中
# 序列化
with open('person.pkl', 'wb') as f:
pickle.dump(Person, f)
# 反序列化
with open('person.pkl', 'rb') as f:
Person = pickle.load(f)
aa = Person('gg', 23, '6')
aa.work()
#----------------------------------------json 模块--------------------------------------------
#ickle可以很方便地序列化所有对象。不过json作为更为标准的格式,具有更好的可读性(pickle是二进制数据)和跨平台性。
#是个不错的选择。json使用的四个函数名和pickle一致。
#序列化为字符串
#可以看到,dumps函数将对象转换成了字符串。loads函数又将其恢复成字典。
dic = {'age': 23, 'job': 'student'}
dic_str = json.dumps(dic)
print(type(dic_str), dic_str)
# out: <class 'str'> {"age": 23, "job": "student"}
dic_obj = json.loads(dic_str)
print(type(dic_obj), dic_obj)
# out: <class 'dict'> {'age': 23, 'job': 'student'}
#存储为json文件
dic = {'age': 23, 'job': 'student'}
with open('abc.json', 'w', encoding='utf-8') as f:
json.dump(dic, f)
with open('abc.json', encoding='utf-8') as f:
obj = json.load(f)
print(obj)
#存储自定义对象
#需要编写一个转换函数可以将对象的字段转换为字典与字典转换为对象字段的函数,作为dump与load的一个参数传入
class Person:
def __init__(self, name, age, job):
self.name = name
self.age = age
self.job = job
def work(self):
print(self.name, 'is working...')
#这两个转换函数是写在外面的,这点要注意,转换函数的名称可以随便取,有意义就行
def person2dict(person):
return {'name': person.name,
'age': person.age,
'job': person.job}
def dict2person(dic):
return Person(dic['name'], dic['age'], dic['job'])
#使用如下
with open('abc.json', 'w', encoding='utf-8') as f:
json.dump(aa, f, default=person2dict)
with open('abc.json', encoding='utf-8') as f:
obj = json.load(f, object_hook=dict2person)
print(obj.name, obj.age, obj.job)
obj.work()