o e1@sddlZddlZddlZddlZddlZddlmZmZmZm Z m Z m Z m Z ddl ZddlmZddlmZddlmZddlmZmZmZddlmZdd lmZdd lmZdd lmZmZmZe rhddl Z!Gd d d eZ"ee#ej$e"efZ%ede%dZ&e e'ee'e#e e#ffZ(e e'ee#e e#ffZ)GdddeZ*Gdddee*Z+Gddde,Z-ej.dEde&dee/de&fddZ0ej. dFde%dee/dee1deeej$e e'e fdffdd Z2Gd!d"d"eZ3Gd#d$d$eZ4Gd%d&d&eZ5Gd'd(d(eZ6ej. ) * + ,dGde%d-e'd.e'd/e'd0e'de5f d1d2Z7ej. ) 3 + ,dHdee#ej$efd-e'd.e'd/e'd0e'de6f d4d5Z8ej.de%de)fd6d7Z9de%ddfd8d9Z:d:e'de'fd;d<Z;de%de'fd=d>ZdCdDZ.dS)IN)UnionMutableMappingOptionalDictSequence TYPE_CHECKINGList)curried)TypeVar)import_pyarrow_interchange)sanitize_dataframesanitize_arrow_table_DataFrameLike)sanitize_geo_interface)AltairDeprecationWarning)PluginRegistry)Protocol TypedDictLiteralc@seZdZUeed<dS)_SupportsGeoInterface__geo_interface__N)__name__ __module__ __qualname__r__annotations__rrAD:\Projects\ConvertPro\env\Lib\site-packages\altair/utils/data.pyrs r _TDataType)boundc@seZdZdedefddZdS)DataTransformerTypedatareturncKsdSNr)selfr!kwargsrrr__call__1szDataTransformerType.__call__N)rrr _DataType_VegaLiteDataDictr&rrrrr 0sr c@s>eZdZddiZedefddZejdeddfddZdS) DataTransformerRegistryconsolidate_datasetsTr"cCs |jdSNr*_global_settings)r$rrrr*8s z,DataTransformerRegistry.consolidate_datasetsvalueNcCs||jd<dSr+r,)r$r.rrrr*<s)rrrr-propertyboolr*setterrrrrr)5s r)c@seZdZdZdS) MaxRowsErrorz+Raised when a data model has too many rows.N)rrr__doc__rrrrr2Bsr2r!max_rowsr"cst|fdd}t|dr |jddkr|jd}n8|j}n4t|tjr)|}n+t|tr9d|vr7|d}n|St|drTt}||}d urR|j krR||Sd urat |kra||S) zzRaise MaxRowsError if the data model has more than max_rows. If max_rows is None, then do not perform any check. cstdd)NzHThe number of rows in your dataset is greater than the maximum allowed (aG). Try enabling the VegaFusion data transformer which raises this limit by pre-evaluating data transformations in Python. >> import altair as alt >> alt.data_transformers.enable("vegafusion") Or, see https://altair-viz.github.io/user_guide/large_datasets.html for additional information on how to plot large datasets.)r2rr5rrraise_max_rows_errorPs z(limit_rows..raise_max_rows_errorrtypeZFeatureCollectionfeaturesvalues __dataframe__N) check_data_typehasattrr isinstancepd DataFramedictr from_dataframeZnum_rowslen)r!r5r7r:pipa_tablerr6r limit_rowsHs*        rFnfraczpyarrow.lib.TablecCst|t|tjr|j||dSt|trr?r@samplerA ValueErrorintrCrandomr=r rBrangeZtake)r!rGrHr:rDrEindicesrrrrI{s6      rIc@eZdZUeded<dS)_JsonFormatDictjsonr8NrrrrrrrrrrP rPc@rO)_CsvFormatDictcsvr8NrRrrrrrTrSrTc@eZdZUeed<eed<dS)_ToJsonReturnUrlDicturlformatN)rrrstrrrPrrrrrW  rWc@rV)_ToCsvReturnUrlDictrXrYN)rrrrZrrTrrrrr\r[r\ altair-datarQ{prefix}-{hash}.{extension}prefix extensionfilenameurlpathcClt|}t|}|j|||d}t|d }||Wdn1s%wYtj||ddidS)zQ Write the data model to a .json file and return a url based data model. r`hashrawNr8rQrXrY)_data_to_json_string_compute_data_hashrYopenwriteospathjoin)r!r`rarbrcZ data_json data_hashfrrrto_jsons   rrrUcCrd)zFWrite the data model to a .csv file and return a url based data model.rergNr8rUrh)_data_to_csv_stringrjrYrkrlrmrnro)r!r`rarbrcZdata_csvrprqrrrto_csvs   rtcCst|t|drt|tjrt|}t|j}d|iSt|tjr.t|}d|jddiSt|t r=d|vr;t d|St|drRt }t | |}d|iStdt|)z0Replace a DataFrame by a data model with values.rr:records)orient.values expected in data dict, but not present.r;zUnrecognized data type: {})r<r=r>r?r@r rrto_dictrAKeyErrorr rrB to_pylistrJrYr8)r!Zdata_sanitizedrDrErrr to_valuess$       r{cs@tttjfstfdddDstdtdSdS)Nc3s|]}t|VqdSr#)r=).0attrr!rr s  z"check_data_type..)rr;zBExpected dict, DataFrame or a __geo_interface__ attribute, got: {})r>rAr?r@any TypeErrorrYr8r~rr~rr<sr<data_strcCst|Sr#)hashlibmd5encode hexdigest)rrrrrjsrjcCst|t|drt|tjrt|}t|j}t |St|tjr.t|}|j dddSt|t rDd|vr;t dtj |dddSt|d rXt }||}t |Std ) z5Return a JSON string representation of the input datarru)rvZdouble_precisionr:rwT) sort_keysr;zBto_json only works with data expressed as a DataFrame or as a dict)r<r=r>r?r@r rrrQdumpsrrrAryr rBrzNotImplementedError)r!rDrErrrris(        ricCst|t|dr tdt|tjrt|}|jddSt|tr6d|vr*t dtj |djddSt|dr_t }dd l }dd l m}||}|}||||Std ) z4return a CSV string representation of the input datarzLto_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=rr>r?r@r rtrAry from_dictr pyarrowZ pyarrow.csvrUrBZBufferOutputStreamZ write_csvgetvalueZ to_pybytesdecode)r!rDpaZpa_csvrEZ csv_bufferrrrrs#s.        rscGs"tjdtddtj|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)warningswarnrr pipe)r!funcsrrrrBs rcOs tjdtddtj|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.r r)rrrr curry)argsr%rrrrQs r)r4)NN)r]rQr^r_)r]rUr^r_)?rQrmrLrrtypingrrrrrrrZpandasr?Ztoolzr r Z _importersr corer rrr deprecationrZplugin_registryrrrrZ pyarrow.librrrAr@r'rrZr(Z_ToValuesReturnTyper r) Exceptionr2rrKrFfloatrIrPrTrWr\rrrtr{r<rjrirsrrrrrs$        2 '