PapaTutuWawa/etesync-server
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

msgpack route: use the encode/decode functions from the utils module.

Browse Source
This commit is contained in:
Tom Hacohen 2020-12-29 15:44:52 +02:00
parent dbdff06e68
commit 62eb46ec4e
2 changed files with 7 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
etebase_fastapi/msgpack.py
View File

@ -1,10 +1,11 @@
import typing as t
import msgpack
from fastapi.routing import APIRoute, get_request_handler
from pydantic import BaseModel
from starlette.requests import Request
from starlette.responses import Response
from .utils import msgpack_encode, msgpack_decode
class MsgpackRequest(Request):
media_type = "application/msgpack"
@ -12,7 +13,7 @@ class MsgpackRequest(Request):
async def json(self) -> bytes:
if not hasattr(self, "_json"):
body = await super().body()
self._json = msgpack.unpackb(body, raw=False)
self._json = msgpack_decode(body)
return self._json
@ -25,9 +26,7 @@ class MsgpackResponse(Response):
if isinstance(content, BaseModel):
content = content.dict()
ret = msgpack.packb(content, use_bin_type=True)
assert ret is not None
return ret
return msgpack_encode(content)
class MsgpackRoute(APIRoute):

4
etebase_fastapi/utils.py
View File

@ -47,7 +47,9 @@ def is_collection_admin(collection, user):
def msgpack_encode(content):
return msgpack.packb(content, use_bin_type=True)
ret = msgpack.packb(content, use_bin_type=True)
assert ret is not None
return ret
def msgpack_decode(content):