3j";ddlZddlZddlmZddlZddlZddlm Z m Z ddl m Z m Z mZddlmZmZddlmZmZdd lmZdd lmZd d gZd ZdZdZdZGddZdZGdde Z Gdde Z!Gdde Z"Gdde"Z#Gdde"Z$Gdde Z%Gd d!e Z&Gd"d#e Z'd$Zd%Zy)&N)StringIO)decodeencode) BaseHandlerregister unregister)REVERSE_TYPE_MAPTYPE_MAP) b64decode b64encode)register_handlers)unregister_handlersrrc nt|tjr|jSt |fi|SN) isinstancenpgenericitemr)objkwargss B/DATA/.local/lib/python3.12/site-packages/jsonpickle/ext/pandas.py pd_encoders,#rzz"xxz #   c t|fi|Sr)r)srs r pd_decoders ! v rc|sgSg}|d}d}|ddD]$}||k(r|dz }|j||g|}d}&|j||g|S)z Encodes a list of type codes using Run-Length Encoding (RLE). This allows for object columns in dataframes to contain items of different types without massively bloating the encoded representation. rrN)append) types_listencoded current_typecounttyps r rle_encoder&ss  Ga=L E!"~ ,  QJE NNL%0 1LE  NNL%() NrcHg}|D]\}}|j|g|z|S)z\ Decodes a Run-Length Encoded (RLE) list back into the original list of type codes. )extend) encoded_listdecodedr%r$s r rle_decoder+4s.G" Uuu}%# Nrc&eZdZdefdZddZdZy)PandasProcessoric ||_||_y)a :param size_threshold: nonnegative int or None valid values for 'size_threshold' are all nonnegative integers and None. If size_threshold is None, dataframes are always stored as csv strings :param compression: a compression module or None valid values for 'compression' are {zlib, bz2, None} if compression is None, no compression is applied N)size_threshold compression)selfr/r0s r__init__zPandasProcessor.__init__?s-&rNc|jft||jkDrN|jr.|jj|j }d|d<t ||d<d|d<n ||d<d|d<||d<|S)NTcompvaluesFtxtmeta)r/lenr0compressrr )r1bufdatar7s rflatten_pandaszPandasProcessor.flatten_pandasLs    *s3x$:M:M/M&&// =#V &s^DNDK DNDKV  rc|jddr|d}nIt|d}|jddr)|jj|j }|jdi}||fS)Nr6Tr5r4Fr7)getr r0 decompressr)r1r;r:r7s rrestore_pandaszPandasProcessor.restore_pandasZsn 88E4 x.CDN+Cxx&&&11#6==?xx#T{rr)__name__ __module__ __qualname__zlibr2r<r@rrr-r->s&)t '  rr-c|j|jdid}|jddg}g}i}g}i}i}|jD]u\} } | jdr|j | | || <.| jdr t || <I| jdr|j | d || <q| || <wt |||| ||fS) NdtypesFresetheaderrdatetimecomplex timedeltaobject)dtyperJ parse_dates converters)restorer>items startswithr rLdict) r7context meta_dtypesrJrPrQ timedeltasparse_datetime_v2rOkvs rmake_read_csv_paramsr\fs//$((8R"8/FKXXh $FKJJ E!!#1 << #   q !#$ a \\) $#JqM \\+ &   a E!HE!H$ KJ    rc,eZdZeZdZdZdZy)PandasDfHandlerct}t|jtjr;|jDcgc] }t |}}|jj }d}n2|jj}|jj}d}t|jtjrE|jjDcgc] }t |} }|jj } d} n2|jj} |jj} d} i} g} |jD]}||}|jj}|dk(r|jdjr7|jdj}|| |<| jdz|j| |<| jtj |d|j| |<| jtj |dt#| d}t%| }||| t#|d|| |d}t#| d}|j'|||}|Scc}wcc}w) NTFrNc`t|ttttt j fSr)rlistrUsettuplerndarrayxs rz)PandasDfHandler.flatten..sjT4eRZZ,PQrct|Sr)rres rrgz)PandasDfHandler.flatten..srpy/jpkeys) dtypes_rleindex index_namescolumns column_names is_multiindexis_multicolumns)r-rropd MultiIndexrcnamestolistnamermr5rOapplyanyr r r>rr&r<)r1rr;ppcolrorprridx index_valuesrnrq data_columns type_codescol_data dtype_nameserialized_values index_encoded rle_typesr7 data_encodeds rflattenzPandasDfHandler.flattens0   ckk2== 1-0[[9[cuSz[G9;;,,L"Okk((*G;;++L#O cii /25))2B2BC2B3E#J2BLC))//K M99++-L))..K!M  ;;C3xH!,,JX%>>Q#%)17J(K(R(R(T%(9L%%%g.)1(9L%%%hll:x&HI%-OO$5 S!!!(,,z8"DE).|$7 z* $"&gD1(*. l6   tT: }:Ds JJc|jj|\}} t|d} |d}t|}t|dd}|jd d r1tjj||jd  }n|}i} i} t||D]M\} } || } | d k(r| Dcgc] }t|}}|| | <.| | | <tj| d}|| | <Otj| }||_|j D]A} | j| d} t#j$|}|| j'||| <Ct|dd}|jdd r1tjj||jd }n&tj(||jd}||_d |vrT|jd d r"|jd |j _|S|jd |j _|S#t$r+t j dd|j |cYSwxYw#t$r+t j dd|j |cYSwxYwcc}w#t$rd| d}t j |YwxYw)NTrjzjsonpickle versions at and above v3.4 have a different encoding scheme for pandas dataframes. If you're not decoding an object encoded in pre-v3.4 jsonpickle, please file a bug report on our GitHub!) stacklevelrlzjsonpickle versions at and above v3.4 have a different encoding scheme for pandas dataframes. Please update your jsonpickle and re-encode these objects!rorrFrprurirNz9jsonpickle was unable to properly deserialize the column zL into its inferred dtype. Please file a bugreport on the jsonpickle GitHub! rmrqrnrw)rzr@r Exceptionwarningswarn restore_v3_3KeyErrorr+r>rsrt from_tupleszipr DataFramerorrOastypeIndexrmrurw)r1rrr7r~rrcolumns_decodedrodf_datarGr{ type_coderr col_values dtype_strdfrOmsgr}rms rrRzPandasDfHandler.restores!WW33C8 d *!,T:L *\*I * !it< 88%u -mm//txx'?0G&G!':6NC#C(HG#7?@xtfTlx @) ' ,00HE 's 7\\' " ::C 31I #+S'../3 d7m$7 88OU +MM--DHH]$;.EHH\0GHE T !xx)51#'88N#;   #'((>":  m * MM@  $$S) ) * * MMR  $$S) ) *8A$ #""%'IJ  c"  #s: H9I07J'1,J,91I-,I-01J$#J$,$KKc|jj|\}}t||j\}}}|j dd}|dj rt jt|fi|nt j}|D]} t j|| || <|j|}|jt|dd|r||j_|S)Ncolumn_level_namesr5rmT)inplace)rzr@r\rVr>striprsread_csvrr to_timedeltar set_indexrroru) r1r;csvr7paramsrXrYrrr{s rrzPandasDfHandler.restore_v3_3)sGG**40 T0DT4<<0X- -"XX&:DAH~##% KK 0 0 Coobg.BsG YY( ) VDM*D 9 1BJJ  rN)rArBrCr-rzrrRrrErrr^r^s  BBHZxrr^c&eZdZeZdZdZy)PandasSeriesHandlerc|j|d<|jj|jd|d<|jj|jd|d<|S)z/Flatten the index and values for reconstructionrwFrHrmr5)rwrVrrmr5)r1rr;s rrzPandasSeriesHandler.flattenBsVxxV  ,,SYYe,DW --cjj-FX rc|d}|jj|dd}|jj|dd}tj|||S)zRestore the flattened datarwrmFrHr5)rmrw)rVrRrsSeries)r1r;rwrmr5s rrRzPandasSeriesHandler.restoreJsXF| $$T']%$@%%d8nE%Byyu488rNrArBrCr-rzrrRrErrrr?s  B9rrcDeZdZeZej ZdZdZ dZ y)PandasIndexHandlercd|jiS)Nrwrr1rs r name_bundlerzPandasIndexHandler.name_bundlerVs!!rc|j|}tddt|ji|}t |j }|j j|||}|S)NrOrE)rrUstrrOrrvrzr<)r1rr; name_bundler7r:s rrzPandasIndexHandler.flattenYsY'', 8#cii.8K8SZZ\"ww%%c46 rc|jj|\}}|jdd}|jDcic]\}}|dvrd|tnd|}}}|j t |fd|i|}|Scc}}w)NrO>rwrurwc|SrrEres rrgz,PandasIndexHandler.restore..ds1r)rzr@r>rSrcindex_constructorr) r1r;r:r7rOrZr[rr|s rrRzPandasIndexHandler.restore`sGG**40 T$'  $1%% rs 880'?C  5 6!.%%P@vkvr9+9&4'1'$0$ [  +(K, < r