o $ h@sUddlmZddlmZmZmZddlmZmZddl m Z ddl m Z ddl mZddlmZddlmZdd lmZmZdd lmZerYdd lmZdd lmZdd lmZedZded<dddZ dddZ!Gddde Z"dS) ) annotations) TYPE_CHECKINGLiteralcast)Self TypeAlias)DeltaGenerator)compute_and_register_element_id)StreamlitAPIException)Block) ForwardMsg)enqueue_messageget_script_run_ctx)register_widget) TracebackType)Cursor)WidgetCallback)smalllargemediumr DialogWidthwidthreturn'BlockProto.Dialog.DialogWidth.ValueTypecCs.|dkr tjjjS|dkrtjjjStjjjS)zMaps the user-provided literal to a value of the DialogWidth proto enum. Returns the mapped enum field for "small" by default and otherwise the mapped type. rr) BlockProtoDialogrLARGEMEDIUMSMALL)rrZ/var/www/html/mmpi2/mmpi_env/lib/python3.10/site-packages/streamlit/elements/lib/dialog.py_process_dialog_width_input)s    r! should_openboolNonecCs.t}|r|r|jrtdd|_dSdSdS)aCheck whether a dialog has already been opened in the same script run. Only one dialog is supposed to be opened. The check is implemented in a way that for a script run, the open function can only be called once. One dialog at a time is a product decision and not a technical one. Raises ------ StreamlitAPIException Raised when a dialog has already been opened in the current script run. zOnly one dialog is allowed to be opened at the same time. Please make sure to not call a dialog-decorated function more than once in a script run.TN)rhas_dialog_openedr )r"script_run_ctxrrr !_assert_first_dialog_to_be_opened8s  r'cspeZdZeddddd/ddZd0fdd Zd1ddZd2dd Zd2d!d"Zd3fd$d% Z d4fd-d. Z Z S)5rTrignore) dismissibler on_dismissparentrtitlestrr)r#rrr*+Literal['ignore', 'rerun'] | WidgetCallbackrc Cs|dvrt|std|dt}||j_||j_t||j_|dk}d}|rWt}t dd||||t|s;t |ndd}||j_ t |t|rK|nddd d d |d d |j jr`|j jjng} td |j|td} | | _|| _| S)N)r(rerunzYou have passed zI to `on_dismiss`. But only 'ignore', 'rerun', or a callable is supported.r(dialogcallback)user_keydgr,r)rr*cS|SNrxrrr ~z Dialog._create..cSr4r5rr6rrr r8r9 trigger_value)on_change_handler deserializer serializerctx value_typer) block_protodg_type)callabler rr0r,r)r!rrr r-idr _active_dg_cursor delta_pathr_blockr _delta_path_current_proto) r+r,r)rr*r@is_dismiss_activated element_idr>rFr0rrr _createRsH     zDialog._createroot_container int | Nonecursor Cursor | NoneDeltaGenerator | None block_type str | Noner$cs"t||||d|_d|_dSr5)super__init__rIrH)selfrMrOr+rR __class__rr rUs zDialog.__init__r"cCsn|jdus |jdurtdt|t}|j|jjdd<|jj |j||jjj _ |jj|_t |dS)zISend an updated proto message to indicate the open-status for the dialog.Nz;Dialog not correctly initialized. This should never happen.) rIrH RuntimeErrorr'r metadatarFdelta add_blockCopyFromr0is_openr )rVr"msgrrr _updates   zDialog._updatecC|ddS)NTr`rVrrr openz Dialog.opencCra)NFrbrcrrr closerez Dialog.closercst|Sr5)rT __enter__rcrWrr rgs zDialog.__enter__exc_typetype[BaseException] | Noneexc_valBaseException | Noneexc_tbTracebackType | NoneLiteral[False]cst|||Sr5)rT__exit__)rVrhrjrlrWrr roszDialog.__exit__) r+rr,r-r)r#rrr*r.rr) rMrNrOrPr+rQrRrSrr$r"r#rr$)rr$)rr)rhrirjrkrlrmrrn) __name__ __module__ __qualname__ staticmethodrLrUr`rdrfrgro __classcell__rrrWr rQs>  rN)rrrrrp)# __future__rtypingrrrtyping_extensionsrrstreamlit.delta_generatorrstreamlit.elements.lib.utilsr streamlit.errorsr streamlit.proto.Block_pb2r rstreamlit.proto.ForwardMsg_pb2r 7streamlit.runtime.scriptrunner_utils.script_run_contextr rstreamlit.runtime.statertypesrstreamlit.cursorrrr__annotations__r!r'rrrrr s$