Sau đây là một ví dụ về cách bạn có thể viết và đọc một tệp pickle. Lưu ý rằng nếu bạn tiếp tục thêm dữ liệu kén chọn vào tệp, bạn sẽ cần phải tiếp tục đọc từ tệp cho đến khi bạn tìm thấy những gì bạn muốn hoặc một ngoại lệ được tạo ra khi đến cuối tệp. Đó là những gì chức năng cuối cùng làm.
import os
import pickle
PICKLE_FILE = 'pickle.dat'
def main():
add_to_pickle(PICKLE_FILE, 123)
add_to_pickle(PICKLE_FILE, 'Hello')
add_to_pickle(PICKLE_FILE, None)
add_to_pickle(PICKLE_FILE, b'World')
add_to_pickle(PICKLE_FILE, 456.789)
for item in read_from_pickle(PICKLE_FILE):
print(repr(item))
os.remove(PICKLE_FILE)
def add_to_pickle(path, item):
with open(path, 'ab') as file:
pickle.dump(item, file, pickle.HIGHEST_PROTOCOL)
def read_from_pickle(path):
with open(path, 'rb') as file:
try:
while True:
yield pickle.load(file)
except EOFError:
pass
if __name__ == '__main__':
main()
open('filename', 'wb')