o hc@sdZddlmZddlmZmZddlmZddlm Z er.ddl m Z m Z m Z ddlmZd$d d Z d%d&ddZd'ddZd(ddZd)d d!Zd*d"d#Zd S)+ztData marshalling utilities for ArrowTable protobufs, which are used by CustomComponent for dataframe serialization. ) annotations) TYPE_CHECKINGAny)dataframe_util)pandas_styler_utils) DataFrameIndexSeries) ArrowTableitemrreturncCst|tr t|S|S)z;Convert a tuple to a list. Leave as is if it's not a tuple.) isinstancetuplelist)r r[/var/www/vscode/kcb/lib/python3.10/site-packages/streamlit/components/v1/component_arrow.py_maybe_tuple_to_list srNprotoArrowTableProtodata default_uuid str | NoneNonecCsHt|r t|||t|}t||jt||jt ||dS)a:Marshall data into an ArrowTable proto. Parameters ---------- proto : proto.ArrowTable Output. The protobuf for a Streamlit ArrowTable proto. data : pandas.DataFrame, pandas.Styler, numpy.ndarray, Iterable, dict, or None Something that is or can be converted to a dataframe. N) ris_pandas_stylerrmarshall_stylerconvert_anything_to_pandas_df_marshall_indexindex_marshall_columnscolumns_marshall_data)rrrdfrrrmarshall%s    r"rrcC.ddl}tt|j}||}t||_dS)aHMarshall pandas.DataFrame index into an ArrowTable proto. Parameters ---------- proto : proto.ArrowTable Output. The protobuf for a Streamlit ArrowTable proto. index : pd.Index Index to use for resulting frame. Will default to RangeIndex (0, 1, 2, ..., n) if no index is provided. rN)pandasmaprvaluesrr convert_pandas_df_to_arrow_bytesr)rrpdindex_dfrrrr<  rrr cCr#)a[Marshall pandas.DataFrame columns into an ArrowTable proto. Parameters ---------- proto : proto.ArrowTable Output. The protobuf for a Streamlit ArrowTable proto. columns : Series Column labels to use for resulting frame. Will default to RangeIndex (0, 1, 2, ..., n) if no column labels are provided. rN)r$r%rr&rrr'r)rrr( columns_dfrrrrPr*rr!rcCst||_dS)zMarshall pandas.DataFrame data into an ArrowTable proto. Parameters ---------- proto : proto.ArrowTable Output. The protobuf for a Streamlit ArrowTable proto. df : pandas.DataFrame A dataframe to marshall. N)rr'r)rr!rrrr ds r cCsftdr tdddl}t|j}t|j}t|j}|j| | j | j dS)zConvert ArrowTable proto to pandas.DataFrame. Parameters ---------- proto : proto.ArrowTable Output. pandas.DataFrame z14.0.1zThe installed pyarrow version is not compatible with this component. Please upgrade to 14.0.1 or higher: pip install -U pyarrowrN)rr) ris_pyarrow_version_less_than RuntimeErrorr$ convert_arrow_bytes_to_pandas_dfrrrrto_numpyTtolist)rr(rrrrrrarrow_proto_to_dataframess      r2)r rr r)N)rrrrrrr r)rrrrr r)rrrr r r)rrr!rr r)rrr r)__doc__ __future__rtypingrr streamlitrstreamlit.elements.librr$rrr streamlit.proto.Components_pb2r rrr"rrr r2rrrrs