o hM@sUddlmZddlmZmZmZddlmZmZddl m Z m Z ddl m Z ddlmZddlmZddlmZe rAdd lmZd Zd ed <d Zd ed<eegZded<eGdddeeefZdddZdS)) annotations)IterableIteratorMutableMapping) dataclassfield) TYPE_CHECKINGFinal)parse)StreamlitAPIException) ForwardMsg)get_script_run_ctx)SupportsKeysAndGetItemembedz Final[str]EMBED_QUERY_PARAM embed_optionsEMBED_OPTIONS_QUERY_PARAMzFinal[list[str]]EMBED_QUERY_PARAMS_KEYSc@seZdZUdZeedZded<d:ddZd;d d Z dd?ddZ d@ddZdAd d!ZdBd"d#ZdCd$d%ZdCd&d'ZdDd)d*ZdEd,d-ZdFd0d1ZdGdHd5d6Zd7d8Zd9S)I QueryParamszA lightweight wrapper of a dict that sends forwardMsgs when state changes. It stores str keys with str and List[str] values. )default_factoryzdict[str, list[str] | str] _query_paramsreturn Iterator[str]cCs |tdd|jDS)Ncss|] }|tvr|VqdSNr.0keyrX/var/www/vscode/kcb/lib/python3.10/site-packages/streamlit/runtime/state/query_params.py 1sz'QueryParams.__iter__..)_ensure_single_query_api_usediterrkeysselfrrr__iter__.szQueryParams.__iter__rstrcCsn|z%|tvrtt||j|}t|tr't|dkr"WdS|dWS|WSty6tt|w)zRetrieves a value for a given key in query parameters. Returns the last item in a list or an empty string if empty. If the key is not present, raise KeyError. r)r!rKeyErrormissing_key_error_messager isinstancelistlenr%rvaluerrr __getitem__7s       zQueryParams.__getitem__r0str | Iterable[str]NonecCs |||||dSr)r!_QueryParams__set_item_internal_send_query_param_msgr/rrr __setitem__L  zQueryParams.__setitem__cCsht|tr td|d|tvrtdt|tr+t|ts+dd|D|j|<dSt||j|<dS)Nz#You cannot set a query params key `z` to a dictionary.zVQuery param embed and embed_options (case-insensitive) cannot be set programmatically.cSsg|]}t|qSr)r')ritemrrr ^sz3QueryParams.__set_item_internal..)r,dictr rrr'rr/rrr__set_item_internalQs  zQueryParams.__set_item_internalcCsN|z|tvrtt||j|=|WdSty&tt|wr)r!rr*r+rr5)r%rrrr __delitem__bs   zQueryParams.__delitem__rother\Iterable[tuple[str, str | Iterable[str]]] | SupportsKeysAndGetItem[str, str | Iterable[str]]kwdscKs|t|drt|dr|D] }||||qn |D] \}}|||q |D] \}}|||q/|dS)Nr#r1)r!hasattrr#r4itemsr5)r%r=r?rr0rrrupdatels    zQueryParams.update list[str]cCs<|||jvs |tvrgS|j|}t|tr|S|gSr)r!rrr,r-r/rrrget_alls  zQueryParams.get_allintcCs|tdd|jDS)NcSsh|]}|tvr|qSrrrrrr sz&QueryParams.__len__..)r!r.rr$rrr__len__szQueryParams.__len__cCs|t|jSr)r!r'rr$rrr__str__s zQueryParams.__str__cCsLt}|dur dS|t}tj|jdd|j_|jj|_||dS)NT)doseq) r r!r r urlencoderpage_info_changed query_stringenqueue)r%ctxmsgrrrr5s z!QueryParams._send_query_param_msgcCs ||jdd|dSNTpreserve_embed)r!clear_with_no_forward_msgr5r$rrrclearr7zQueryParams.cleardict[str, str]csfddjDS)Ncsi|] }|tvr||qSrrrr$rr s z'QueryParams.to_dict..)r!rr$rr$rto_dicts zQueryParams.to_dict_dictcCsF||j}|jddz||WdSty"||_wrP)r!rcopyrSrBr )r%rX old_valuerrr from_dicts   zQueryParams.from_dictvallist[str] | strcCs||j|<dSr)r)r%rr\rrrset_with_no_forward_msgsz#QueryParams.set_with_no_forward_msgFrRboolcsfdd|jD|_dS)Ncs"i|] \}}|tvrr||qSrr)rrr0rQrrrVs  z9QueryParams.clear_with_no_forward_msg..)rrA)r%rRrrQrrSs  z%QueryParams.clear_with_no_forward_msgcCst}|dur dS|dSr)r !mark_production_query_params_used)r%rNrrrr!s z)QueryParams._ensure_single_query_api_usedN)rrrr'rr')rr'r0r2rr3)rr'rr3)r)r=r>r?r')rr'rrC)rrE)rr')rr3)rrU)rXr>)rr'r\r]rr3)F)rRr_rr3)__name__ __module__ __qualname____doc__rr:r__annotations__r&r1r6r4r<rBrDrGrHr5rTrWr[r^rSr!rrrrr&s(             rrr'rcCs d|dS)Nzst.query_params has no key "z".r)rrrrr+s r+Nra) __future__rcollections.abcrrr dataclassesrrtypingrr urllibr streamlit.errorsr streamlit.proto.ForwardMsg_pb2r 7streamlit.runtime.scriptrunner_utils.script_run_contextr _typeshedrrrfrrr'rr+rrrrs&        &