L iddlmZddlmZddlmZddlmZmZm Z m Z m Z m Z m Z mZmZmZmZerddlmZdddddd d dZ d d Zy) ) annotations) TYPE_CHECKING)StreamlitAPIException) RegisterWidgetResultTValueFieldName WidgetArgsWidgetCallbackWidgetDeserializer WidgetKwargsWidgetMetadataWidgetSerializerWidgetValuePresenteruser_key_from_element_id)ScriptRunContextN) callbackson_change_handlerargskwargs presenterc ~| | tdt|||||||||r |jnd|  } t| |S)a Register a widget with Streamlit, and return its current value. NOTE: This function should be called after the proto has been filled. Parameters ---------- element_id : str The id of the element. Must be unique. deserializer : WidgetDeserializer[T] Called to convert a widget's protobuf value to the value returned by its st. function. serializer : WidgetSerializer[T] Called to convert a widget's value to its protobuf representation. ctx : ScriptRunContext or None Used to ensure uniqueness of widget IDs, and to look up widget values. callbacks : dict[str, WidgetCallback] or None A dictionary of callbacks for multi-callback support. on_change_handler : WidgetCallback or None An optional callback invoked when the widget's value changes. args : WidgetArgs or None Positional arguments to pass to the `on_change_handler` or `callbacks`. kwargs : WidgetKwargs or None Keyword arguments to pass to the `on_change_handler` or `callbacks`. value_type: ValueFieldName The value_type the widget is going to use. We use this information to start with a best-effort guess for the value_type of each widget. Once we actually receive a proto for a widget from the frontend, the guess is updated to be the correct type. Unfortunately, we're not able to always rely on the proto as the type may be needed earlier. Thankfully, in these cases (when value_type == "trigger_value"), the static table here being slightly inaccurate should never pose a problem. presenter : WidgetValuePresenter or None An optional hook that allows a widget to customize how its value should be presented. Returns ------- register_widget_result : RegisterWidgetResult[T] Provides information on which value to return to the widget caller, and whether the UI needs updating. - Unhappy path: - Our ScriptRunContext doesn't exist (meaning that we're running as a "bare script" outside streamlit). - We are disconnected from the SessionState instance. In both cases we'll return a fallback RegisterWidgetResult[T]. - Happy path: - The widget has already been registered on a previous run but the user hasn't interacted with it on the client. The widget will have the default value it was first created with. We then return a RegisterWidgetResult[T], containing this value. - The widget has already been registered and the user *has* interacted with it. The widget will have that most recent user-specified value. We then return a RegisterWidgetResult[T], containing this value. For both paths a widget return value is provided, allowing the widgets to be used in a non-streamlit setting. Nz# J  "/2C++ H )3 77c| tj|jS|j}t |}|j j ||S)a$Register a widget and return its value, using an already constructed `WidgetMetadata`. This is split out from `register_widget` to allow caching code to replay widgets by saving and reusing the completed metadata. See `register_widget` for details on what this returns. )r )rfailurer idr session_stater%)r#r" widget_iduser_keys r$rrsN {$++9N9NOO I' 2H    , ,Xx @@r&)rstrr zWidgetDeserializer[T]r!zWidgetSerializer[T]r"ScriptRunContext | Nonerz dict[str, WidgetCallback] | NonerzWidgetCallback | NonerzWidgetArgs | NonerzWidgetKwargs | NonerrrzWidgetValuePresenter | NonereturnRegisterWidgetResult[T])r#zWidgetMetadata[T]r"r.r/r0) __future__rtypingrstreamlit.errorsrstreamlit.runtime.state.commonrrrr r r r r rrrstreamlit.runtime.scriptrunnerrr%rr&r$r7s# 2    ?37/3""&-1Z8Z8(Z8$ Z8 ! Z8 0 Z8-Z8 Z8 Z8Z8+Z8Z8zAA AAr&