a  dn @sdZddlmZmZddlmZddlZddlZddlZddl Z ddl Z ddl Z ddl m Z mZmZmZmZmZmZmZmZmZddlmZddlZddlZddlZddlmZddl m!Z!dd l"m#Z#m$Z$m%Z&e j'd krdd l m(Z(n dd l)m(Z(dd l m*Z*m+Z+m,Z,e,rdd l-m#Z.edZ/e(dZ0Gddde+Z1ddddddZ2dde23DZ4gdZ5gdZ6gdZ7e*dZ8e9ee8ee8e:ffddd Z;eeee?d'd(d)Z@ej%ej%d*d+d,ZAd-d.ZBdDeeeCefeCfeeej%e1fe?e?e?e?eeCefd1d2d3ZDee#d4fee8ee8e:ffd5d6d7ZEe e0efd8d9d:ZFdEeee?ed;dd?d@ZHeeedAdBdCZIdS)Gz Utility routines )MappingMutableMapping)deepcopyN) CallableTypeVarAnyUnionDictOptionalTupleSequenceTypecast) ModuleType) infer_dtype) SchemaBase)Column DtypeKind DataFrame) ) ParamSpec)LiteralProtocol TYPE_CHECKING)r_V_Pc@seZdZedddZdS)_DataFrameLike)returncOsdSN)selfargskwargsr r Z/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/altair/utils/core.py __dataframe__1sz_DataFrameLike.__dataframe__N)__name__ __module__ __qualname__ DfiDataFramer%r r r r$r0srONQTG)ordinalnominal quantitativetemporalZgeojsoncCsi|]\}}||qSr r ).0kvr r r$ =r6)ZargmaxZargminZaveragecountZdistinctmaxmeanZmedianminmissingproductZq1Zq3Zci0Zci1stderrstdevZstdevpsumZvalidvaluesZvarianceZ variancep) Z row_numberZrankZ dense_rankZ percent_rankZ cume_distZntileZlagZleadZ first_value last_valueZ nth_value)PyearZquartermonthweekdayZ dayofyeardatehoursminutessecondsZ millisecondsZ yearquarterZyearquartermonthZ yearmonthZ yearmonthdateZyearmonthdatehoursZyearmonthdatehoursminutesZ yearmonthdatehoursminutessecondsZyearweekZ yearweekdayZyearweekdayhoursZyearweekdayhoursminutesZyearweekdayhoursminutessecondsZ yeardayofyearZ quartermonthZ monthdateZmonthdatehoursZmonthdatehoursminutesZmonthdatehoursminutessecondsweekdayZ weeksdayhoursZweekdayhoursminutesZweekdayhoursminutessecondsZdayhoursZdayhoursminutesZdayhoursminutessecondsZ hoursminutesZhoursminutessecondsZminutessecondsZsecondsmillisecondsZutcyearZ utcquarterZutcmonthZutcweekZutcdayZ utcdayofyearZutcdateZutchoursZ utcminutesZ utcsecondsZutcmillisecondsZutcyearquarterZutcyearquartermonthZ utcyearmonthZutcyearmonthdateZutcyearmonthdatehoursZutcyearmonthdatehoursminutesZ#utcyearmonthdatehoursminutessecondsZ utcyearweekZutcyearweekdayZutcyearweekdayhoursZutcyearweekdayhoursminutesZ!utcyearweekdayhoursminutessecondsZutcyeardayofyearZutcquartermonthZ utcmonthdateZutcmonthdatehoursZutcmonthdatehoursminutesZutcmonthdatehoursminutessecondsZ utcweekdayZutcweeksdayhoursZutcweekdayhoursminutesZutcweekdayhoursminutessecondsZ utcdayhoursZutcdayhoursminutesZutcdayhoursminutessecondsZutchoursminutesZutchoursminutessecondsZutcminutessecondsZutcsecondsmilliseconds)r/r0r1r2)datarcCsvt|dd}|dvrdS|dkrBt|drB|jjrBd|jjfS|dvrNd S|d vrZd Stjd |d dd SdS)z From an array-like input, infer the correct vega typecode ('ordinal', 'nominal', 'quantitative', or 'temporal') Parameters ---------- data: object F)Zskipna)floatingzmixed-integer-floatintegerz mixed-integercomplexr1 categoricalcatr/)stringbytesrPbooleanmixedunicoder0)datetime datetime64 timedeltaZ timedelta64rGtimeZperiodr2zJI don't know how to infer vegalite type from '{}'. Defaulting to nominal. stacklevelN) rhasattrrQZordered categoriestolistwarningswarnformat)rLtypr r r$infer_vegalite_types  re)featrc sLfdddD}zd|d}WnttfyF|}Yn0|S)zc Merge properties with geometry * Overwrites 'type' and 'geometry' entries if existing csi|]}||qSr r )r3r4rfr r$r6r7z$merge_props_geom..typegeometryZ properties)updateAttributeErrorKeyError)rfZgeomZ props_geomr rgr$merge_props_geoms  rn)georcCst|}|D],}tt||jdr||||<qtt |}|ddkr|d}t |dkrt |D]\}}t |||<qvn |ddkrt |}n d|d}|S)zSantize a geo_interface to prepare it for serialization. * Make a copy * Convert type array or _Array to list * Convert tuples to lists (using json.loads/dumps) * Merge properties with geometry )Z_ArrayarrayriZFeatureCollectionfeaturesrZFeaturerh) rkeysstrrir& startswithr`jsonloadsdumpslen enumeratern)rokeyZgeo_dctidxrfr r r$sanitize_geo_interfaces       r|)dtypesubtyperc Cs,zt||WSttfy&YdS0dS)NF)npZ issubdtypeNotImplementedError TypeError)r}r~r r r$numpy_is_subtype!sr)dfrcCs~|}t|jtjr$|jt|_|jD]}t|ts*td|q*t|j tj r^tdt|jtj rttddd}|j D]}t t|d}|d}t|}|dkr||t}||d||<q|d kr||t}||d||<q|d kr"||t||<q|d krP||t}||d||<q|d sh|d r||dddd||<q|drtdj||dq|drqq|dvr||t}||d||<qt|tjr||t||<qt|tjrJ||}|t|B}|t|d||<q|tkr||j|dd}||d||<q|S)aSanitize a DataFrame to prepare it for serialization. * Make a copy * Convert RangeIndex columns to strings * Raise ValueError if column names are not strings * Raise ValueError if it has a hierarchical index. * Convert categoricals to strings. * Convert np.bool_ dtypes to Python bool objects * Convert np.int dtypes to Python int objects * Convert floats to objects and replace NaNs/infs with None. * Convert DateTime dtypes into appropriate string representations * Convert Nullable integers to objects and replace NaN with None * Convert Nullable boolean to objects and replace NaN with None * convert dedicated string column to objects and replace NaN with None * Raise a ValueError for TimeDelta dtypes zKDataframe contains invalid column name: {0!r}. Column names must be stringsz"Hierarchical indices not supportedcSst|tjr|S|SdSr) isinstancerZndarrayr`)valr r r$to_list_if_arrayJs z,sanitize_dataframe..to_list_if_arrayrr[categoryNrRboolrTrW timestampcSs|Sr) isoformat)xr r r$rr7z$sanitize_dataframe..ZNaTrYField "{col_name}" has type "{dtype}" which is not supported by Altair. Please convert to either a timestamp or a numerical value.col_namer}rj> ZFloat64ZInt16ZUInt8ZFloat32ZInt8ZInt64ZUInt16ZInt32ZUInt32ZUInt64F)Z convert_dtype)copyrcolumnspdZ RangeIndexZastypers ValueErrorrcindexZ MultiIndexZdtypesitemsrobjectwhereZnotnullrtapplyreplacerrrNrMZisnullisinf)rrrZ dtype_itemr}Z dtype_namecolZ bad_valuesr r r$sanitize_dataframe(sl        rcCsddl}ddlm}g}|j}|jD]d}||}||j}t|dr\| | |q$t|dr~t dj ||dq$| |q$|j j||jdS)z+Sanitize arrow table for JSON serializationrNrdurationrr)names)ZpyarrowZpyarrow.computeZcomputeschemarfieldrirsrtappendstrftimerrcZTableZ from_arrays)Zpa_tablepaZpcZarraysrnamerpr}r r r$sanitize_arrow_tables"    rTF) shorthandrLparse_aggregatesparse_window_opsparse_timeunits parse_typesrc s>ddlm}siStttt}ddd|ddddtd dttd dt d g}|r| d g| d g|r| dg| dg|r| dg| dg|rtt j dd|D}fdd|D} t tr} ntfdd| D} d| vr>> data = pd.DataFrame({'foo': ['A', 'B', 'A', 'B'], ... 'bar': [1, 2, 3, 4]}) >>> parse_shorthand('name') == {'field': 'name'} True >>> parse_shorthand('name:Q') == {'field': 'name', 'type': 'quantitative'} True >>> parse_shorthand('average(col)') == {'aggregate': 'average', 'field': 'col'} True >>> parse_shorthand('foo:O') == {'field': 'foo', 'type': 'ordinal'} True >>> parse_shorthand('min(foo):Q') == {'aggregate': 'min', 'field': 'foo', 'type': 'quantitative'} True >>> parse_shorthand('month(col)') == {'field': 'col', 'timeUnit': 'month', 'type': 'temporal'} True >>> parse_shorthand('year(col):O') == {'field': 'col', 'timeUnit': 'year', 'type': 'ordinal'} True >>> parse_shorthand('foo', data) == {'field': 'foo', 'type': 'nominal'} True >>> parse_shorthand('bar', data) == {'field': 'bar', 'type': 'quantitative'} True >>> parse_shorthand('bar:O', data) == {'field': 'bar', 'type': 'ordinal'} True >>> parse_shorthand('sum(bar)', data) == {'aggregate': 'sum', 'field': 'bar', 'type': 'quantitative'} True >>> parse_shorthand('count()', data) == {'aggregate': 'count', 'type': 'quantitative'} True r)pyarrow_availablez (?P.*)z (?P{})|z(?Pcount)z (?Pcount)z(?P{})z (?P{})z(?P{}))rriZ agg_countZop_count aggregateZ window_optimeUnitz{agg_count}\(\)z{aggregate}\({field}\)z{op_count}\(\)z{window_op}\({field}\)z{timeUnit}\({field}\)z{field}css|]}|d|fVqdS)z:{type}Nr r3pr r r$ )r7z"parse_shorthand..c3s0|](}td|jfidtjVqdS)z\Az\ZN)recompilercDOTALLr)unitsr r$r+sc3s*|]"}|dur|VqdSr)match groupdict)r3exp)rr r$r3srirr8r1rr2Nr%r\rr[sort:z"{}" z0is not one of the valid encoding data types: {}.z, zi For more details, see https://altair-viz.github.io/user_guide/encodings/index.html#encoding-data-types. z>If you are trying to use a column name that contains a colon, zPprefix it with a backslash; for example "column\:name" instead of "column:name".)!Zaltair.utils._importersrlist TYPECODE_MAPINV_TYPECODE_MAPrcjoin AGGREGATESWINDOW_AGGREGATES TIMEUNITSextend itertoolschainrdictnextgetr^r%rZ column_namesZget_column_by_name"infer_vegalite_type_for_dfi_columnrrrretuplerrfindrsplitrA)rrLrrrrrZvalid_typecodespatternsZregexpsattrsZdfiZunescaped_fieldcolumnr )rrr$parse_shorthandsR               "   r PandasColumn)rrc Csddlm}z|jd}WnRtyl}z:d|jdvsFd|jdvrTWYd}~dS|WYd}~n d}~00|tjkr|jdr|jddur|jd}||}d|fS|tj tjtj fvrd S|tj tj tj fvrd S|tjkrdStd |dS) Nr)column_to_arrayrXrr2Z is_orderedr_r/r0r1zUnexpected DtypeKind: )Z"pyarrow.interchange.from_dataframerr}rr"rZ CATEGORICALZdescribe_categoricalZ to_pylistSTRINGZBOOLINTZUINTFLOATZDATETIMEr)rrkindeZcategories_columnZcategories_arrayr r r$rys.     rObjcs(tdtftttfdfdd }|S)zEApply call signature and documentation of Obj to the decorated method.)frcs|j|_|_jrxj}djd|d<|jrP|jd|dd}n d|}z ||_WntyvYn0|S)NzRefer to :class:``r r[)__init__ __wrapped__Z_uses_signature__doc__ splitlinesr&rrl)rZdoclinesdocrr r$decorates    zuse_signature..decorate)rrr)rrr rr$ use_signatures$r)originalrkrrcCsd|r t|}|D]J\}}t|trV||i}t|trLt||||<q^|||<q|||<q|S)aUpdate nested dictionaries Parameters ---------- original : MutableMapping the original (nested) dictionary, which will be updated in-place update : Mapping the nested dictionary of updates copy : bool, default False if True, then copy the original dictionary rather than modifying it Returns ------- original : MutableMapping a reference to the (modified) original dict Examples -------- >>> original = {'x': {'b': 2, 'c': 4}} >>> update = {'x': {'b': 5, 'd': 6}, 'y': 40} >>> update_nested(original, update) # doctest: +SKIP {'x': {'b': 5, 'c': 4, 'd': 6}, 'y': 40} >>> original # doctest: +SKIP {'x': {'b': 5, 'c': 4, 'd': 6}, 'y': 40} )rrrrrr update_nested)rrkrrzrZorig_valr r r$rs     r) in_ipythoncCsFt}|r ddlm}|}nd}|dur8||n tj|dS)Nr) get_ipython)sysexc_infoZIPython.core.getipythonrZ showtraceback tracebackprint_exception)rrripr r r$display_tracebacks  r)r"r#channelsc s"fddtD}dd|D}dd|D}i|D]D\}}|i}|jdrdd}n|jdrvd }nd }|||<q>|D]t} t| ttfrt| d krt | d } nt | } | | d } | d urt d | | |vrt d | | || <qfddfdd|DS)aInfer typed keyword arguments for args and kwargs Parameters ---------- args : Sequence Sequence of function args kwargs : MutableMapping Dict of function kwargs channels : ModuleType The module containing all altair encoding channel classes. Returns ------- kwargs : dict All args and kwargs in a single dict, with keys and types based on the channels mapping. c3s|]}t|VqdSr)getattr)r3r)rr r$r r7z'infer_encoding_types..css&|]}t|trt|tr|VqdSr)rri issubclassrr3cr r r$r scSsi|] }||jqSr )Z_encoding_namerr r r$r6sz(infer_encoding_types..ZDatumZdatumValuevaluerrNzpositional of type {}zencoding {} specified twice.cst|tr|St|tr d|i}t|ttfrBfdd|DSvrbtjddd|S}d|vrz|dn|d}z|j|d d WSt j y|YS0dS) Nrcsg|]}|qSr r )r3Zsubobj)_wrap_in_channel_classencodingr r$ 3r7zHinfer_encoding_types.._wrap_in_channel_class..z"Unrecognized encoding channel '{}'r[r\rrF)validate) rrrsrrrarbrc from_dict jsonschemaZValidationError)objrclassescls)rname_to_channel)rr$r+s"   z4infer_encoding_types.._wrap_in_channel_classcsi|]\}}|||qSr r )r3rr)rr r$r6Fs)dirr setdefaultr&endswithrrrrxrirrrcr) r"r#rZ channel_objsZchannel_to_nameZchanrZchansrzargtype_rr )rrrr$infer_encoding_typess<       r)NTFTT)F)T)Jrcollections.abcrrrrrurrrrratypingrrrrr r r r r rtypesrrZpandasrnumpyrZpandas.api.typesrZaltair.utils.schemapirZaltair.utils._dfi_typesrrrr) version_inforZtyping_extensionsrrrZ*pandas.core.interchange.dataframe_protocolrrrrrrrrrrZ_InferredVegaLiteTyperrrerrnr|rrrrrsrrrrrrr r r r$s 0      T ,!w C  %  *