U 淀g@sdZddlmZddlZddlZeedr|ddlmZzddlm Z Wn e k rfddlm Z YnXdZ Gd d d Z nd Z dd lmZ d dZddlmZmZmZmZmZddlmZmZdZdZdZdZdZdZdZdZ dZ!dZ"dZ#e$e%fddZ&ddZ'ddZ(dZ)de)e!fdde!fdde!fdd e"fdd!e"fdd"e"fd#d$de)fd%d&d'd(d)d*d+dd,e"fdd-e"fdd.e"fd/d0e"fd1d2e"fde)e fdde fdde fddefddefddefddefd3Z*Gd4d5d5Z+Gd6d7d7Z,dS)8z.Fallback pure Python implementation of msgpack)datetimeNpypy_version_info) newlist_hint) BytesBuilder) StringBuilderTc@s&eZdZd ddZddZddZdS) StringIOcCs,|r tt||_|j|nt|_dSN)rlenbuilderappendselfsrJ/opt/hc_python/lib/python3.8/site-packages/pip/_vendor/msgpack/fallback.py__init__szStringIO.__init__cCs6t|tr|}nt|tr&t|}|j|dSr ) isinstance memoryviewtobytes bytearraybytesr r r rrrwrites    zStringIO.writecCs |jSr )r buildrrrrgetvalue!szStringIO.getvalueN)r)__name__ __module__ __qualname__rrrrrrrrs rF)BytesIOcCsgSr r)sizerrr(rr!) BufferFull OutOfData ExtraData FormatError StackError)ExtType TimestampicCs(|||kr|||kS|||kSdSr r)objttypetuplerrr_check_type_strict?s  r2cCst|}|jdkrtd|S)Nr"z$cannot unpack from multi-byte object)ritemsize ValueError)r.viewrrr_get_data_from_bufferFs r6cKsztddt|i|}||z |}Wn2tk rHtdYntk r^tYnX|rvt || |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_sizezUnpack failed: incomplete input)N) Unpackerr feed_unpackr$r4RecursionErrorr'_got_extradatar%_get_extradata)packedkwargsunpackerretrrrunpackbMs     rB>H>IBbz>Hbz>Ib)r,z>f)z>d)r*rD)r,rE)rGz>Q)r"b)r*z>h)r,z>i)rGz>qb1sb2sb4srGb8sb16s)c@seZdZdZdddddddddddedddddfdd Zd d Zd d ZddZddZ ddZ d*ddZ d+ddZ ddZ efddZddZddZeZd d!Zd"d#Zd$d%Zd&d'Zd(d)ZdS),r8a Streaming unpacker. Arguments: :param file_like: File-like object having `.read(n)` method. If specified, unpacker reads serialized data from it and `.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). :param bool strict_map_key: If true (default), only str or bytes are accepted for map (dict) keys. :param object_hook: When specified, it should be callable. Unpacker calls it with a dict argument after unpacking msgpack map. (See also simplejson) :param 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() 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. NrTFi@cCs| dkr d} |dkrd|_nt|js.td||_d|_t|_d|_d|_| sVd} | dkrb| } |dkrn| }|dkrz| }|dkr| d}|dkr| }| |_ ||j krt d |pt |j d |_ t ||_t ||_| |_||_d|krd ksnt d ||_| |_||_||_| |_| |_||_||_||_||_d|_| dk r\t| s\td |dk rxt|sxtd|dk rt|std|dk r|dk rtdt| stddS)NstrictTz!`file_like.read` must be callableFrirkr*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_sizer4min _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)rrq read_sizeuse_listraw timestampstrict_map_key object_hookobject_pairs_hook list_hookunicode_errorsr7ext_hook max_str_len max_bin_len max_array_len max_map_len max_ext_lenrrrrsl      zUnpacker.__init__cCsr|js tt|}t|j|jt||jkr4t|jdkrb|jd|j=|j|j8_d|_|j |dSNr) rmAssertionErrorr6r rrrsrur#rtextend)r next_bytesr5rrrr9Bs  z Unpacker.feedcCs"|j|j|j7_|j|_dS)z)Gets rid of the used parts of the buffer.N)rrsrtrrrr_consumeQszUnpacker._consumecCs|jt|jkSr )rsr rrrrrrr<VszUnpacker._got_extradatacCs|j|jdSr )rrrsrrrrr=YszUnpacker._get_extradatacCs|j|dd}||S)NFraise_outofdata)_readrrnrArrr read_bytes\szUnpacker.read_bytescCs8|j||d|j}|j|||}|t||_|S)Nr)_reserversrrr )rrrirArrrras zUnpacker._readcCst|j|j|}|dkr dS|jr2|j|_t|jdkr`|jd|j=|j|j8_d|_| }|t|j|jkr~t|dkrt|j |}|j |}|sqt |t st|j|7_|t|8}q~t|j||jkr|rd|_tdSr)r rrrsrmrtr$rur#maxrwrqrorrr)rrr remain_bytes to_read_bytes read_datarrrris0   zUnpacker._reservecCs,t}d}d}|d|j|j}|jd7_|d@dkrD|}n|d@dkr`d|dA}n|d@dkr|d@}t}||jkrt|d |jd ||}n||d @d kr|d @}t}||j krt|d|j d n@|d @dkr"|d @}t }||j kr"t|d|j d n|dkr4d}n|dkrFd}n|dkrXd}nd|krpdkrnnt |\}}}||t |dkrt||j|jd}n |j|j}|j|7_||jkrt|d|jd ||}n&d|krdkrnnlt |\}}}||t||j|j\}}|j|7_||jkrvt|d|jd ||}nd|krdkrnnZt |\}}||t |dkrt||j|jd}n |j|j}|j|7_n(d|krdkrnnjt |\}}}|j|krFt|d|jd ||dt||j|j\}}|j|d7_nd|krd kr"nnt |\}}}||t |dkrt||j|j\}n |j|j}|j|7_||jkrt|d |jd ||}nd!|kr:d"krnn^t |\}}}||t||j|j\}|j|7_||j kr"t|d|j d nd#|krd$krnn^t |\}}}||t||j|j\}|j|7_||j kr"t|d|j d n td%||||fS)&Nrr"rkz exceeds max_str_len()z exceeds max_array_len(z exceeds max_map_len(FTrOrQz exceeds max_bin_len(rRrTz exceeds max_ext_len(rUr^r_rcrdrfrgrhrirjzUnknown header: 0x%x)TYPE_IMMEDIATErrrrsTYPE_RAWrr4r TYPE_ARRAYrTYPE_MAPr_MSGPACK_HEADERSr struct unpack_fromrrr&)rtyprr.rHr fmtLrrr _read_headers                             zUnpacker._read_headerc s~\}}}|tkr*|tkr&td|S|tkrF|tkrBtd|S|tkr|tkrrt|D]}tq^dSt |}t|D]}| t qj dk r |}j r|St|S|tkr|tkrt|D]}ttqdSjdk r fddt|D}ni}t|D]b}t }jrht|ttfkrhtdtt|t|tr~t|}t ||<q,jdk r|}|S|tkrdS|tkrjrt|}n|dj}|S|tkrt|S|tkrl|dkr\tt|} j dkr.| !Sj d krB| "Sj d krV| #S| Sn$|t|S|t%kszt&|S) NzExpected arrayz Expected mapc3s"|]}ttfVqdSr )r: EX_CONSTRUCT).0_rrr sz#Unpacker._unpack..z%s is not allowed for map keyutf_8rkr"r*r+)'rEX_READ_ARRAY_HEADERrr4EX_READ_MAP_HEADERrEX_SKIPranger:rr rr~r|r1rrzr0strrrsysinternrrrydecoder{TYPE_BINTYPE_EXTr) from_bytesr}to_unix to_unix_nano to_datetimerrr) rexecuterrr.rrArkeytsrrrr:s~                          zUnpacker._unpackcCs|Sr rrrrr__iter__5szUnpacker.__iter__cCsTz|t}||WStk r8|tYntk rNtYnXdSr )r:rrr$ StopIterationr;r'rrArrr__next__8s zUnpacker.__next__cCs|t|dSr )r:rrrrrrskipEs z Unpacker.skipcCs4z|t}Wntk r&tYnX||Sr )r:rr;r'rrrrrunpackIs  zUnpacker.unpackcCs|t}||Sr )r:rrrrrrread_array_headerQs zUnpacker.read_array_headercCs|t}||Sr )r:rrrrrrread_map_headerVs zUnpacker.read_map_headercCs|jSr )rrrrrtell[sz Unpacker.tell)T)T)rrr__doc__r(rr9rr<r=rrrrrr:rrnextrrrrrrrrrr8sHa [  "` J r8c@seZdZdZd#ddZeeefddZd d Z d d Z d dZ ddZ ddZ ddZddZefddZddZddZddZdd Zd!d"ZdS)$Packera9 MessagePack Packer Usage:: packer = Packer() astream.write(packer.pack(a)) astream.write(packer.pack(b)) Packer's constructor has some keyword arguments: :param default: When specified, it should be callable. 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. :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. NFTcCsV||_||_||_||_t|_t||_|p0d|_|dk rLt |sLt d||_ dS)Nrlzdefault must be callable) _strict_types _use_float _autoreset _use_bin_typerrrrx _datetimer{rnrp_default)rdefaultuse_single_float autoreset use_bin_type strict_typesrrrrrrs   zPacker.__init__c Csd}|jr|}t}nttf}|dkr,td|dkr@|jdS||trf|rZ|jdS|jdS||trd|krdkrnn|jt d|Sd |krdkrnn|jt d |Sd|krd krnn|jt d d |Sd|krdkr*nn|jt dd|Sd |krBdkr\nn|jt dd|Sd|krtdkrnn|jt dd|Sd|krdkrnn|jt dd|Sd|krdkrnn|jt dd|Sd|kr dkr$nn|jt dd|Sd |krt+d=|qdS)?NFrzrecursion limit exceededrBirHrBBrWiz>Bbr[>BHrXiz>Bhr\>BIrYiz>Bir]lz>BQrZlz>Bqr^TzInteger value out of rangez%s is too largezutf-8zString is too largezMemoryview is too largez>BfrUz>BdrVrkr"r*r,rGrM>BBrRrSrTzCannot serialize z where tzinfo=None),rlistr1r4rrrrxintrpackr OverflowErrorrrr r0r_pack_bin_headerrencoder{_pack_raw_headerrnbytesfloatrr(r)to_bytescodedatarr_pack_array_headerr_packdict_pack_map_pairsitemsr _DateTimetzinfo from_datetimerp) rr. nest_limitcheckcheck_type_strict default_used list_typesrrrrrrrrr s                                   z Packer._packcCsFz||Wnt|_YnX|jrB|j}t|_|SdSr )r rrrrr)rr.rArrrr6s z Packer.packcCs0|t|||jr,|j}t|_|SdSr )r r rrrrr)rpairsrArrrpack_map_pairsAs  zPacker.pack_map_pairscCs6|dkr t|||jr2|j}t|_|SdSNr)r4rrrrrrrrrrpack_array_headerHs  zPacker.pack_array_headercCs6|dkr t|||jr2|j}t|_|SdSr)r4_pack_map_headerrrrrrrrrrpack_map_headerQs  zPacker.pack_map_headercCsJt|tstdd|kr&dks0ntdt|tsBtdt|}|dkrZtd|dkrp|jd n|d kr|jd n|d kr|jd n|dkr|jdnt|dkr|jdn^|dkr|jdt d|n<|dkr|jdt d|n|jdt d||jt d||j|dS)Nztypecode must have int type.rztypecode should be 0-127zdata must have bytes typerzToo large datar"rr*rr,rrGrrMrrrrrDrE) rrrpr4rr rrrrr)rtypecoderrrrr pack_ext_typeZs4   zPacker.pack_ext_typecCsh|dkr |jtdd|S|dkr>|jtdd|S|dkr\|jtdd |Std dS) NrrrrrrgrrrhzArray is too largerrrrrr4rrrrrrwszPacker._pack_array_headercCsh|dkr |jtdd|S|dkr>|jtdd|S|dkr\|jtdd |Std dS) NrrrrrrirrrjzDict is too larger!r"rrrrszPacker._pack_map_headercCs<|||D](\}}|||d|||dqdS)Nr")rr )rrrrkvrrrr s  zPacker._pack_map_pairscCs|dkr"|jtdd|nn|jrH|dkrH|jtdd|nH|dkrh|jtdd |n(|d kr|jtd d |ntd dS)NrrrrrrdrrrerrrfzRaw is too large)rrrrrrr4r"rrrrszPacker._pack_raw_headercCsv|js||S|dkr.|jtdd|S|dkrL|jtdd|S|dkrj|jtdd |Std dS) NrrrOrrrPrrrQzBin is too large)rrrrrrrr4r"rrrrs zPacker._pack_bin_headercCs |jS)z/Return internal buffer contents as bytes object)rrrrrrrrsz Packer.bytescCs t|_dS)zYReset internal buffer. This method is useful only when autoreset=False. N)rrrrrrrresetsz Packer.resetcCstrt|S|jSdS)zReturn view of internal buffer.N)USING_STRINGBUILDERrrrr getbufferrrrrr's zPacker.getbuffer)NFTTFFN)rrrrrDEFAULT_RECURSE_LIMITrr2r rrrrr rrr rrrr%r'rrrrr_s4F  |       r)-rrr rrhasattrZ__pypy__rZ__pypy__.buildersrr ImportErrorr&rior exceptionsr#r$r%r&r'extr(r)rrrrrrrrrrr(r0r1r2r6rB_NO_FORMAT_USEDrr8rrrrrs     [