o
ai @ s d Z ddlmZ ddlZddlZejd dkZer"eefZ dd Z
n
eZ e Z e
Zdd Z
ejdk r8eZd d
Znd d
Zeed rjdd
lmZ zddlmZ W n ey^ ddlmZ Y nw dZG dd deZn dZddl m Z dd Zddl m Z m Z m!Z!m"Z"m#Z# ddl$m%Z%m&Z& dZ'dZ(dZ)dZ*dZ+dZ,dZ-dZ.dZ/d Z0d Z1e2e3fd d Z4d d! Z5d"d# Z6ejd$k rd+d%d&Z7nej8Z7G d'd( d(eZ9G d)d* d*eZ:dS ),z.Fallback pure Python implementation of msgpack )datetimeN c C | S N) iteritemsd r /builddir/build/BUILDROOT/alt-python310-pip-21.3.1-5.el8.x86_64/opt/alt/python310/lib/python3.10/site-packages/pip/_vendor/msgpack/fallback.pydict_iteritems r c C r r )itemsr r r r
r r ) c C s. t | jdkot| jd to| jd dS )N r z maximum recursion depth exceeded)lenargs
isinstancestr
startswither r r
_is_recursionerror s
r c C s dS )NTr r r r r
r ' pypy_version_info) newlist_hint) BytesBuilder)
StringBuilderTc @ s& e Zd Zd ddZdd Zdd ZdS )
StringIO c C s. |rt t|| _| j| d S t | _d S r )r r builderappendselfsr r r
__init__7 s zStringIO.__init__c C s6 t |tr
| }n t |trt|}| j| d S r )r
memoryviewtobytes bytearraybytesr r! r" r r r
write> s
zStringIO.writec C
| j S r )r buildr# r r r
getvalueE s
zStringIO.getvalueN)r )__name__
__module__ __qualname__r% r* r. r r r r
r 6 s
r F)BytesIOc C s g S r r )sizer r r
M s r4 r )
BufferFull OutOfData ExtraData FormatError
StackError)ExtType Timestampr r i c C s$ |||u r || |v S || |u S r r )objttypetupler r r
_check_type_strictd s rA c C s t | }|jdkr
td|S )Nr z$cannot unpack from multi-byte object)r& itemsize
ValueError)r= viewr r r
_get_data_from_bufferk s
rE c
K s t ddt| i|}|| z| }W n ty! td ty3 } zt|r.t d}~ww | r?t
|| |S )aw
Unpack an object from `packed`.
Raises ``ExtraData`` when *packed* contains extra bytes.
Raises ``ValueError`` when *packed* is incomplete.
Raises ``FormatError`` when *packed* is not valid msgpack.
Raises ``StackError`` when *packed* contains too nested.
Other exceptions can be raised during unpacking.
See :class:`Unpacker` for options.
Nmax_buffer_sizez Unpack failed: incomplete inputr ) Unpackerr feed_unpackr6 rC RecursionErrorr r9 _got_extradatar7 _get_extradata)packedkwargsunpackerretr r r r
unpackbr s
rQ )r c C s t | t||S )z0Explicit type cast for legacy struct.unpack_from)struct unpack_fromr) )fbor r r
_unpack_from s rY c @ s e Zd ZdZdddddddddddedddddfdd Zd
d Zd d
Zdd Zdd Z dd Z
d*ddZ d*ddZ e
fddZe
fddZd d Zd d ZeZd d! Zd"d# Zd$d% Zd&d' Zd(d) ZdS )+rG a Streaming unpacker.
Arguments:
:param file_like:
File-like object having `.read(n)` method.
If specified, unpacker reads serialized data from it and :meth:`feed()` is not usable.
:param int read_size:
Used as `file_like.read(read_size)`. (default: `min(16*1024, max_buffer_size)`)
:param bool use_list:
If true, unpack msgpack array to Python list.
Otherwise, unpack to Python tuple. (default: True)
:param bool raw:
If true, unpack msgpack raw to Python bytes.
Otherwise, unpack to Python str by decoding with UTF-8 encoding (default).
:param int timestamp:
Control how timestamp type is unpacked:
0 - Timestamp
1 - float (Seconds from the EPOCH)
2 - int (Nanoseconds from the EPOCH)
3 - datetime.datetime (UTC). Python 2 is not supported.
:param bool strict_map_key:
If true (default), only str or bytes are accepted for map (dict) keys.
:param callable object_hook:
When specified, it should be callable.
Unpacker calls it with a dict argument after unpacking msgpack map.
(See also simplejson)
:param callable object_pairs_hook:
When specified, it should be callable.
Unpacker calls it with a list of key-value pairs after unpacking msgpack map.
(See also simplejson)
:param str unicode_errors:
The error handler for decoding unicode. (default: 'strict')
This option should be used only when you have msgpack data which
contains invalid UTF-8 string.
:param int max_buffer_size:
Limits size of data waiting unpacked. 0 means 2**32-1.
The default value is 100*1024*1024 (100MiB).
Raises `BufferFull` exception when it is insufficient.
You should set this parameter when unpacking data from untrusted source.
:param int max_str_len:
Deprecated, use *max_buffer_size* instead.
Limits max length of str. (default: max_buffer_size)
:param int max_bin_len:
Deprecated, use *max_buffer_size* instead.
Limits max length of bin. (default: max_buffer_size)
:param int max_array_len:
Limits max length of array.
(default: max_buffer_size)
:param int max_map_len:
Limits max length of map.
(default: max_buffer_size//2)
:param int max_ext_len:
Deprecated, use *max_buffer_size* instead.
Limits max size of ext type. (default: max_buffer_size)
Example of streaming deserialize from file-like object::
unpacker = Unpacker(file_like)
for o in unpacker:
process(o)
Example of streaming deserialize from socket::
unpacker = Unpacker(max_buffer_size)
while True:
buf = sock.recv(1024**2)
if not buf:
break
unpacker.feed(buf)
for o in unpacker:
process(o)
Raises ``ExtraData`` when *packed* contains extra bytes.
Raises ``OutOfData`` when *packed* is incomplete.
Raises ``FormatError`` when *packed* is not valid msgpack.
Raises ``StackError`` when *packed* contains too nested.
Other exceptions can be raised during unpacking.
Nr TFi @c C s |
d u rd}
|d u rd| _ nt|jstd|| _d| _ t | _d| _d| _| s+d} |
dkr1| }
|dkr7| }|dkr=| }|dkrE| d }|dkrK| }| | _ || j krWt
d |p^t | j d
| _ t
|| _t
|| _|
| _|| _d| kr}d kst
d t
d || _| | _|| _|| _| | _|
| _|| _|| _|| _|| _d| _ | d urt| std
|d urt|std|d urt|std|d ur|d urtdt| stdd S )NstrictTz!`file_like.read` must be callableFr irZ r z.read_size must be smaller than max_buffer_sizei @ r ztimestamp must be 0..3z`list_hook` is not callablez `object_hook` is not callablez#`object_pairs_hook` is not callablez8object_pairs_hook and object_hook are mutually exclusivez`ext_hook` is not callable) _feedingcallableread TypeError file_liker( _buffer_buff_i_buf_checkpoint_max_buffer_sizerC min
_read_sizebool_raw_strict_map_key_unicode_errors _use_list
_timestamp
_list_hook _object_hook_object_pairs_hook _ext_hook _max_str_len _max_bin_len_max_array_len _max_map_len _max_ext_len_stream_offset)r# r` read_sizeuse_listraw timestampstrict_map_key object_hookobject_pairs_hook list_hookunicode_errorsrF ext_hook max_str_len max_bin_len
max_array_len max_map_len max_ext_lenr r r
r% sv
zUnpacker.__init__c C sr | j sJ t|}t| j| j t| | jkrt| jdkr1| jd | j= | j| j8 _d| _| j| d S Nr ) r\ rE r ra rb rd r5 rc extend)r#
next_bytesrD r r r
rH U s
z
Unpacker.feedc C s" | j | j| j 7 _ | j| _dS )z+ Gets rid of the used parts of the buffer. N)rv rb rc r- r r r
_consumed s zUnpacker._consumec C s | j t| jk S r )rb r ra r- r r r
rK i zUnpacker._got_extradatac C s | j | jd S r )ra rb r- r r r
rL l r zUnpacker._get_extradatac C s | j |dd}| |S )NFraise_outofdata)_readr r# nrP r r r
read_byteso s zUnpacker.read_bytesc C s8 | j ||d | j}| j||| }|t| | _|S )Nr )_reserverb ra r )r# r r irP r r r
r t s
zUnpacker._readc C s t | j| j | }|dkrd S | jr| j| _t| jdkr0| jd | j= | j| j8 _d| _| }|dkr^t| j|}| j |}|sFnt
|t sMJ | j|7 _|t |8 }|dks7t | j|| j k ro|rqd| _td S d S r ) r ra rb r\ rc r6 maxrf r` r^ r r) )r# r r remain_bytes
to_read_bytes read_datar r r
r | s0
zUnpacker._reservec C s
t }d}d }| d | j| j }| jd7 _|d@ dkr"|}n|d@ dkr0d|dA }n|d@ dkrO|d@ }t}|| jkrHtd || j| |}n|d
@ d kri|d @ }t}|| j krgtd
|| j n|d
@ dkr|d @ }t
}|| j krtd|| j n|dkrd }nw|dkrd}no|dkrd}ng|dkrt }| d | j| j }| jd7 _|| j
krtd|| j
f | |}n:|dkrt }| d td| j| jd }| jd7 _|| j
krtd|| j
f | |}n |dkr,t }| d td| j| jd }| jd7 _|| j
kr%td|| j
f | |}n|d kr_t}| d td | j| j\}}| jd7 _|| jkrXtd || jf | |}n|d krt}| d td!| j| j\}}| jd 7 _|| jkrtd || jf | |}np|d"krt}| d# td$| j| j\}}| jd#7 _|| jkrtd || jf | |}n=|d%kr| d td&| j| jd }| jd7 _n |d'kr| d( td)| j| jd }| jd(7 _n|d*kr| d | j| j }| jd7 _n|d+kr5| d td| j| jd }| jd7 _n|d,krR| d td| j| jd }| jd7 _n|d-kro| d( td.| j| jd }| jd(7 _n|d/kr| d td0| j| jd }| jd7 _nv|d1kr| d td2| j| jd }| jd7 _nY|d3kr| d td4| j| jd }| jd7 _n<|d5kr| d( td6| j| jd }| jd(7 _n |d7krt}| jdk rtd d| jf | d td8| j| j\}}| jd7 _n|d9kr?t}| jdk r'td d| jf | d td:| j| j\}}| jd 7 _n|d;krmt}| jdk rUtd d| jf | d# td<| j| j\}}| jd#7 _n|d=krt}| jd(k rtd d(| jf | d> td?| j| j\}}| jd>7 _ng|d@krt}| jdAk rtd dA| jf | dB tdC| j| j\}}| jdB7 _n9|dDkrt}| d | j| j }| jd7 _|| jkrtd || j| |}n |dEkr%t}| d td| j| j\}| jd7 _|| jkr td || j| |}n|dFkrTt}| d td| j| j\}| jd7 _|| jkrNtd || j| |}n|dGkr~t}| d td| j| j\}| jd7 _|| j kr}td
|| j n|dHkrt}| d td| j| j\}| jd7 _|| j krtd
|| j nZ|dIkr| d td| j| j\}| jd7 _|| j krtd|| j t
}n0|dJkr| d td| j| j\}| jd7 _|| j krtd|| j t
}ntdK| |||fS )LNr r rZ z%s exceeds max_str_len(%s) z %s exceeds max_array_len(%s)z%s exceeds max_map_len(%s) F T z%s exceeds max_bin_len(%s) r >H r< >I Bbz%s exceeds max_ext_len(%s) r z>Hb r z>Ib z>f z>d z>Q rW z>h z>i z>q b1s b2s b4s b8s b16s zUnknown header: 0x%x)TYPE_IMMEDIATEr ra rb TYPE_RAWrq rC r
TYPE_ARRAYrs TYPE_MAPrt TYPE_BINrr rY TYPE_EXTru r8 )r# executetypr r= rW Lr r r
_read_header s
zUnpacker._read_headerc
sr |\}}}|tkr|tkrtd|S |tkr$|tkr"td|S |tkr`|tkr:t|D ]} t q0d S t |}t|D ]
}|
t qB j d urW |} j
r\|S t|S |tkr|tkr{t|D ] } t t qld S jd ur fddt|D }|S i }t|D ]0} t } jrt|ttfvrtdtt| tst|tu rt|} t ||< q jd urш |}|S |tkrd S |tkr jrt|}|S |d j }|S |t krt|S |t kr0|dkr(t t|} j!dkr| " S j!d kr | # S j!d
kr&| $ S | S %|t|S |t&ks7J |S ) NzExpected arrayz Expected mapc 3 s$ | ]
} t tfV qd S r )rI EX_CONSTRUCT).0_r- r r
s
z#Unpacker._unpack..z %s is not allowed for map keyutf_8rZ r r r )'r EX_READ_ARRAY_HEADERr rC EX_READ_MAP_HEADERr EX_SKIPxrangerI r r! r rm rk r@ ro ri r? unicoder) r PY2sysinternrn r rh decoderj r r r;
from_bytesrl to_unix to_unix_nano to_datetimerp r )
r# r r r r= r rP r keytsr r- r
rI p s
zUnpacker._unpackc C s | S r r r- r r r
__iter__ r zUnpacker.__iter__c C sD z | t}| |W S ty | t ty! tw r )rI r r r6
StopIterationrJ r9 r# rP r r r
__next__ s
zUnpacker.__next__c C s | t | d S r )rI r r r- r r r
skip s
z
Unpacker.skipc C s. z| t}W n ty tw | |S r )rI r rJ r9 r r r r r
unpack s zUnpacker.unpackc C | t}| |S r )rI r r r r r r
read_array_header
zUnpacker.read_array_headerc C r r )rI r r r r r r
read_map_header r zUnpacker.read_map_headerc C s | j S r )rv r- r r r
tell s z
Unpacker.tell)T)r/ r0 r1 __doc__r: r% rH r rK rL r r r r r rI r r nextr r r r r r r r r
rG sL a
]
UM rG c @ s e Zd ZdZ d#ddZeeefddZd d
Z d d Z
d
d Z dd Z dd Z
dd Zdd ZefddZdd Zdd Zd d Zd d Zd!d" ZdS )$Packera8
MessagePack Packer
Usage::
packer = Packer()
astream.write(packer.pack(a))
astream.write(packer.pack(b))
Packer's constructor has some keyword arguments:
:param callable default:
Convert user type to builtin type that Packer supports.
See also simplejson's document.
:param bool use_single_float:
Use single precision float type for float. (default: False)
:param bool autoreset:
Reset buffer after each pack and return its content as `bytes`. (default: True).
If set this to false, use `bytes()` to get content and `.reset()` to clear buffer.
:param bool use_bin_type:
Use bin type introduced in msgpack spec 2.0 for bytes.
It also enables str8 type for unicode. (default: True)
:param bool strict_types:
If set to true, types will be checked to be exact. Derived classes
from serializable types will not be serialized and will be
treated as unsupported type and forwarded to default.
Additionally tuples will not be serialized as lists.
This is useful when trying to implement accurate serialization
for python types.
:param bool datetime:
If set to true, datetime with tzinfo is packed into Timestamp type.
Note that the tzinfo is stripped in the timestamp.
You can get UTC datetime with `timestamp=3` option of the Unpacker.
(Python 2 is not supported).
:param str unicode_errors:
The error handler for encoding unicode. (default: 'strict')
DO NOT USE THIS!! This option is kept for very specific usage.
Example of streaming deserialize from file-like object::
unpacker = Unpacker(file_like)
for o in unpacker:
process(o)
Example of streaming deserialize from socket::
unpacker = Unpacker()
while True:
buf = sock.recv(1024**2)
if not buf:
break
unpacker.feed(buf)
for o in unpacker:
process(o)
Raises ``ExtraData`` when *packed* contains extra bytes.
Raises ``OutOfData`` when *packed* is incomplete.
Raises ``FormatError`` when *packed* is not valid msgpack.
Raises ``StackError`` when *packed* contains too nested.
Other exceptions can be raised during unpacking.
NFTc C sf || _ || _|| _|| _t | _tr|rtdt|| _ |p d| _
|d ur.t |s.t d|| _
d S )Nz%datetime is not supported in Python 2r[ zdefault must be callable)
_strict_types
_use_float
_autoreset
_use_bin_typer ra r rC rg _datetimerj r] r_ _default)r# defaultuse_single_float autoreset use_bin_type strict_typesr r r r r
r% , s
zPacker.__init__c C s d}| j r
|}t}nttf} |dk rtd|d u r!| jdS ||tr4|r.| jdS | jdS ||tr6d| krDdk rPn n
| jt d |S d
| krZdk rfn n
| jt d |S d| krpd kr}n n | jt d
d|S d| krdk rn n | jt dd|S d | k rdkrn n | jt dd|S d| krdk rn n | jt dd|S d| k rdkrn n | jt dd|S d| krdk rn n | jt d d |S d| k rd krn n | jt d d |S d!| krdk r!n n | jt d"d#|S |s2| j
d ur2|
|}d}qt d$||t t
fr[t|}|d%krPtd&t|j | | | j|S ||tr|d'| j}t|}|d%krutd(| | | j|S ||trt||j }|d%krtd)| | | j|S ||tr| jr| jt d*d+|S | jt d,d-|S ||ttfrk||trd.}| } n|j }|j } t |t sJ t | t sJ t| }
|
d/kr| jd0 n]|
d1kr| jd2 nQ|
d3kr| jd4 nE|
d5kr | jd6 n9|
d7kr,| jd8 n-|
d kr=| jt d9d:|
n |
dkrN| jt dd;|
n | jt dd<|
| jt d | | j| d S |||rt|}| !| t"|D ]
} | #|| |d/ q~d S ||t$r| %t|t&||d/ S | j'r||t(r|j)d urt*|}d/}q|s| j
d ur|
|}d/}qt+d=|f )>NFTr zrecursion limit exceeded r BirW r BBr iz>Bbr >BHr i z>Bhr >BIr i z>Bir l z>BQr l z>Bqr zInteger value out of range z%s is too largezutf-8zString is too largezMemoryview is too largez>Bfr z>Bdr rZ r r r< r r >BBr r r zCannot serialize %r),r listr@ rC ra r* rg int_typesrT packr
OverflowErrorr) r( r r? r/ _pack_bin_headerr encoderj _pack_raw_headerr& rB floatr r: r; to_bytescodedatar int_pack_array_headerr _packdict_pack_map_pairsr r _DateTimetzinfo
from_datetimer_ ) r# r=
nest_limitcheckcheck_type_strict default_used
list_typesr r r r r r r r
r D s
z Packer._packc C s@ z| | W n t | _ | jr | j }t | _|S d S r )r r ra r r. )r# r= rP r r r
r s
z Packer.packc C s0 | t|| | jr| j }t | _|S d S r )r r r ra r. r )r# pairsrP r r r
pack_map_pairs s
zPacker.pack_map_pairsc C 6 |dkrt | | | jr| j }t | _|S d S Nr
)rC r r ra r. r r r r r
pack_array_header
zPacker.pack_array_headerc C r+ r, )rC _pack_map_headerr ra r. r r r r r
pack_map_header r. zPacker.pack_map_headerc C sN t |ts tdd| krdkstd tdt |ts$tdt|}|dkr0td|dkr;| jd nZ|d
krF| jd nO|d krQ| jd
nD|dkr\| jd n9|dkrg| jd n.|dkrx| jdt d| n |dkr| jdt d| n | jdt d| | jt d| | j| d S )Nz typecode must have int type.r ztypecode should be 0-127zdata must have bytes typer zToo large datar r r r r< r
r r r r r r r r r )
r r r_ rC r) r ra r* rT r )r# typecoder r r r r
pack_ext_type s8
zPacker.pack_ext_typec C d |dkr| j tdd| S |dkr | j tdd|S |dkr.| j tdd |S td
) Nr r r r r r r r r zArray is too largera r* rT r rC r# r r r r
r zPacker._pack_array_headerc C r7 ) Nr r r r r r r r r zDict is too larger8 r9 r r r
r/ r: zPacker._pack_map_headerc C s< | | |D ]\}}| ||d | ||d qd S )Nr )r/ r )r# r r) r$ kvr r r
r s
zPacker._pack_map_pairsc C s |dkr| j tdd| d S | jr&|dkr&| j tdd| d S |dkr7| j tdd | d S |d
krH| j td d | d S td
)Nr r r r r r r r r r r r zRaw is too large)ra r* rT r r rC r9 r r r
r s zPacker._pack_raw_headerc C sr | j s| |S |dkr| jtdd|S |dkr&| jtdd|S |dkr5| jtdd |S td
) Nr r r r r r r r r zBin is too large)r r ra r* rT r rC r9 r r r
r # s
zPacker._pack_bin_headerc C r+ )z/Return internal buffer contents as bytes object)ra r. r- r r r
r) / s
z Packer.bytesc C s t | _dS )zYReset internal buffer.
This method is useful only when autoreset=False.
N)r ra r- r r r
reset3 s z Packer.resetc C s t str
t| S | j S )z Return view of internal buffer.)USING_STRINGBUILDERr r& r) ra getbufferr- r r r
r? : s
zPacker.getbuffer)NFTTFFN)r/ r0 r1 r r% DEFAULT_RECURSE_LIMITr rA r r r* r- r0 r6 r r/ r r r r) r= r? r r r r
r s6 F
z r )r );r r r! r rT version_infor r longr r r r ranger RuntimeErrorrJ r hasattrZ__pypy__r Z__pypy__.buildersr r ImportErrorr> objectr ior2
exceptionsr5 r6 r7 r8 r9 extr: r; r r r r r r r r r r r@ r? r@ rA rE rQ rY rU rG r r r r r
sj
S