mirror of
https://github.com/msgpack/msgpack-python.git
synced 2025-11-09 05:51:02 +00:00
Make strict_map_key default to True (#392)
This commit is contained in:
parent
0fc0eb2f16
commit
d8e3cf0563
8 changed files with 19 additions and 20 deletions
|
|
@ -131,7 +131,7 @@ cdef inline int get_data_from_buffer(object obj,
|
|||
|
||||
|
||||
def unpackb(object packed, *, object object_hook=None, object list_hook=None,
|
||||
bint use_list=True, bint raw=False, bint strict_map_key=False,
|
||||
bint use_list=True, bint raw=False, bint strict_map_key=True,
|
||||
unicode_errors=None,
|
||||
object_pairs_hook=None, ext_hook=ExtType,
|
||||
Py_ssize_t max_str_len=-1,
|
||||
|
|
@ -221,9 +221,7 @@ cdef class Unpacker(object):
|
|||
Otherwise, unpack to Python str by decoding with UTF-8 encoding (default).
|
||||
|
||||
:param bool strict_map_key:
|
||||
If true, only str or bytes are accepted for map (dict) keys.
|
||||
It's False by default for backward-compatibility.
|
||||
But it will be True from msgpack 1.0.
|
||||
If true (default), only str or bytes are accepted for map (dict) keys.
|
||||
|
||||
:param callable object_hook:
|
||||
When specified, it should be callable.
|
||||
|
|
@ -305,7 +303,7 @@ cdef class Unpacker(object):
|
|||
self.buf = NULL
|
||||
|
||||
def __init__(self, file_like=None, *, Py_ssize_t read_size=0,
|
||||
bint use_list=True, bint raw=False, bint strict_map_key=False,
|
||||
bint use_list=True, bint raw=False, bint strict_map_key=True,
|
||||
object object_hook=None, object object_pairs_hook=None, object list_hook=None,
|
||||
unicode_errors=None, Py_ssize_t max_buffer_size=0,
|
||||
object ext_hook=ExtType,
|
||||
|
|
|
|||
|
|
@ -175,9 +175,7 @@ class Unpacker(object):
|
|||
Otherwise, unpack to Python str by decoding with UTF-8 encoding (default).
|
||||
|
||||
:param bool strict_map_key:
|
||||
If true, only str or bytes are accepted for map (dict) keys.
|
||||
It's False by default for backward-compatibility.
|
||||
But it will be True from msgpack 1.0.
|
||||
If true (default), only str or bytes are accepted for map (dict) keys.
|
||||
|
||||
:param callable object_hook:
|
||||
When specified, it should be callable.
|
||||
|
|
@ -249,7 +247,7 @@ class Unpacker(object):
|
|||
read_size=0,
|
||||
use_list=True,
|
||||
raw=False,
|
||||
strict_map_key=False,
|
||||
strict_map_key=True,
|
||||
object_hook=None,
|
||||
object_pairs_hook=None,
|
||||
list_hook=None,
|
||||
|
|
|
|||
|
|
@ -92,7 +92,7 @@ def test_array32():
|
|||
|
||||
def match(obj, buf):
|
||||
assert packb(obj) == buf
|
||||
assert unpackb(buf, use_list=0) == obj
|
||||
assert unpackb(buf, use_list=0, strict_map_key=False) == obj
|
||||
|
||||
|
||||
def test_match():
|
||||
|
|
|
|||
|
|
@ -5,7 +5,7 @@ from msgpack import unpackb
|
|||
|
||||
|
||||
def check(src, should, use_list=0, raw=True):
|
||||
assert unpackb(src, use_list=use_list, raw=raw) == should
|
||||
assert unpackb(src, use_list=use_list, raw=raw, strict_map_key=False) == should
|
||||
|
||||
|
||||
def testSimpleValue():
|
||||
|
|
|
|||
|
|
@ -87,11 +87,11 @@ def test_max_map_len():
|
|||
d = {1: 2, 3: 4, 5: 6}
|
||||
packed = packb(d)
|
||||
|
||||
unpacker = Unpacker(max_map_len=3)
|
||||
unpacker = Unpacker(max_map_len=3, strict_map_key=False)
|
||||
unpacker.feed(packed)
|
||||
assert unpacker.unpack() == d
|
||||
|
||||
unpacker = Unpacker(max_map_len=2)
|
||||
unpacker = Unpacker(max_map_len=2, strict_map_key=False)
|
||||
with pytest.raises(UnpackValueError):
|
||||
unpacker.feed(packed)
|
||||
unpacker.unpack()
|
||||
|
|
|
|||
|
|
@ -33,7 +33,10 @@ def test_decode_pairs_hook():
|
|||
packed = packb([3, {1: 2, 3: 4}])
|
||||
prod_sum = 1 * 2 + 3 * 4
|
||||
unpacked = unpackb(
|
||||
packed, object_pairs_hook=lambda l: sum(k * v for k, v in l), use_list=1
|
||||
packed,
|
||||
object_pairs_hook=lambda l: sum(k * v for k, v in l),
|
||||
use_list=1,
|
||||
strict_map_key=False,
|
||||
)
|
||||
assert unpacked[1] == prod_sum
|
||||
|
||||
|
|
@ -70,10 +73,10 @@ def bad_complex_decoder(o):
|
|||
def test_an_exception_in_objecthook1():
|
||||
with raises(DecodeError):
|
||||
packed = packb({1: {"__complex__": True, "real": 1, "imag": 2}})
|
||||
unpackb(packed, object_hook=bad_complex_decoder)
|
||||
unpackb(packed, object_hook=bad_complex_decoder, strict_map_key=False)
|
||||
|
||||
|
||||
def test_an_exception_in_objecthook2():
|
||||
with raises(DecodeError):
|
||||
packed = packb({1: [{"__complex__": True, "real": 1, "imag": 2}]})
|
||||
unpackb(packed, list_hook=bad_complex_decoder, use_list=1)
|
||||
unpackb(packed, list_hook=bad_complex_decoder, use_list=1, strict_map_key=False)
|
||||
|
|
|
|||
|
|
@ -14,7 +14,7 @@ from msgpack import packb, unpackb, Unpacker, Packer, pack
|
|||
|
||||
|
||||
def check(data, use_list=False):
|
||||
re = unpackb(packb(data), use_list=use_list)
|
||||
re = unpackb(packb(data), use_list=use_list, strict_map_key=False)
|
||||
assert re == data
|
||||
|
||||
|
||||
|
|
@ -166,7 +166,7 @@ def testMapSize(sizes=[0, 5, 50, 1000]):
|
|||
bio.write(packer.pack(i * 2)) # value
|
||||
|
||||
bio.seek(0)
|
||||
unpacker = Unpacker(bio)
|
||||
unpacker = Unpacker(bio, strict_map_key=False)
|
||||
for size in sizes:
|
||||
assert unpacker.unpack() == dict((i, i * 2) for i in range(size))
|
||||
|
||||
|
|
@ -186,7 +186,7 @@ def test_pairlist():
|
|||
pairlist = [(b"a", 1), (2, b"b"), (b"foo", b"bar")]
|
||||
packer = Packer()
|
||||
packed = packer.pack_map_pairs(pairlist)
|
||||
unpacked = unpackb(packed, object_pairs_hook=list)
|
||||
unpacked = unpackb(packed, object_pairs_hook=list, strict_map_key=False)
|
||||
assert pairlist == unpacked
|
||||
|
||||
|
||||
|
|
|
|||
|
|
@ -132,7 +132,7 @@ def test_unpack_tell():
|
|||
pack(m, stream)
|
||||
offsets.append(stream.tell())
|
||||
stream.seek(0)
|
||||
unpacker = Unpacker(stream)
|
||||
unpacker = Unpacker(stream, strict_map_key=False)
|
||||
for m, o in zip(messages, offsets):
|
||||
m2 = next(unpacker)
|
||||
assert m == m2
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue