def _get_binary(
data: Any, view: Any, position: int, obj_end: int, opts: CodecOptions, dummy1: Any
) -> Tuple[Union[Binary, uuid.UUID], int]:
"""Decode a BSON binary to bson.binary.Binary or python UUID."""
length, subtype = _UNPACK_LENGTH_SUBTYPE_FROM(data, position)
position += 5
if subtype == 2:
length2 = _UNPACK_INT_FROM(data, position)[0]
position += 4
if length2 != length - 4:
raise InvalidBSON("invalid binary (st 2) - lengths don't match!")
length = length2
end = position + length
if length < 0 or end > obj_end:
raise InvalidBSON("bad binary object length")
# Convert UUID subtypes to native UUIDs.
if subtype in ALL_UUID_SUBTYPES:
uuid_rep = opts.uuid_representation
binary_value = Binary(data[position:end], subtype)
if (
(uuid_rep == UuidRepresentation.UNSPECIFIED)
or (subtype == UUID_SUBTYPE and uuid_rep != STANDARD)
or (subtype == OLD_UUID_SUBTYPE and uuid_rep == STANDARD)
):
return binary_value, end
return binary_value.as_uuid(uuid_rep), end
# Decode subtype 0 to 'bytes'.
if subtype == 0:
value = data[position:end]
else:
value = Binary(data[position:end], subtype)
return value, end
def _binary_or_uuid(data, subtype, json_options):
# special handling for UUID
if subtype in ALL_UUID_SUBTYPES:
uuid_representation = json_options.uuid_representation
binary_value = Binary(data, subtype)
if uuid_representation == UuidRepresentation.UNSPECIFIED:
return binary_value
if subtype == UUID_SUBTYPE:
# Legacy behavior: use STANDARD with binary subtype 4.
uuid_representation = UuidRepresentation.STANDARD
elif uuid_representation == UuidRepresentation.STANDARD:
# subtype == OLD_UUID_SUBTYPE
# Legacy behavior: STANDARD is the same as PYTHON_LEGACY.
uuid_representation = UuidRepresentation.PYTHON_LEGACY
return binary_value.as_uuid(uuid_representation)
if PY3 and subtype == 0:
return data
return Binary(data, subtype)
def _binary_or_uuid(data: Any, subtype: int,
json_options: JSONOptions) -> Union[Binary, uuid.UUID]:
# special handling for UUID
if subtype in ALL_UUID_SUBTYPES:
uuid_representation = json_options.uuid_representation
binary_value = Binary(data, subtype)
if uuid_representation == UuidRepresentation.UNSPECIFIED:
return binary_value
if subtype == UUID_SUBTYPE:
# Legacy behavior: use STANDARD with binary subtype 4.
uuid_representation = UuidRepresentation.STANDARD
elif uuid_representation == UuidRepresentation.STANDARD:
# subtype == OLD_UUID_SUBTYPE
# Legacy behavior: STANDARD is the same as PYTHON_LEGACY.
uuid_representation = UuidRepresentation.PYTHON_LEGACY
return binary_value.as_uuid(uuid_representation)
if subtype == 0:
return cast(uuid.UUID, data)
return Binary(data, subtype)
def _get_binary(data, view, position, obj_end, opts, dummy1):
"""Decode a BSON binary to bson.binary.Binary or python UUID."""
length, subtype = _UNPACK_LENGTH_SUBTYPE_FROM(data, position)
position += 5
if subtype == 2:
length2 = _UNPACK_INT_FROM(data, position)[0]
position += 4
if length2 != length - 4:
raise InvalidBSON("invalid binary (st 2) - lengths don't match!")
length = length2
end = position + length
if length < 0 or end > obj_end:
raise InvalidBSON('bad binary object length')
# Convert UUID subtypes to native UUIDs.
# TODO: PYTHON-2245 Decoding should follow UUID spec in PyMongo 4.0+
if subtype in ALL_UUID_SUBTYPES:
uuid_representation = opts.uuid_representation
binary_value = Binary(data[position:end], subtype)
if uuid_representation == UuidRepresentation.UNSPECIFIED:
return binary_value, end
if subtype == UUID_SUBTYPE:
# Legacy behavior: use STANDARD with binary subtype 4.
uuid_representation = UuidRepresentation.STANDARD
elif uuid_representation == UuidRepresentation.STANDARD:
# subtype == OLD_UUID_SUBTYPE
# Legacy behavior: STANDARD is the same as PYTHON_LEGACY.
uuid_representation = UuidRepresentation.PYTHON_LEGACY
return binary_value.as_uuid(uuid_representation), end
# Decode subtype 0 to 'bytes'.
if subtype == 0:
value = data[position:end]
else:
value = Binary(data[position:end], subtype)
return value, end
