mxi$ddlmZddlZddlmZddlmZmZmZddl m Z ddl m Z m Z ddlmZddlmZdd lmZdd lmZmZerdd lmZeeZ d dd Zdd ZddZddZeGddZ y)) annotationsN) dataclass) TYPE_CHECKINGAnycast) ARROW_REF_KEY)convert_anything_to_arrow_bytesis_dataframe_like) get_logger) BidiComponent) MixedData)AttributeDictionarycalc_md5)BidiComponentStatec|i}i}|jD]=\}}t|r( t|}t|}|||<t|i||<9|||<?|S#t $r&}t jd|||||<Yd}~ld}~wwxYw)aExtract dataframe-like objects from a dictionary and replace them with placeholders. This function traverses the first level of a dictionary, detects any dataframe-like objects, stores their Arrow bytes representation in the `arrow_blobs` dictionary, and replaces them with JSON-serializable placeholder objects. Parameters ---------- data The dictionary to process. Only the first level is checked for dataframe-like objects. arrow_blobs The dictionary to store the extracted Arrow bytes in, keyed by a unique reference ID. Returns ------- dict[str, Any] A new dictionary with dataframe-like objects replaced by placeholders. NzAArrow serialization failed for key %r, keeping original value: %s)itemsr r rr Exception_LOGGERdebug)data arrow_blobsprocessed_datakeyvalue arrow_bytesref_ides k/home/htdocs/ttos/venv/lib/python3.12/site-packages/streamlit/components/v2/bidi_component/serialization.py_extract_dataframes_from_dictr"s2 Njjl( U U # ,=eD "+.&1 F#'4f&=s##(N3 -(0  , W ',s# ,s&A B"BBc"i}t|tr t||}n|}|rmt} t j ||_|jD]\}}||j|_ |jj|y t j ||_y#t $r&t j t||_YwxYw#t $r&t j t||_YywxYw)a4Serialize mixed data with automatic dataframe detection into a protobuf message. This function detects dataframe-like objects in the first level of a dictionary, extracts them into separate Arrow blobs, and populates a `MixedDataProto` protobuf message for efficient serialization. Parameters ---------- data The data structure to serialize. If it is a dictionary, its first level will be scanned for dataframe-like objects. bidi_component_proto The protobuf message to populate with the serialized data. N) isinstancedictrMixedDataProtojsondumps TypeErrorstrrrrmixedCopyFrom)rbidi_component_protorr mixed_protorrs rserialize_mixed_datar,[s %'K$6t[I$&  ?#zz.9K $/#4#4#6 ? FK3>K # #F + 0 ? ""++K8 H(, >(B % ? $zz#n*=>K  ? H)- 3~3F(G % Hs#B-C-,CC,D Dcj|y tj|S#tj$r|cYSwxYw)a;Deserialize a JSON string, returning the string itself if it's not valid JSON. Parameters ---------- s The string to deserialize. Returns ------- Any The deserialized JSON object, or the original string if parsing fails. Returns `None` if the input is `None`. N)r$loadsJSONDecodeError)ss rhandle_deserializer1s9 yzz!}   s 22cHt|}|yt|tr|S|gS)aDeserialize trigger aggregator payloads as a list. For bidirectional components, the frontend always sends a JSON array of payload objects. This deserializer normalizes older or singular payloads into a list while preserving ``None`` for cleared values. Parameters ---------- s The JSON string to deserialize, hopefully representing a list of payloads. Returns ------- list[Any] or None A list of payloads, or `None` if the input was `None`. N)r1r!list)r0rs rdeserialize_trigger_listr4s,$ q !E }% 7Nc0eZdZUdZdZded<ddZddZy) BidiComponentSerdeaSerialization and deserialization logic for a bidirectional component. This class handles the conversion of component state between the frontend (JSON strings) and the backend (Python objects). The canonical shape is a flat mapping of state keys to values. Parameters ---------- default A dictionary of default values to be applied to the state when deserializing, if the corresponding keys are not already present. Nzdict[str, Any] | Nonedefaultct|tr|}n=t|tr+ tj|}t|tr|ni}ni}|j,|jjD]\}}||vs |||<tdt|}|S#tj t f$r$}tjd||i}Yd}~d}~wwxYw)a_Deserialize the component's state from a frontend value. Parameters ---------- ui_value The value received from the frontend, which can be a JSON string, a dictionary, or `None`. Returns ------- BidiComponentState The deserialized state as a flat mapping. z7Failed to deserialize component state from frontend: %s)exc_infoNr) r!r"r'r$r.r/r&rwarningr8rrr)selfui_valuedeserialized_valueparsedr default_key default_valuestates r deserializezBidiComponentSerde.deserializes* h %!)  # & (H-/9&$/GVR""$  << #.2ll.@.@.B D* ]&886C&{3 D%) "56H"I%  '(()4 (M  &("  (s)B C9CCc,tj|S)aSerialize the component's state into a JSON string for the frontend. Parameters ---------- value The component state to serialize. Returns ------- str A JSON string representation of the value. )r$r%)r<rs r serializezBidiComponentSerde.serializeszz%  r5)r=zstr | dict[str, Any] | Nonereturnr)rrrFr')__name__ __module__ __qualname____doc__r8__annotations__rCrEr5rr7r7s &*G ").`!r5r7)N)rdict[str, Any]rzdict[str, bytes] | NonerFrM)rrr*BidiComponentProtorFNone)r0 str | NonerFr)r0rPrFzlist[Any] | None)! __future__rr$ dataclassesrtypingrrr0streamlit.components.v2.bidi_component.constantsrstreamlit.dataframe_utilr r streamlit.loggerr !streamlit.proto.BidiComponent_pb2r rNr r#streamlit.utilrr,streamlit.components.v2.bidi_component.staterrGrrr,r1r4r7rLr5rrZs# !++JW'QI8O X BF6 6'>66r0Hf.4 P!P! P!r5