Cutlery/calibre
1
0
Fork 0
mirror of https://github.com/kovidgoyal/calibre.git synced 2025-07-08 02:34:06 -04:00
Code Issues Packages Projects Releases Wiki Activity

Serialization of FieldMetadata

Browse Source
This commit is contained in:
Kovid Goyal 2017-05-01 13:02:23 +05:30
parent 37309b8c56
commit 40bbe91c0c
No known key found for this signature in database
GPG Key ID: 06BC317B515ACE7C
1 changed files with 12 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
src/calibre/utils/serialize.py
View File

@ -8,6 +8,8 @@ import json, base64
from functools import partial from functools import partial
from datetime import datetime from datetime import datetime
import msgpack
from calibre.utils.iso8601 import parse_iso8601 from calibre.utils.iso8601 import parse_iso8601
MSGPACK_MIME = 'application/x-msgpack' MSGPACK_MIME = 'application/x-msgpack'
@ -23,16 +25,18 @@ def encoder(obj, for_json=False):
return encoded(0, unicode(obj.isoformat())) return encoded(0, unicode(obj.isoformat()))
if isinstance(obj, (set, frozenset)): if isinstance(obj, (set, frozenset)):
return encoded(1, tuple(obj)) return encoded(1, tuple(obj))
if obj.__class__.__name__ == 'Metadata': if hasattr(obj, '__calibre_serializable__'):
from calibre.ebooks.metadata.book.base import Metadata from calibre.ebooks.metadata.book.base import Metadata
from calibre.library.field_metadata import FieldMetadata, fm_as_dict
if isinstance(obj, Metadata): if isinstance(obj, Metadata):
from calibre.ebooks.metadata.book.serialize import metadata_as_dict from calibre.ebooks.metadata.book.serialize import metadata_as_dict
return encoded(2, metadata_as_dict(obj, encode_cover_data=for_json)) return encoded(2, metadata_as_dict(obj, encode_cover_data=for_json))
elif isinstance(obj, FieldMetadata):
return encoded(3, fm_as_dict(obj))
raise TypeError('Cannot serialize objects of type {}'.format(type(obj))) raise TypeError('Cannot serialize objects of type {}'.format(type(obj)))
def msgpack_dumps(data): def msgpack_dumps(data):
import msgpack
return msgpack.packb(data, use_bin_type=True, default=encoder) return msgpack.packb(data, use_bin_type=True, default=encoder)
@ -55,10 +59,16 @@ def decode_metadata(x, for_json):
return obj return obj
def decode_field_metadata(x, for_json):
from calibre.library.field_metadata import fm_from_dict
return fm_from_dict(x)
decoders = ( decoders = (
lambda x, fj: parse_iso8601(x, assume_utc=True), lambda x, fj: parse_iso8601(x, assume_utc=True),
lambda x, fj: set(x), lambda x, fj: set(x),
decode_metadata, decode_metadata,
decode_field_metadata,
) )
@ -70,7 +80,6 @@ def decoder(obj, for_json=False):
def msgpack_loads(data): def msgpack_loads(data):
import msgpack
return msgpack.unpackb(data, encoding='utf-8', object_hook=decoder) return msgpack.unpackb(data, encoding='utf-8', object_hook=decoder)