d)ddlZddlZddlZddlZddlZddlZddlmZddl m Z ddl m Z ddl m Z ddlmZddlmZe ZGd d eeZGd d eZejddZejd dZej d!dZej d"dZejdZdZdZdZdZdZ dZdZ!dS)#N)curried)Callable)sanitize_dataframe)sanitize_geo_interface)AltairDeprecationWarning)PluginRegistrycLeZdZddiZedZejdZdS)DataTransformerRegistryconsolidate_datasetsTc|jdSNr _global_settings)selfs 1lib/python3.11/site-packages/altair/utils/data.pyr z,DataTransformerRegistry.consolidate_datasetss$%;<<c||jd<dSrr)rvalues rr z,DataTransformerRegistry.consolidate_datasetss8=4555rN)__name__ __module__ __qualname__rpropertyr setterrrr r sW.5 ==X= >>! >>>rr ceZdZdZdS) MaxRowsErrorz+Raised when a data model has too many rows.N)rrr__doc__rrrrr6s55Drrct|t|dr'|jddkr|jd}n[|j}nSt|tjr|}n6t|t rd|vr |d}n|St|dr|}|&t||krtd|d |S) zzRaise MaxRowsError if the data model has more than max_rows. If max_rows is None, then do not perform any check. __geo_interface__typeFeatureCollectionfeaturesvalues __dataframe__NzHThe number of rows in your dataset is greater than the maximum allowed (z). See https://altair-viz.github.io/user_guide/large_datasets.html for information on how to plot large datasets, including how to install third-party data management tools and, in the right circumstance, disable the restriction) check_data_typehasattrr! isinstancepd DataFramedictlenr)datamax_rowsr%s r limit_rowsr0<s Dt())   !& )-@ @ @+J7FF+FF D", ' ' D$   t  (^FFK  ' 'F h 6 6 A)1 A A A    Krcxt|t|tjr|||St|t rJd|vrD|d}|r|nt |t|z}tj||}d|iSdSt|drt}| |}|r|nt |t|z}tjtt||}| |SdS)zBReduce the size of the data model by sampling without replacement.)nfracr%r&N)r'r)r*r+sampler,intr-randomr(import_pyarrow_interchangefrom_dataframerangetake)r.r2r3r%pipa_tableindicess rr4r4]s5D$ %%&{{QT{*** D$   & t  (^F3Cs6{{ 233A]61--Ff% %    ' '& ' ) )$$T** 1AAD3x==011-c(mm 4 4a88}}W%%% &&r altair-datajson{prefix}-{hash}.{extension}c2t|}t|}||||}t|d5}||dddn #1swxYwYt j||ddidS)zQ Write the data model to a .json file and return a url based data model. prefixhash extensionwNr"r?urlformat)_data_to_json_string_compute_data_hashrJopenwriteospathjoin)r.rDrFfilenameurlpath data_json data_hashfs rto_jsonrWrs%T**I"9--If9 RRH h    7<<22vv>N O OOA))A-0A-csvc2t|}t|}||||}t|d5}||dddn #1swxYwYt j||ddidS)zFWrite the data model to a .csv file and return a url based data model.rCrGNr"rYrH)_data_to_csv_stringrLrJrMrNrOrPrQ)r.rDrFrRrSdata_csvrUrVs rto_csvr]s#4((H"8,,If9 RRH h   7<<22vuo N NNrXc.t|t|drAt|tjrt |}t |j}d|iSt|tjr't |}d|diSt|trd|vrtd|St|dr9t}| |}d| iSdS)z0Replace a DataFrame by a data model with values.r!r%records)orient.values expected in data dict, but not present.r&N)r'r(r)r*r+rrr!to_dictr,KeyErrorr7r8 to_pylistr.r;r<s r to_valuesrfsDt())0 dBL ) ) ,%d++D%d&<==$ D", ' ' 0!$''$,,i,8899 D$  0 4  KLL L  ' '0 ' ) )$$T**(,,..// 00rctttjfsJt fddDs1t dtdSdS)z-Raise if the data is not a dict or DataFrame.c38K|]}t|VdSN)r().0attrr.s r z"check_data_type..s>>> $d>>>>>>r)r!r&zBExpected dict, DataFrame or a __geo_interface__ attribute, got: {}N)r)r,r*r+any TypeErrorrJr")r.s`rr'r's dT2<0 1 1 #>>>>(N>>>;;  P W WT          rcrtj|Sri)hashlibmd5encode hexdigest)data_strs rrLrLs( ;x(( ) ) 3 3 5 55rct|t|drQt|tjrt |}t |j}tj |St|tjr&t |}| ddSt|tr/d|vrtdtj |ddSt|d rIt}||}tj |St!d ) z5Return a JSON string representation of the input datar!r_)r`double_precisionr%raT) sort_keysr&zBto_json only works with data expressed as a DataFrame or as a dict)r'r(r)r*r+rrr!r?dumpsrWr,rcr7r8rdNotImplementedErrorres rrKrKsADt()) dBL ) ) ,%d++D%d&<==z$ D", ' ' !$''||9r|BBB D$    4  KLL Lz$x.D9999  ' ' ' ) )$$T**z(,,..///! S   rct|t|drtdt|tjr%t |}|dSt|trLd|vrtdtj |ddSt|drt}dd l }dd l m}||}|}||||Std ) z4return a CSV string representation of the input datar!zLto_csv does not work with data that contains the __geo_interface__ attributeF)indexr%z-values expected in data dict, but not presentr&rNzAto_csv only works with data expressed as a DataFrame or as a dict)r'r(rzr)r*r+rr]r,rc from_dictr7pyarrow pyarrow.csvrYr8BufferOutputStream write_csvgetvalue to_pybytesdecode)r.r;papa_csvr< csv_buffers rr[r[s|Dt()) ! 7    D", ' ' !$''{{{''' D$   4  JKK K|%%d8n55<<5<III  ' '  ' ) )$$$$$$$$T****,, :...""$$//1188:::! R   rc^tjdtdtj|g|RS)zi Pipe a value through a sequence of functions Deprecated: use toolz.curried.pipe() instead. zdalt.pipe() is deprecated, and will be removed in a future release. Use toolz.curried.pipe() instead.r stacklevel)warningswarnrrpipe)r.funcss rrrsB  M ,   < %u % % %%rc\tjdtdtj|i|S)zRCurry a callable function Deprecated: use toolz.curried.curry() instead. zfalt.curry() is deprecated, and will be removed in a future release. Use toolz.curried.curry() instead.rr)rrrrcurry)argskwargss rrr s?  M -   =$ )& ) ))rcddl} |dddlm}|S#|j$r}t d|d}~w|j$r}t d|d}~wwxYw)Nrzpyarrow>=11.0.0zdUsage of the DataFrame Interchange Protocol requires the package 'pyarrow', but it is not installed.zThe installed version of 'pyarrow' does not meet the minimum requirement of version 11.0.0. Please update 'pyarrow' to use the DataFrame Interchange Protocol.) pkg_resourcesrequirepyarrow.interchange interchangeDistributionNotFound ImportErrorVersionConflict)rr;errs rr7r7s/000((((((  - r     ( Q   s# A= A AA)r)NN)r>r?r@rA)r>rYr@rA)"r?rOr6rprpandasr*toolzrtypingrcorerr deprecationrplugin_registryr DataTransformerTyper Exceptionrrr0r4rWr]rfr'rLrKr[rr7rrrrsI $$$$$$((((((111111++++++  > > > > >n-@A > > >>     9    @ &&&&(   *  PPPP$   *  O O O O  000,    "666   4   > & & & * * *r