chore: reuse json util #712 (#713)

This commit is contained in:
Oleg A 2021-10-06 00:59:43 +03:00 committed by GitHub
parent daf085ff9e
commit 100848b889
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
3 changed files with 23 additions and 5 deletions

View file

@ -1,8 +1,8 @@
import json
import pathlib
import pickle
import typing
from aiogram.utils import json
from .memory import MemoryStorage

View file

@ -20,28 +20,46 @@ for json_lib in (RAPIDJSON, UJSON):
break
if mode == RAPIDJSON:
def dump(*args, **kwargs):
return json.dump(*args, **kwargs)
def load(*args, **kwargs):
return json.load(*args, **kwargs)
def dumps(data):
return json.dumps(data, ensure_ascii=False)
def loads(data):
return json.loads(data, number_mode=json.NM_NATIVE)
elif mode == UJSON:
def dump(*args, **kwargs):
return json.dump(*args, **kwargs)
def load(*args, **kwargs):
return json.load(*args, **kwargs)
def loads(data):
return json.loads(data)
def dumps(data):
return json.dumps(data, ensure_ascii=False)
else:
import json
def dump(*args, **kwargs):
return json.dump(*args, **kwargs)
def load(*args, **kwargs):
return json.load(*args, **kwargs)
def dumps(data):
return json.dumps(data, ensure_ascii=False)
def loads(data):
return json.loads(data)

View file

@ -26,7 +26,7 @@ class FakeTelegram(aresponses.ResponsesMockServer):
@staticmethod
def parse_data(message_data):
import json
from aiogram.utils import json
from aiogram.utils.payload import _normalize
_body = '{"ok":true,"result":' + json.dumps(_normalize(message_data)) + '}'