o h@sddlmZddlmZddlmZmZmZddlm Z ddl m Z er6ddl m Z ddlmZddlmZd.ddZd.ddZd/ddZd0ddZedeeefdZd1ddZ d2d3d%d&Zd4d)d*Zd5d+d,Zd-S)6) annotations)Mapping) TYPE_CHECKINGAnyTypeVar)dataframe_util)StreamlitAPIException) DataFrame)Styler)Arrowproto ArrowProtostylerr default_uuidstrreturnNonecCsddl}|j}|jt|jjjjkr&td|jd|jjjjd|jdt |||| | dd}t ||t |||t|||dS)aoMarshall pandas.Styler into an Arrow proto. Parameters ---------- proto : proto.Arrow Output. The protobuf for Streamlit Arrow proto. styler : pandas.Styler Helps style a DataFrame or Series according to the data with HTML and CSS. default_uuid : str If pandas.Styler uuid is not provided, this value will be used. rNzThe dataframe has `zc` cells, but the maximum number of cells allowed to be rendered by Pandas Styler is configured to `z`. To allow more cells to be styled, you can change the `"styler.render.max_elements"` config. For example: `pd.set_option("styler.render.max_elements", z)`F)pandasdatasizeintoptionsrrender max_elementsr_marshall_uuid_compute _translate_marshall_caption_marshall_styles_marshall_display_values)r rrpdstyler_data_df pandas_stylesr#^/var/www/vscode/kcb/lib/python3.10/site-packages/streamlit/elements/lib/pandas_styler_utils.pymarshall_stylers       r%cCs&|jdur ||t|j|j_dS)atMarshall pandas.Styler uuid into an Arrow proto. Parameters ---------- proto : proto.Arrow Output. The protobuf for Streamlit Arrow proto. styler : pandas.Styler Helps style a DataFrame or Series according to the data with HTML and CSS. default_uuid : str If pandas.Styler uuid is not provided, this value will be used. N)uuidset_uuidrr)r rrr#r#r$rGs  rcCs|jdur |j|j_dSdS)aMarshall pandas.Styler caption into an Arrow proto. Parameters ---------- proto : proto.Arrow Output. The protobuf for Streamlit Arrow proto. styler : pandas.Styler Helps style a DataFrame or Series according to the data with HTML and CSS. N)captionr)r rr#r#r$r\s rstylesMapping[str, Any]cCsg}d|vr!|d}t|}|D]}td||jdd}||qd|vr@|d}t|}|D]}td||jdd}||q/t|dkrOd||j_d Sd S) aRMarshall pandas.Styler styles into an Arrow proto. Parameters ---------- proto : proto.Arrow Output. The protobuf for Streamlit Arrow proto. styler : pandas.Styler Helps style a DataFrame or Series according to the data with HTML and CSS. styles : dict pandas.Styler translated styles. table_styles ) separator cellstyle cell_style_r N)_trim_pandas_styles_pandas_style_to_cssr&appendlenjoinrr))r rr) css_rulesr+styleruler.r#r#r$rls$    rM)boundlist[M]cCsdd|DS)zFilter out empty styles. Every cell will have a class, but the list of props may just be [['', '']]. Parameters ---------- styles : list pandas.Styler translated styles. cSs&g|]}tdd|dDr|qS)css|]}t|VqdS)Nany).0yr#r#r$ sz1_trim_pandas_styles...propsr=)r?xr#r#r$ s&z'_trim_pandas_styles..r#)r)r#r#r$r2s r2r0 style_typer8r&r-cCsg}|dD]\}}t|dt|}||qd|}|dkr-|dg} n|d} g} | D] } | ||| q5d| } d|} | d | d }|S) aTConvert pandas.Styler translated style to CSS. Parameters ---------- style_type : str Either "table_styles" or "cell_style". style : dict pandas.Styler translated style. uuid : str pandas.Styler uuid. separator : str A string separator used between table and cell selectors. rBz: z#T_r+selector selectorsz, z; z { z })rstripr4r6)rEr8r&r- declarations css_property css_value declarationtable_selectorcell_selectorsrG cell_selectorrFdeclaration_blockrule_setr#r#r$r3s     r3dfr cCst||}t||j_dS)a0Marshall pandas.Styler display values into an Arrow proto. Parameters ---------- proto : proto.Arrow Output. The protobuf for Streamlit Arrow proto. df : pandas.DataFrame A dataframe with original values. styles : dict pandas.Styler translated styles. N)_use_display_valuesr convert_pandas_df_to_arrow_bytesrdisplay_values)r rRr)new_dfr#r#r$rs  rc Csddl}|t}|d}d|vrA|d}|D](}|D]#}d|vr?||d}r?tt|\} } t|d|j| | f<qq|S)zCreate a new pandas.DataFrame where display values are used instead of original ones. Parameters ---------- df : pandas.DataFrame A dataframe with original values. styles : dict pandas.Styler translated styles. rNzrow(\d+)_col(\d+)bodyid display_value) reastypercompilematchmaprgroupsiloc) rRr)rZrVcell_selector_regexrowsrowcellr]rcr#r#r$rSs  rSN)r r rr rrrr)r r rr rr)r r rr r)r*rr)r)r<rr<)r0) rErr8r*r&rr-rrr)r r rRr r)r*rr)rRr r)r*rr ) __future__rcollections.abcrtypingrrr streamlitrstreamlit.errorsrrr pandas.io.formats.styler streamlit.proto.Arrow_pb2r r r%rrrrr:r2r3rrSr#r#r#r$s&         )  )  7