dnrops
/
remarshal
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

style: sort encode and decode functions

This commit is contained in:
D. Bohdan 2023-09-14 14:19:46 +00:00
parent f1900464d6
commit ff55a8b54a
1 changed files with 17 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

34
remarshal.py
View File

@ -337,6 +337,15 @@ def traverse(
Document = Union[bool, bytes, datetime.datetime, Mapping, None, Sequence, str]
def _decode_cbor(input_data: bytes) -> Document:
try:
doc = cbor2.loads(input_data)
return cast(Document, doc)
except cbor2.CBORDecodeError as e:
msg = f"Cannot parse as CBOR ({e})"
raise ValueError(msg)
def _decode_json(input_data: bytes) -> Document:
try:
doc = json.loads(
@ -358,15 +367,6 @@ def _decode_msgpack(input_data: bytes) -> Document:
raise ValueError(msg)
def _decode_cbor(input_data: bytes) -> Document:
try:
doc = cbor2.loads(input_data)
return cast(Document, doc)
except cbor2.CBORDecodeError as e:
msg = f"Cannot parse as CBOR ({e})"
raise ValueError(msg)
def _decode_toml(input_data: bytes) -> Document:
try:
# Remove TOML Kit's custom classes.
@ -493,6 +493,14 @@ def _stringify_special_keys(key: Any) -> Any:
return str(key)
def _encode_cbor(data: Document) -> bytes:
try:
return bytes(cbor2.dumps(data))
except cbor2.CBOREncodeError as e:
msg = f"Cannot convert data to CBOR ({e})"
raise ValueError(msg)
def _json_default(obj: Any) -> str:
if isinstance(obj, datetime.datetime):
return obj.isoformat()
@ -541,14 +549,6 @@ def _encode_msgpack(data: Document) -> bytes:
raise ValueError(msg)
def _encode_cbor(data: Document) -> bytes:
try:
return bytes(cbor2.dumps(data))
except cbor2.CBOREncodeError as e:
msg = f"Cannot convert data to CBOR ({e})"
raise ValueError(msg)
def _encode_toml(
data: Mapping[Any, Any],
*,