o h$#@sdUddlmZddlmZmZddlmZmZmZm Z m Z ddl m Z m Z ddlmZmZddlmZddlmZddlmZmZerKdd lmZmZe eZd ed <d Zd ed<e dZ dFddZ!dGddZ"dHd!d"Z# #dIdJd'd(Z$e d)ed*Z%e d+ed*Z&d,Z'dKd1d2Z(dLd5d6Z)e dMd;d<Z*e dNd>d<Z*d?d<Z*e dOdAdBZ+e dPdDdBZ+dEdBZ+d#S)Q) annotations)EnumEnumMeta) TYPE_CHECKINGAnyFinalTypeVaroverload)configlogger)OptionSequenceconvert_anything_to_list)StreamlitAPIException)RegisterWidgetResult)Tcheck_python_comparable)IterableSequencer_LOGGERgdy=z Final[float]_FLOAT_EQUALITY_EPSILON_ValueiterableIterable[_Value]xreturnintcCs`t|D]"\}}||kr|St|tr&t|tr&t||tkr&|Sqtt|d)aReturn zero-based index of the first item whose value is equal to x. Raises a ValueError if there is no such item. We need a custom implementation instead of the built-in list .index() to be compatible with NumPy array and Pandas Series. Parameters ---------- iterable : list, tuple, numpy.ndarray, pandas.Series x : Any Returns ------- int z is not in iterable) enumerate isinstancefloatabsr ValueErrorstr)rrivaluer$a/var/www/vscode/kcb/lib/python3.10/site-packages/streamlit/elements/lib/options_selector_utils.pyindex_&sr&opt Sequence[Any]default_valuesSequence[Any] | Any | Nonelist[int] | NonecsH|durdSt|}|D]}|vrtd|dq fdd|DS)zIPerform validation checks and return indices based on the default values.NzThe default value 'zd' is not part of the options. Please make sure that every default values also exists in the options.csg|]}|qSr$)index).0r#r'r$r% Osz0check_and_convert_to_indices..)r r)r'r)r#r$r.r%check_and_convert_to_indices?s r0optionsOptionSequence[T] Sequence[T]cCst|}t||SN)r r)r1indexable_optionsr$r$r%(convert_to_sequence_and_check_comparableRsr6Nr5default list[int]cCs"t||}|dur |}|Sg}|Sr4)r0)r5r7default_indicesr$r$r%get_default_indicesXs  r:E1)boundE2)offnameOnly nameAndValuefrom_enum_value to_enum_classtype[E2]E1 | E2cCst|tstdt|t|tstdt|t||r#|Std}|tvr7tdtd|d|dkr=|S|j }|j |j ksb|dkrTt |j t |j ksb|d krkt |j t |j krktd |||Std ||||jS) aAttempt to coerce an Enum value to another EnumMeta. An Enum value of EnumMeta E1 is considered coercable to EnumType E2 if the EnumMeta __qualname__ match and the names of their members match as well. (This is configurable in streamlist configs) z,Expected an Enum in the first argument. Got z6Expected an EnumMeta/Type in the second argument. Got zrunner.enumCoercionzEInvalid value for config option runner.enumCoercion. Expected one of z , but got 'z'.r>r?r@zFailed to coerce %s to class %szCoerced %s to class %s)rrr typerr get_option&_ALLOWED_ENUM_COERCION_CONFIG_SETTINGSr __class__ __qualname__set_member_names__value2member_map_rdebug_name_)rArB coercion_typefrom_enum_classr$r$r% _coerce_enumfsB        rQ Iterable[Any]rcsLz t|}tt|Wn tyYdSwtfdd|Dr$SdS)ziReturn the common class of all elements in a iterable if they share one. Otherwise, return None. Nc3s|] }t|uVqdSr4)rE)r-item first_classr$r% sz2_extract_common_class_from_iter..)iterrEnext StopIterationall)r inner_iterr$rTr%_extract_common_class_from_iters r\register_widget_resultRegisterWidgetResult[Enum] type[Enum] opt_sequencecCdSr4r$r]r1r`r$r$r%maybe_coerce_enumrcRegisterWidgetResult[T]cCrar4r$rbr$r$r%rcrdcCsHt|jts|St|tr|}n t|}|dur|Stt|j||jS)zMaybe Coerce a RegisterWidgetResult with an Enum member value to RegisterWidgetResult[option] if option is an EnumType, otherwise just return the original RegisterWidgetResult. N)rr#rrr\rrQ value_changed)r]r1r` coerce_classr$r$r%rcs   RegisterWidgetResult[list[T]]cCrar4r$rbr$r$r%maybe_coerce_enum_sequencerdri!RegisterWidgetResult[tuple[T, T]]cCrar4r$rbr$r$r%rirdcsbtdd|jDs |St|tr|n t|dur|Stt|jfdd|jD|jS)zMaybe Coerce a RegisterWidgetResult with a sequence of Enum members as value to RegisterWidgetResult[Sequence[option]] if option is an EnumType, otherwise just return the original RegisterWidgetResult. css|]}t|tVqdSr4)rrr-valr$r$r%rVsz-maybe_coerce_enum_sequence..Nc3s|]}t|VqdSr4)rQrkrgr$r%rVs  )rZr#rrr\rrErfrbr$rmr%ris )rrrrrr)r'r(r)r*rr+)r1r2rr3r4)r5r3r7r*rr8)rAr;rBrCrrD)rrRrr)r]r^r1r_r`r(rr^)r]rer1r2r`r3rre)r]rhr1r2r`r3rrh)r]rjr1r2r`r3rrj), __future__renumrrtypingrrrrr streamlitr r streamlit.dataframe_utilr r streamlit.errorsrstreamlit.runtime.state.commonrstreamlit.type_utilrrcollections.abcrr get_logger__name__r__annotations__rrr&r0r6r:r;r=rGrQr\rcrir$r$r$r%sB          6