AHjJdZddlmZddlmZmZGddeZd dZd dZ y) a An encoding / decoding format suitable for serializing data structures to binary. This is based on https://en.wikipedia.org/wiki/Bencode with some extensions. The following data types may be encoded: - None - int - bool - bytes - str - list - tuple - dict ) annotations)AnyCallableceZdZdZy) DecodeErrorzA problem decoding data.N)__name__ __module__ __qualname____doc__N/root/tools/cai/cai_env/lib/python3.12/site-packages/textual/_binary_encode.pyrrs"r rc d d}d d}d d}dd}dd}d fd }d fd }d fd }td |t|t|t|t|t |t |t|i d fd  |S)zEncodes a data structure into bytes. Args: data: Data structure Returns: A byte string encoding the data. cy)z Encodes a None value. Args: datum: Always None. Returns: None encoded. Nr )_datums r encode_nonezdump..encode_none&sr c|rdSdS)z Encode a boolean value. Args: datum: The boolean value to encode. Returns: The encoded bytes. TFr datums r encode_boolzdump..encode_bool2st&$&r c d|zS)z Encode an integer value. Args: datum: The integer value to encode. Returns: The encoded bytes. si%ier rs r encode_intzdump..encode_int>sr c"dt||fzS)z Encode a bytes value. Args: datum: The bytes value to encode. Returns: The encoded bytes. s%i:%s)lenrs r encode_byteszdump..encode_bytesJs3u:u---r cD|jd}dt||fzS)z Encode a string value. Args: datum: The string value to encode. Returns: The encoded bytes. utf-8ss%i:%s)encoder)r encoded_datas r encode_stringzdump..encode_stringVs'||G, C -|<<ddjfd|DzS)z Encode a list value. Args: datum: The list value to encode. Returns: The encoded bytes. sl%ser c3.K|] }|ywNr .0elementr!s r z,dump..encode_list..m!Gg&/!Gjoinrr!s r encode_listzdump..encode_listc!G!GGGGr c>ddjfd|DzS)z Encode a tuple value. Args: datum: The tuple value to encode. Returns: The encoded bytes. st%ser c3.K|] }|ywr&r r's rr*z-dump..encode_tuple..yr+r,r-r/s r encode_tuplezdump..encode_tupleor1r cZddjfd|jDzS)z Encode a dictionary value. Args: datum: The dictionary value to encode. Returns: The encoded bytes. sd%ser c3JK|]\}}d||fzyw)s%s%sNr )r(keyvaluer!s rr*z,dump..encode_dict..s," 7AsEGvc{F5M2 2" s #)r.itemsr/s r encode_dictzdump..encode_dict{s0" EJ[[]"    r Ncd t|}||S#t$r tddwxYw)zRecursively encode data. Args: datum: Data suitable for encoding. Raises: TypeError: If `datum` is not one of the supported types. Returns: Encoded data bytes. zCan't encode {datum!r}N)typeKeyError TypeError)rdecoderENCODERSs rr!zdump..encodesD @tE{+Gu~ @454 ? @s/)rNonereturnbytes)rboolrBrC)rintrBrC)rrCrBrC)rstrrBrC)rlistrBrC)rtuplerBrC)rdictrBrC)robjectrBrC)r<rDrErCrFrGrHrI) datarrrrr#r0r4r:r@r!s @@rdumprLss  '  . = H H   T K k Z | ] k | k 2H$ $<r c " tts tdt d d fd d fd d fd d fd }d fd d fd }d fd }d fd }d fd }|||||d d ddd fd S)zLoad an encoded data structure from bytes. Args: encoded: Encoded data in bytes. Raises: DecodeError: If an error was encountered decoding the string. Returns: Decoded data. z must be bytesrcBk\r tddz}dz |S)zGet an encoded byte and advance position. Raises: DecodeError: If the end of the data was reached Returns: A bytes object with a single byte. zMore data expected)r) characterencoded max_positionpositions rget_bytezload..get_bytes8 | #23 3Hx!|4 A r cdzS)zGet the byte at the current position, but don't advance position. Returns: A bytes object with a single byte. rOr )rQrSsr peek_bytezload..peek_bytes x(Q,//r cT|z}t||k7r td|z |S)zGet a number of bytes of encode data. Args: size: Number of bytes to retrieve. Raises: DecodeError: If there aren't enough bytes. Returns: A bytes object. sMissing bytes in {bytes_data!r})rr)size bytes_datarQrSs r get_byteszload..get_bytess=X48 z?d "@A ADr cZd}x}dk7r||z }x}dk7rt|S)zWDecode an int from the encoded data. Returns: An integer. r erE) int_bytesbyterTs r decode_intzload..decode_ints>  z!td*  I z!td*9~r cfx}dk7r||z }x}dk7rt|}|S)zcDecode a bytes string from the encoded data. Returns: A bytes object. :r]) size_bytesr_ bytes_stringrTrZs r decode_byteszload..decode_bytessC  z!td* $ J z!td* Z1 r cd}x}dk7r||z }x}dk7rt|}|jdd}|S)zgDecode a (utf-8 encoded) string from the encoded data. Returns: A string. r rbr replace)errors)rEdecode)rcr_rddecoded_stringrTrZs r decode_stringzload..decode_strings]  z!td* $ J z!td* Z1 %,,WY,Gr cvg}|j}dk7r|dk7r|S)zEDecode a list. Returns: A list of data. r\)appendelements add_elementrirTrVs r decode_listzload..decode_lists= "$oo kT!  !kT! r cg}|j}dk7r|dk7rt|S)zODecode a tuple. Returns: A tuple of decoded data. r\)rmrHrns r decode_tuplezload..decode_tuplesB "$oo kT!  !kT! Xr ci}|j}dk7r|dk7r|S)zMDecode a dict. Returns: A dict of decoded data. r\) __setitem__rns r decode_dictzload..decode_dict"sC *,** kT! &( +kT! r cy)NTr r r rzload..5r cy)NFr r r rrxzload..6ryr cyr&r r r rrxzload..7ryr )isltdrrrcXjx}d}||S|S)zMRecursively decode data. Returns: Decoded data. N)get)initialr?DECODERSrerTs rrizload..decode:s4 ,,(*4wd; ?( (yr )rBrC)rXrErBrC)rBrE)rcrCrBrC)rBrF)rBz list[object])rBztuple[object, ...])rBzdict[object, object])rBrJ) isinstancerCr>r)rQr`rkrqrsrvrrirerTrZrRrVrSs` @@@@@@@@rloadrs gu %((wrs+$# #)#JZ\r