mxidZddlmZddlmZmZddlmZddlm Z erddl m Z m Z ddl mZddZ d dd Zdd Zdd Zdd ZddZy )ztData marshalling utilities for ArrowTable protobufs, which are used by CustomComponent for dataframe serialization. ) annotations) TYPE_CHECKINGAny)dataframe_util)pandas_styler_utils) DataFrameIndex) ArrowTablec<t|tr t|S|S)z;Convert a tuple to a list. Leave as is if it's not a tuple.) isinstancetuplelist)items ^/home/htdocs/ttos/venv/lib/python3.12/site-packages/streamlit/components/v1/component_arrow.py_maybe_tuple_to_listr s#D%04::d:Nctj|rtj|||tj|}t ||j t||jt||y)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)protodata default_uuiddfs rmarshallr %sZ&&t,++E4F  5 5d ;BE288$eRZZ(5"rcddl}ttt|j}|j |}t j||_y)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) pandasrmaprvaluesrr convert_pandas_df_to_arrow_bytesr)rrpd index_valuesindex_dfs rrr<s=0%,,?@L||L)H AA(KEKrcddl}ttt|j}|j |}t j||_y)aZMarshall pandas.DataFrame columns into an ArrowTable proto. Parameters ---------- proto : proto.ArrowTable Output. The protobuf for a Streamlit ArrowTable proto. columns : Index Column labels to use for resulting frame. Will default to RangeIndex (0, 1, 2, ..., n) if no column labels are provided. rN) r"rr#rr$rrr%r)rrr& column_values columns_dfs rrrPs=17>>BCMm,J"CCJOEMrc8tj||_y)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)rrs rrrds @@DEJrctjdr tdddl}tj|j }tj|j }tj|j}|j|j|jjj|jjjS)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&rrrs rarrow_proto_to_dataframer4ss228< I    : :5:: FD  ; ;EKK HE==emmLG << nn  '')  "$$++-  r)rrreturnr)N)rArrowTableProtorrrz str | Noner5None)rr6r Index[Any]r5r7)rr6rr8r5r7)rr6rrr5r7)rr6r5r)__doc__ __future__rtypingrr streamlitrstreamlit.elements.librr"rr streamlit.proto.Components_pb2r r6rr rrrr4rrr@si#%$6'L; CG "%5? .L(P( Er