o @j݁@shUddlmZddlZddlZddlZddlZddlZddlZddl Z ddl m Z m Z ej dkr6ddlmZddlmZmZmZmZmZddlmZmZddlmZmZdd lmZdd lmZdd lm Z dd l!m"Z"m#Z#m$Z$dd lm%Z%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z,m-Z-m.Z.m/Z/m0Z0m1Z1m2Z2ddl3Z3ddl4m5Z5ddl6m7Z7m8Z8m9Z9m:Z:m;Z;mm?Z?ddl@mAZAmBZBmCZCddlDmEZEmFZFmGZGddlHmIZIddlJmKZKddlLmMZMddl3mNZNmOZOmPZPmQZQddlRmSZSe%rej dkrddlTaTndaTddlUaUddlVaVddlWmXZXddlYmZZZe/deXe9BdZ[ndaVdaTdaUe&Z[ddd Z\dd!d"Z]dd#d$Z^e0ee_e.e0ee_ffZ`e0ee_e.e0ee_ffZaed%Zbd&ecd'<ed%Zdd(ecd)<Gd*d+d+ZeGd,d-d-ZfGd.d/d/egZhGd0d1d1ZiGd2d3d3ZjGd4d5d5ZkGd6d7d7ZlGd8d9d9eZmGd:d;d;e9Zne/d<ZoeNe0eodfeifZpeNeoejfZqe/d=erdZseNesekfZteNeselfZueZveNeoevfZw dddEdFZxGdGdHdHe_ZyGdIdJdJe ZzGdKdLdLezZ{GdMdNdNezZ|GdOdPdPezZ}GdQdRdRe}Z~GdSdTdTe}ZGdUdVdVeZGdWdXdXee)eoZGdYdZdZe ZGd[d\d\e|eZGd]d^d^e|eZGd_d`d`eZGdadbdbe|eZGdcdddde+e_e-e_fZGdedfdfeZdddkdlZdddqdrZ g g dddvdwZddgdgddxdd|d}ZdddZdddZdddZ > gddddZdddZdddZdddZdddZid>fdddZdddZdS)) annotationsN)ABCabstractmethod )BooleanOptionalAction)SUPPRESSArgumentParser NamespaceRawDescriptionHelpFormatter_SubParsersAction) defaultdictdeque)asdict is_dataclass)Enum)Path)dedent)BuiltinFunctionType FunctionTypeSimpleNamespace) TYPE_CHECKINGAnyCallableDictGenericIteratorMappingNoReturnOptionalSequenceTypeVarUnioncastoverload) dotenv_values) AliasChoices AliasPath BaseModelJson RootModelSecret TypeAdapter)Representation) WithArgsTypesorigin_is_union typing_base) deep_updateis_model_classlenient_issubclass)is_pydantic_dataclass) FieldInfo)PydanticUndefined) Annotated_AnnotatedAliasget_args get_origin)path_type_labelr )PydanticDataclass) BaseSettings PydanticModel)boundreturnNonec Cs>tdurdSzddlaWdSty}ztd|d}~ww)NrzBPyYAML is not installed, run `pip install pydantic-settings[yaml]`)yaml ImportErrorerIh/var/www/www-root/data/www/ai-bot.pdev.uz/venv/lib/python3.10/site-packages/pydantic_settings/sources.py import_yamlFs rKc Cs`tjdkr$tdur dSzddlaWdSty#}ztd|d}~wwtdur*dSddladS)Nr=rzAtomli is not installed, run `pip install pydantic-settings[toml]`)sys version_infotomlirFtomllibrGrIrIrJ import_tomlPs   rPc CsNzddlmaddlmaddlmaWdSty&}ztd|d}~ww)Nr)TokenCredential)ResourceNotFoundError) SecretClientzdAzure Key Vault dependencies are not installed, run `pip install pydantic-settings[azure-key-vault]`)azure.core.credentialsrQazure.core.exceptionsrRazure.keyvault.secretsrSrFrGrIrIrJimport_azure_key_vault`s  rWPathType DEFAULT_PATH DotenvTypeENV_FILE_SENTINELc@eZdZdZdS)NoDecodez0Annotation to prevent decoding of a field value.N__name__ __module__ __qualname____doc__rIrIrIrJr^yr^c@r]) ForceDecodez.Annotation to force decoding of a field value.Nr_rIrIrIrJrerdrec@ eZdZdS) SettingsErrorNr`rarbrIrIrIrJrgrgc@rf)_CliSubCommandNrhrIrIrIrJrjrirjc@rf)_CliPositionalArgNrhrIrIrIrJrkrirkc@rf)_CliImplicitFlagNrhrIrIrIrJrlrirlc@rf)_CliExplicitFlagNrhrIrIrIrJrmrirmcs.eZdZddfdd Zdfd d ZZS)_CliInternalArgParserTcli_exit_on_errorboolkwargsrrCrDc stjdi|||_dS)NrI)super__init___cli_exit_on_error)selfrorq __class__rIrJrss z_CliInternalArgParser.__init__messagestrrcs$|js td|t|dS)Nzerror parsing CLI: )rtrgrrerror)rurxrvrIrJrzsz_CliInternalArgParser.error)T)rorprqrrCrD)rxryrCr)r`rarbrsrz __classcell__rIrIrvrJrnsrnc@rf)CliMutuallyExclusiveGroupNrhrIrIrIrJr|rir|T _CliBoolFlagTmodel is_requiredrpro bool | NoneOptional[PydanticModel]c Cst|}|durt|r|jd}t|tr|}|durd}g}t|D]\}}t|j vrCt ||dur>t ||S| |q'|r]|rQdd |dnd}|rYt |t|dS)a Get the subcommand from a model. Args: model: The model to get the subcommand from. is_required: Determines whether a model must have subcommand set and raises error if not found. Defaults to `True`. cli_exit_on_error: Determines whether this function exits with error if no subcommand is found. Defaults to model_config `cli_exit_on_error` value if set. Otherwise, defaults to `True`. Returns: The subcommand model if found, otherwise `None`. Raises: SystemExit: When no subcommand is found and is_required=`True` and cli_exit_on_error=`True` (the default). SettingsError: When no subcommand is found and is_required=`True` and cli_exit_on_error=`False`. NroTz#Error: CLI subcommand is required {, }z@Error: CLI subcommand is required but no subcommands were found.)typer3 model_configget isinstancerp_get_model_fieldsitemsrjmetadatagetattrappendjoin SystemExitrg) rrro model_cls model_default subcommands field_name field_info error_messagerIrIrJget_subcommands*    rc@rf) EnvNoneTypeNrhrIrIrIrJrrirc@seZdZdZd(ddZd)d d Zd*ddZed+ddZed,ddZ e d-ddZ d.ddZ d/d!d"Z d0d#d$Ze d+d%d&Zd'S)1PydanticBaseSettingsSourcezi Abstract base class for settings sources, every settings source classes should inherit from it. settings_clstype[BaseSettings]cCs||_|j|_i|_i|_dSN)rrconfig_current_state_settings_sources_data)rurrIrIrJrss z#PydanticBaseSettingsSource.__init__statedict[str, Any]rCrDcC ||_dS)z Record the state of settings from the previous settings sources. This should be called right before __call__. Nr)rurrIrIrJ_set_current_state z-PydanticBaseSettingsSource._set_current_statestatesdict[str, dict[str, Any]]cCr)z Record the state of settings from all previous settings sources. This should be called right before __call__. Nr)rurrIrIrJ_set_settings_sources_datarz5PydanticBaseSettingsSource._set_settings_sources_datacC|jS)z` The current state of the settings, populated by the previous settings sources. rrurIrIrJ current_statez(PydanticBaseSettingsSource.current_statecCr)z= The state of all previous settings sources. rrrIrIrJsettings_sources_datarz0PydanticBaseSettingsSource.settings_sources_datafieldr6rrytuple[Any, str, bool]cCdS)a Gets the value, the key for model creation, and a flag to determine whether value is complex. This is an abstract method that should be overridden in every settings source classes. Args: field: The field. field_name: The field name. Returns: A tuple contains the key, value and a flag to determine whether value is complex. NrIrurrrIrIrJget_field_valuez*PydanticBaseSettingsSource.get_field_valuerpcCst|j|jS)z Checks whether a field is complex, in which case it will attempt to be parsed as JSON. Args: field: The field. Returns: Whether the field is complex. )_annotation_is_complex annotationr)rurrIrIrJfield_is_complexs z+PydanticBaseSettingsSource.field_is_complexvaluervalue_is_complexcCs(|dur||s |r||||S|S)aF Prepares the value of a field. Args: field_name: The field name. field: The field. value: The value of the field that has to be prepared. value_is_complex: A flag to determine whether value is complex. Returns: The prepared value. N)rdecode_complex_value)rurrrrrIrIrJprepare_field_value*s z.PydanticBaseSettingsSource.prepare_field_valuecCs6|rt|jvs|jddurt|jvr|St|S)a Decode the value for a complex field Args: field_name: The field name. field: The field. value: The value of the field that has to be prepared. Returns: The decoded value for further preparation enable_decodingF)r^rrrrejsonloads)rurrrrIrIrJr;s   z/PydanticBaseSettingsSource.decode_complex_valuecCdSrrIrrIrIrJ__call__Oz#PydanticBaseSettingsSource.__call__N)rr)rrrCrD)rrrCrDrCr)rCrrr6rryrCr)rr6rCrp rryrr6rrrrprCr)rryrr6rrrCr)r`rarbrcrsrrpropertyrrrrrrrrrIrIrIrJrs         rcsBeZdZdZddfdd ZdddZdddZdddZZS)DefaultSettingsSourcea Source class for loading default object values. Args: settings_cls: The Settings class. nested_model_default_partial_update: Whether to allow partial updates on nested model default object fields. Defaults to `False`. Nrr#nested_model_default_partial_updatercst|i|_|dur|n|jdd|_|jrN|jD]0\}}t||^}}|d}t t |j r>t |j |j|<qt t |j rM|j |j|<qdSdS)NrFr)rrrsdefaultsrrr model_fieldsr_get_alias_namesrrdefaultrr3 model_dump)rurrrr alias_names_preferred_aliasrvrIrJrs^s"  zDefaultSettingsSource.__init__rr6rryrCrcCrN)NrXFrIrrIrIrJrorz%DefaultSettingsSource.get_field_valuercCrr)rrrIrIrJrsszDefaultSettingsSource.__call__cC|jjd|jdS)Nz%(nested_model_default_partial_update=))rwr`rrrIrIrJ__repr__vszDefaultSettingsSource.__repr__r)rrrrrrrCry r`rarbrcrsrrrr{rIrIrvrJrTs   rcsDeZdZdZ ddfd d ZdddZdddZdddZZS)InitSettingsSourcezX Source class for loading values provided during settings class initialization. Nrr init_kwargsrrrcs8||_t||dur||_dS|jdd|_dS)NrF)rrrrsrrr)rurrrrvrIrJrss  zInitSettingsSource.__init__rr6rryrCrcCrrrIrrIrIrJrrz"InitSettingsSource.get_field_valuecCs$|jrttttf|jS|jSr)rr-rryr dump_pythonrrrIrIrJrs zInitSettingsSource.__call__cC|jjd|jdS)Nz (init_kwargs=r)rwr`rrrIrIrJrzInitSettingsSource.__repr__r)rrrrrrrrrrrIrIrvrJr|s  rcs\eZdZ     d"d#fd d Zd$ddZd%ddZd&ddZd'ddZd(d d!ZZ S))PydanticBaseEnvSettingsSourceNrrcase_sensitiver env_prefix str | Noneenv_ignore_emptyenv_parse_none_strenv_parse_enumsrCrDcst||dur |n|jdd|_|dur|n|jdd|_|dur(|n|jdd|_|dur6|n|jd|_|durF||_dS|jd|_dS)NrFrrXrrr) rrrsrrrrrrr)rurrrrrrrvrIrJrss $z&PydanticBaseEnvSettingsSource.__init__rrycCs|js|S|Sr)rlower)rurrIrIrJ_apply_case_sensitivez3PydanticBaseEnvSettingsSource._apply_case_sensitiverr6rlist[tuple[str, str, bool]]cCs4g}t|jttfr|j}n|j}|rct|trX|D]:}t|tr7||||t |dkr2dndfqt|trVt t|d}||||t |dkrRdndfqn ||||df|rl|j ddrt t|jrt|j|jr||||j|df|S||||j|df|S)a Extracts field info. This info is used to get the value of field from environment variables. It returns a list of tuples, each tuple contains: * field_key: The key of field that has to be used in model creation. * env_name: The environment variable name of the field. * value_is_complex: A flag to determine whether the value from environment variable is complex and has to be parsed. Args: field (FieldInfo): The field. field_name (str): The field name. Returns: list[tuple[str, str, bool]]: List of tuples, each tuple contains field_key, env_name, and value_is_complex. TFrpopulate_by_name)rvalidation_aliasr'r(convert_to_aliaseslistryrrlenr$rrr0r;r_union_is_complexrr)rurrrv_aliasalias first_argrIrIrJ_extract_field_infos.   (  z1PydanticBaseEnvSettingsSource._extract_field_info field_valuesrc Csi}|D]q\}}d}|j}t|}tt|jr3t|dkr3td|vr3|D] } | dur2| }nq(|r:t|ds?|||<q|jD]\} } | j sW| | krW| }nqD|s_|||<qt |jt rst |trs||||| <q||| <q|S)ae Replace field names in values dict by looking in models fields insensitively. By having the following models: ```py class SubSubSub(BaseModel): VaL3: str class SubSub(BaseModel): Val2: str SUB_sub_SuB: SubSubSub class Sub(BaseModel): VAL1: str SUB_sub: SubSub class Settings(BaseSettings): nested: Sub model_config = SettingsConfigDict(env_nested_delimiter='__') ``` Then: _replace_field_names_case_insensitively( field, {"val1": "v1", "sub_SUB": {"VAL2": "v2", "sub_SUB_sUb": {"vAl3": "v3"}}} ) Returns {'VAL1': 'v1', 'SUB_sub': {'Val2': 'v2', 'SUB_sub_SuB': {'VaL3': 'v3'}}} Nr)rrr:r0r;rrhasattrrrrr4r)rdict'_replace_field_names_case_insensitively) rurrvaluesnamersub_model_fieldrargsargsub_model_field_namefrIrIrJrs4& zEPydanticBaseEnvSettingsSource._replace_field_names_case_insensitively field_valuecCsJi}|D]\}}t|tst|ts|n||||<qd||<q|S)ze Recursively parse values that are of "None" type(EnvNoneType) to `None` type(None). N)rrrr_replace_env_none_type_values)rurrkeyrrIrIrJr's   z;PydanticBaseEnvSettingsSource._replace_env_none_type_valuesc Csi}|jjD]\}}z |||\}}}Wnty1}ztd|d|jjd|d}~wwz |||||}Wnt yV}ztd|d|jjd|d}~ww|dur|j durrt |t rk| |}nt |trrd}|jst |t r|||||<q|||<q|S)Nzerror getting value for field "z" from source ""zerror parsing value for field ")rrrr Exceptionrgrwr`r ValueErrorrrrrrrr)rudatarrr field_keyrrHrIrIrJr5sH    z&PydanticBaseEnvSettingsSource.__call__)NNNNN)rrrrrrrrrrrrrCrD)rryrCry)rr6rryrCr)rr6rrrCr)rrrCrr) r`rarbrsrrrrrr{rIrIrvrJrs   , Ercs`eZdZdZ      d%d&fdd Zd'fdd Zed(ddZd)d!d"Zd*d#d$Z Z S)+SecretsSettingsSourcezE Source class for loading settings values from secret files. Nrr secrets_dirPathType | NonerrrrrrrrCrDcs:t|||||||dur||_dS|jd|_dS)Nr)rrrsrrr)rurrrrrrrrvrIrJrs^s $zSecretsSettingsSource.__init__rcsi}|jdur |St|jttjfr|jgn|j}dd|D}g|_|D]}|s5td|dq%|j |q%t |jsC|S|jD]}| sUt dt |qFtS)z4 Build fields from "secrets" files. NcSsg|]}t|qSrI)r expanduser).0prIrIrJ wz2SecretsSettingsSource.__call__..z directory "z" does not existz.secrets_dir must reference a directory, not a )rrryosPathLike secrets_pathsexistswarningswarnrris_dirrgr<rrr)rusecrets secrets_dirsrpathrvrIrJrms"     zSecretsSettingsSource.__call__dir_pathr file_nameryrp Path | NonecCsB|D]}|j|kr|S|s|j|kr|SqdS)af Find a file within path's directory matching filename, optionally ignoring case. Args: dir_path: Directory path. file_name: File name. case_sensitive: Whether to search for file name case sensitively. Returns: Whether file path or `None` if file does not exist in directory. N)iterdirrr)clsrrrrrIrIrJfind_case_paths z$SecretsSettingsSource.find_case_pathrr6rrc Cs|||}|d^}}|D]H\}}}t|jD]=} || ||j} | s%q| rE|s6|jddr8||kr8|}|  ||fSt j d| dt | dddqqd ||fS) ag Gets the value for field from secret file and a flag to determine whether value is complex. Args: field: The field. field_name: The field name. Returns: A tuple contains the key, value if the file exists otherwise `None`, and a flag to determine whether value is complex. rrFzattempted to load secret file "z" but found a z instead.) stacklevelN) rreversedrrris_filerr read_textstriprrr<) rurr field_infos preferred_keyrrenv_namer secrets_pathrrIrIrJrs"  z%SecretsSettingsSource.get_field_valuecCr)Nz (secrets_dir=r)rwr`rrrIrIrJrrzSecretsSettingsSource.__repr__NNNNNN)rrrrrrrrrrrrrrrCrDr)rrrryrrprCrrr) r`rarbrcrsr classmethodrrrr{rIrIrvrJrYs  #rcszeZdZdZ      d/d0fdd Zd1ddZd2ddZd3dd Zd4d"d#Z d5d6d'd(Z d7d+d,Z d8d-d.Z Z S)9EnvSettingsSourcezN Source class for loading settings values from environment variables. Nrrrrrrenv_nested_delimiterrrrrCrDcsJt|||||||dur|n|jd|_t|j|_||_ dS)Nr)) rrrsrrr)rrenv_prefix_len_load_env_varsenv_vars)rurrrr)rrrrvrIrJrss  zEnvSettingsSource.__init__Mapping[str, str | None]cCsttj|j|j|jSr)parse_env_varsr environrrrrrIrIrJr+rz EnvSettingsSource._load_env_varsrr6rryrc Csld}|||}|d^}}|D] \}}} |j|}|dur0| s,|jddr.||kr.|}nq||| fS)aq Gets the value for field from environment variables and a flag to determine whether value is complex. Args: field: The field. field_name: The field name. Returns: A tuple contains the key, value if the file exists otherwise `None`, and a flag to determine whether value is complex. NrrF)rr,rr) rurrenv_valr"r#rrr$rrIrIrJrs    z!EnvSettingsSource.get_field_valuerrrrpc Cs||\}}|jrt|j|}|dur|n|}|s|rgt|tr#|S|dur5||||j}|r3|SdSz ||||}Wnt yT} z |sJ| WYd} ~ nd} ~ wwt|t ret |||||jS|S|durm|SdS)a Prepare value for the field. * Extract value for nested field. * Deserialize value to python object for complex field. Args: field: The field. field_name: The field name. Returns: A tuple contains prepared value for the field. Raises: ValuesError: When There is an error in deserializing value for complex field. N) _field_is_complexr_annotation_enum_name_to_valrrrexplode_env_varsr,rrrr2) rurrrr is_complexallow_parse_failureenum_val env_val_builtrHrIrIrJrs4    z%EnvSettingsSource.prepare_field_valuetuple[bool, bool]cCsB||r d}d|fStt|jrt|j|jrd}d|fSdS)za Find out if a field is complex, and if so whether JSON errors should be ignored FT)FF)rr0r;rrr)rurr5rIrIrJr1's z#EnvSettingsSource._field_is_complexFieldInfo | Any | NonerFieldInfo | Nonec Cs|sdSt|tr |jn|}tt|st|tr/t|D]}||||}|r,|SqdSt|s7t |r{t |}| D];\}} | | |D]0\} } } |dusT|rc||ks\| |krb| SqI| | kss| | kry| SqIq?dS)aK Find the field in a sub model by key(env name) By having the following models: ```py class SubSubModel(BaseSettings): dvals: Dict class SubModel(BaseSettings): vals: list[str] sub_sub_model: SubSubModel class Cfg(BaseSettings): sub_model: SubModel ``` Then: next_field(sub_model, 'vals') Returns the `vals` field of `SubModel` class next_field(sub_model, 'sub_sub_model') Returns `sub_sub_model` field of `SubModel` class Args: field: The field. key: The key (env name). case_sensitive: Whether to search for key case sensitively. Returns: Field if it finds the next field otherwise `None`. N)rr6rr0r;r/r: next_fieldr3r5rrrr) rurrrrtype_ type_has_keyfieldsrrrr$rIrIrJr;7s.       zEnvSettingsSource.next_fieldr,rc sbtt|jt}fdd||D}i}|D]\}tfdd|Ds+qjd}|j ^} } } |} |} | D]} | |j } t | trW| |i} qB | | j } | sd|r|r| rp| \}}nd\}}|rz | | |}Wnty}z |s|WYd}~nd}~wwt | tr| | vst |tr| | ikr|| | <q|S)a Process env_vars and extract the values of keys containing env_nested_delimiter into nested dictionaries. This is applied to a single field, hence filtering by env_var prefix. Args: field_name: The field name. field: The field. env_vars: Environment variables. Returns: A dictionary contains extracted values from nested env values. cs g|] \}}}|jqSrI)r))rrr$rrIrJr }sz6EnvSettingsSource.explode_env_vars..c3s|]}|VqdSr) startswith)rprefix)r$rIrJ z5EnvSettingsSource.explode_env_vars..N)TT)r4r;rrrranyr*splitr)r;rr setdefaultr1rrr)rurrr,is_dictprefixesresultr0env_name_without_prefixrkeyslast_keyenv_var target_fieldrr4allow_json_failurerHrI)r$rurJr3msH       z"EnvSettingsSource.explode_env_varscC|jjd|jd|jdS)Nz(env_nested_delimiter=, env_prefix_len=r)rwr`r)r*rrIrIrJrszEnvSettingsSource.__repr__r&)rrrrrrr)rrrrrrrrCrDrCr-rr)rr6rCr8r)rr9rryrrrCr:)rryrr6r,r-rCrr) r`rarbrcrsr+rrr1r;r3rr{rIrIrvrJr(s"    . 67r(cseZdZdZedddddddfd+fdd Zd,ddZedddddd-dd Zd.d!d"Z d,d#d$Z d/fd&d' Z d0d)d*Z Z S)1DotEnvSettingsSourcezB Source class for loading settings values from env files. Nrrenv_fileDotenvType | Noneenv_file_encodingrrrrr)rrrrCrDc sP|tkr|n|jd|_|dur|n|jd|_t||||||| dS)NrSrU)r\rrrSrUrrrs) rurrSrUrrr)rrrrvrIrJrss zDotEnvSettingsSource.__init__r-cCs|Sr)_read_env_filesrrIrIrJr+sz#DotEnvSettingsSource._load_env_varsFencodingr ignore_emptyparse_none_str file_pathrrXrprYrZcCst||pdd}t||||S)Nutf8rX)r&r.)r[rXrrYrZ file_varsrIrIrJ_static_read_env_files z*DotEnvSettingsSource._static_read_env_filecCs|j||j|j|j|jdS)NrW)r_rUrrr)rur[rIrIrJ_read_env_filesz#DotEnvSettingsSource._read_env_filecCs^|j}|dur iSt|ttjfr|g}i}|D]}t|}|r,|| |q|Sr) rSrryr r rrrupdater`)ru env_files dotenv_varsrSenv_pathrIrIrJrVs z$DotEnvSettingsSource._read_env_filesrc st}|jddk}|jD]f\}}|r||vrqd}|jjD]6\}}|||D]'\}} }|| ksRt |j |j sMt t |j rVt|j |j rV|| rVd}nq/|r[nq%|sx|rt||jrt|t|jd} ||| <q|||<q|S)NextraforbidFT)rrrrrr,rrrrrrrr0r;rr?rr) ruris_extra_allowedr$ env_valueenv_usedrrrfield_env_namenormalized_env_namervrIrJrs<       zDotEnvSettingsSource.__call__ryc Cs.|jjd|jd|jd|jd|jd S)Nz (env_file=z, env_file_encoding=, env_nested_delimiter=rPr)rwr`rSrUr)r*rrIrIrJrs zDotEnvSettingsSource.__repr__)rrrSrTrUrrrrrr)rrrrrrrrCrDrQ r[rrXrrrprYrprZrrCr-)r[rrCr-rr)r`rarbrcr\rsr+ staticmethodr_r`rVrrr{rIrIrvrJrRs,  #rRcseZdZdZddddddddddddddddejejejej e fdfd"d# Z e dd%d&Z e dd*d&Z e dd-d&Z ddd.dfd2d&Z e dd4d5Ze dd6d5Zdd7dd9d5Zdd?d@ZddAdBZddGdHZddIdJZddMdNZddTdUZddVdWZddYdZZedd\d]ZddbdcZddddeZejejejej e fddfdgZddpdqZddsdtZddudvZddydzZdd~dZ dddZ!dddZ"ddddZ#dddZ$dddZ%dddZ&Z'S)CliSettingsSourcea Source class for loading settings values from CLI. Note: A `CliSettingsSource` connects with a `root_parser` object by using the parser methods to add `settings_cls` fields as command line arguments. The `CliSettingsSource` internal parser representation is based upon the `argparse` parsing library, and therefore, requires the parser methods to support the same attributes as their `argparse` library counterparts. Args: cli_prog_name: The CLI program name to display in help text. Defaults to `None` if cli_parse_args is `None`. Otherwse, defaults to sys.argv[0]. cli_parse_args: The list of CLI arguments to parse. Defaults to None. If set to `True`, defaults to sys.argv[1:]. cli_parse_none_str: The CLI string value that should be parsed (e.g. "null", "void", "None", etc.) into `None` type(None). Defaults to "null" if cli_avoid_json is `False`, and "None" if cli_avoid_json is `True`. cli_hide_none_type: Hide `None` values in CLI help text. Defaults to `False`. cli_avoid_json: Avoid complex JSON objects in CLI help text. Defaults to `False`. cli_enforce_required: Enforce required fields at the CLI. Defaults to `False`. cli_use_class_docs_for_groups: Use class docstrings in CLI group help text instead of field descriptions. Defaults to `False`. cli_exit_on_error: Determines whether or not the internal parser exits with error info when an error occurs. Defaults to `True`. cli_prefix: Prefix for command line arguments added under the root parser. Defaults to "". cli_flag_prefix_char: The flag prefix character to use for CLI optional arguments. Defaults to '-'. cli_implicit_flags: Whether `bool` fields should be implicitly converted into CLI boolean flags. (e.g. --flag, --no-flag). Defaults to `False`. cli_ignore_unknown_args: Whether to ignore unknown CLI args and parse only known ones. Defaults to `False`. cli_kebab_case: CLI args use kebab case. Defaults to `False`. case_sensitive: Whether CLI "--arg" names should be read with case-sensitivity. Defaults to `True`. Note: Case-insensitive matching is only supported on the internal root parser and does not apply to CLI subcommands. root_parser: The root parser object. parse_args_method: The root parser parse args method. Defaults to `argparse.ArgumentParser.parse_args`. add_argument_method: The root parser add argument method. Defaults to `argparse.ArgumentParser.add_argument`. add_argument_group_method: The root parser add argument group method. Defaults to `argparse.ArgumentParser.add_argument_group`. add_parser_method: The root parser add new parser (sub-command) method. Defaults to `argparse._SubParsersAction.add_parser`. add_subparsers_method: The root parser add subparsers (sub-commands) method. Defaults to `argparse.ArgumentParser.add_subparsers`. formatter_class: A class for customizing the root parser help text. Defaults to `argparse.RawDescriptionHelpFormatter`. NTrr cli_prog_namercli_parse_args)bool | list[str] | tuple[str, ...] | Nonecli_parse_none_strcli_hide_none_typercli_avoid_jsoncli_enforce_requiredcli_use_class_docs_for_groupsro cli_prefixcli_flag_prefix_charcli_implicit_flagscli_ignore_unknown_argscli_kebab_caser root_parserrparse_args_methodCallable[..., Any] | Noneadd_argument_methodadd_argument_group_methodadd_parser_methodadd_subparsers_methodformatter_classrCrDc s|dur|n |jdtjd|_|dur|n|jdd|_|dur%|n|jdd|_|s8|jdur6dnd}||_|durA|n|jd d|_|durO|n|jd d|_ | dur]| n|jd d|_ | durk| n|jd d |_ | dury| n|jdd|_ |j d|_ |j r| ds| ds| dd std| |j d7_ | dur| n|jdd|_| dur| n|jdd|_|dur|n|jdd|_|dur|nd}|s|durtdtj|d|jd|j |d|durt|j |j|jdurdnt|j||j ddn|}|j|||||||d|dvrO|dur1tjdd}nt|ttfsBtdt||j |!|j"|ddSdS)NrprrtFruTrDnullrvrwrorxrXry-r.z'CLI settings source prefix is invalid: rzr{r|zGCase-insensitive matching is only supported on the internal root parser)r)rrrr)roprog descriptionr prefix_chars allow_abbrev)r}r~rrrrrNFrz>cli_parse_args must be List[str] or Tuple[str, ...], recieved  parsed_args)#rrrLargvrprtrursrvrwrorxry_cli_flag_prefixr?endswithreplace isidentifierrgrzr{r|rrrsrnrcr_connect_root_parserrrtuplerr+ _parse_argsr})rurrprqrsrtrurvrwrorxryrzr{r|rr}r~rrrrrrvrIrJrsLs      $        zCliSettingsSource.__init__rcCrrrIrrIrIrJrzCliSettingsSource.__call__r"list[str] | tuple[str, ...] | boolCliSettingsSource[T]cCr)a Parse and load the command line arguments list into the CLI settings source. Args: args: The command line arguments to parse and load. Defaults to `None`, which means do not parse command line arguments. If set to `True`, defaults to sys.argv[1:]. If set to `False`, does not parse command line arguments. Returns: CliSettingsSource: The object instance itself. NrI)rurrIrIrJrrr,Namespace | SimpleNamespace | dict[str, Any]cCr)a Loads parsed command line arguments into the CLI settings source. Note: The parsed args must be in `argparse.Namespace`, `SimpleNamespace`, or vars dictionary (e.g., vars(argparse.Namespace)) format. Args: parsed_args: The parsed args to load. Returns: CliSettingsSource: The object instance itself. NrIrurrIrIrJr)rr)list[str] | tuple[str, ...] | bool | None?Namespace | SimpleNamespace | dict[str, list[str] | str] | None%dict[str, Any] | CliSettingsSource[T]cs~|dur |dur td|dur0|dur|jidS|dur%tjdd}|j||j|dS|dur:|j|dStS)Nz/`args` and `parsed_args` are mutually exclusiveFrTr)rgr+rLrrr}rrr)rurrrvrIrJrs   r-cCrrrIrrIrIrJr+rz CliSettingsSource._load_env_varscCr)a Loads the parsed command line arguments into the CLI environment settings variables. Note: The parsed args must be in `argparse.Namespace`, `SimpleNamespace`, or vars dictionary (e.g., vars(argparse.Namespace)) format. Args: parsed_args: The parsed args to load. Returns: CliSettingsSource: The object instance itself. NrIrrIrIrJr+rr/Mapping[str, str | None] | CliSettingsSource[T]cs,|duriSt|ttfrt|}g}|D]0\}}t|tr)|||||<q|drG|durG|dd|}|j ||}| |q|j D]}| D] }||vr^|j ||<qSqMdd|D}|rt |tdtfdd|Dsd |<tttttf||j|j|j |_|S) N :subcommand:rcSs i|] \}}|ds||qS)r)r)rrvalrIrIrJ ) z4CliSettingsSource._load_env_vars..)rc3s"|] }d|vr|VqdS)rNrI)rrlast_selected_subcommandrIrJrA,s z3CliSettingsSource._load_env_vars..z{})rr rvarsrr_merge_parsed_listrrD_cli_subcommandsrrrsmaxrrCrJr.r$rryrrr,)rurselected_subcommandsrrsubcommand_namesubcommand_destrrIrrJr+s>      parsed_list list[str]rry%tuple[Optional[type], Optional[type]]cCsr|j|t}|tustt|rtddt|Ds"|}||fS|r3t|dks1|ddr3tnt }||fS)Ncss0|]}|tdurt|ttfvr|VqdSr)rr;rrrr<rIrIrJrA?szACliSettingsSource._get_merge_parsed_list_types..rr[) _cli_dict_argsrrr0r;rCr:rr?ry)rurr merge_type inferred_typerIrIrJ_get_merge_parsed_list_types8s &z.CliSettingsSource._get_merge_parsed_list_typesc Cszg}d}|||\}}|D]}}t|tsnu|}|dr.|dr.|dd}|r|}|drC||||}d}n>|dsM|drT|||}n+z ||||}Wn!t y~}z||urk||}||||}WYd}~nd}~wwd}|s0|s||||}q|tur|d WS|t urdd |dWSi} |D] } | t | qt | WSty}z td |d |d}~ww) NFr]r,{TrzParsing error encountered for z: )rrryr!r?r_consume_comma_consume_object_or_array_consume_string_or_numberrrrrarrdumpsrrg) rurr merged_listis_last_consumed_a_valuerrrrH merged_dictitemrIrIrJrKsV    z$CliSettingsSource._merge_parsed_listrrrrpcCs|s|d|ddS)Nz""r)r)rurrrrIrIrJrxs  z CliSettingsSource._consume_commacCsd}|dr dnd}tdt|D]6}||dvr|d7}q||dvrH|d8}|||krH|dkrH||d|d||ddSqtd|d ) Nrrrr)rr)rrrzMissing end delimiter "r)r?rangerrrg)rurrcount close_delimconsumedrIrIrJr}s   z*CliSettingsSource._consume_object_or_arrayrtype[Any] | Nonec Csz|turdnt|}d}|t|kr:||dkr'|dks$||ddkr'| }|s0||dkr0n |d7}|t|ks|r@td|d|}|ttfvrzzt|Wntys||jkrbd}|d vrq|dsqd|d}Ynw| |n=d d | d dD\}}|dr| ds|ds| drtd ||d|d}}| t ||i||dS)NrFrr\rzMismatched quotesr)truefalsercss|]}|VqdSrrI)rkvrIrIrJrAsz>CliSettingsSource._consume_string_or_number..=z1Dictionary key=val parameter is a quoted string: )ryrrgr!rfloatrrsr?rrDrrr) rurrrris_find_end_quote val_stringrrrIrIrJrs: $       ( z+CliSettingsSource._consume_string_or_numberrtype[BaseModel]rr6list[type[BaseModel]]cCst|js |jfnt|j}|jrtdd|D}g}|D];}t|tfddr3td|jd|t|tfddrFtd|jd|t t |sRt t |rY| t |q|S)NcSg|] }|tdur|qSrrrrIrIrJr z5CliSettingsSource._get_sub_models..F)is_include_originz.CliSubCommand is not outermost annotation for rz1CliPositionalArg is not outermost annotation for ) r:rrtr_annotation_contains_typesrjrgr`rkr3_strip_annotatedr5r)rurrr field_types sub_modelsr<rIrIrJ_get_sub_modelssz!CliSettingsSource._get_sub_modelscCst|jvrd}n t|jvrd}ndS|jtur%t|d|jd|dtjdkrB|j t urD|j durFt|d|jd|ddSdSdS)NCliImplicitFlagCliExplicitFlagz argument rz is not of type boolrz, must have default for python versions < 3.9) rlrrmrrprgr`rLrMrr7default_factory)rurrr cli_flag_namerIrIrJ_verify_cli_flag_annotationss    z.CliSettingsSource._verify_cli_flag_annotationslist[tuple[str, FieldInfo]]c Cslggg}}}t|D]\}}t|jvrk|s'td|jd|dt||^}}t|dkr@td|jd|dddt |j D} | D]} t | sbt | sbtd|jd|dqL| ||fqt|jvr|std |jd|dt||^}}t|dkrtd |jd|d| ||fq||||| ||fq|||S) Nzsubcommand argument rz has a default valuerz has multiple aliasescSrrrrrIrIrJr rz6CliSettingsSource._sort_arg_fields..z$ has type not derived from BaseModelzpositional argument )rrrjrrrgr`rrr:rr3r5rrkr) rurpositional_argssubcommand_args optional_argsrrrrr field_typerIrIrJ_sort_arg_fieldss4     z"CliSettingsSource._sort_arg_fieldsr}cCr)z#The connected root parser instance.) _root_parserrrIrIrJr}szCliSettingsSource.root_parser parser_method method_namerqCallable[..., Any]cs^dur jdur dkr tjtr ddfd d }|Sdur-dfdd }|SS)NFr~r}rnr"list[str] | tuple[str, ...] | None namespaceNamespace | NonerCrcsvg}|r tt|ngD]&}djd}td|d|}|r/|d|d}||q|||S)Nrz{1,2}z^(z [^\s=]+)(.*)rr) shlexrDrryrematchgrouprr)r}rrinsensitive_argsr flag_prefixmatched)rrurIrJparse_args_insensitive_methods  zOCliSettingsSource._connect_parser_method..parse_args_insensitive_methodrqcstdd)Nz0cannot connect CLI settings source root parser: z. is set to `None` but is needed for connecting)rgrrq)rrIrJnone_parser_methods zDCliSettingsSource._connect_parser_method..none_parser_method)NN)r}rnrrrrrCr)rrrqrrCr)rrrrn)rurrrrqrrrI)rrrurJ_connect_parser_methods  z(CliSettingsSource._connect_parser_methodcs||ddfdd }|S) NrparserrrqrCcsxdsd|fiSfdddD}|dd7<|fi|}t|ds4td |jd iS) N _is_cli_mutually_exclusive_grouprequiredcs i|] }|vr||qSrI)poprrrqrIrJrrzUCliSettingsSource._connect_group_method..add_group_method..)titlerrz (mutually exclusive)add_mutually_exclusive_groupzcannot connect CLI settings source root parser: group object is missing add_mutually_exclusive_group but is needed for connectingrI)rrrgr)rrqmain_group_kwargsradd_argument_grouprrJadd_group_methods   zACliSettingsSource._connect_group_method..add_group_method)rrrqrrCr)r)rurrrIrrJ_connect_group_methods z'CliSettingsSource._connect_group_methodc Csd dd}||_|dur|jr|ntj}||d|_||d |_|||_||d |_ ||d |_ ||_ i|_ t t|_|j|j|jg|j|jdgtd dS)NrrrqrCr c_stj|i|dSNr)r parse_known_argsrrIrIrJ_parse_known_args0szACliSettingsSource._connect_root_parser.._parse_known_argsr~rrrrr added_args arg_prefixsubcommand_prefixralias_prefixesr)rrrqrrCr )rr{r parse_argsrr _add_argumentr _add_group _add_parser_add_subparsers_formatter_classrrrr_add_parser_argsr}rrr7) rur}r~rrrrrrrIrIrJr&s,    z&CliSettingsSource._connect_root_parserrr r r rr rr c sd} i} tt|stt|rtt||sdn|}||D]\} } ||| | } t| | | |jd\}}|d}t| j vr| D]}| t | dkrO|j n|}||}||}||j |d|<t | dkrpdn| j}|jr|jdur}dnt|j}| dur|j|d|dt | dkr| jnddn| } t| dr| jr| jddd |d nd |d | _|j|j| |||j|jdurdnt|jd |g||d ||d dgtdqBq |jt| jttttt fdd}| o| }i}t!|d<|"| | ||d<|#| j|d<|j$o| %o|tu|d<|r7| j&dur7|s7|||j'dn|||d<|(|||||}|rR|d|vrSq |rmd|d<t| jtt fddrm| j|j)|d<t*| j vr| |+|d<|dg}|d=|d=d|,|| ||r|j-||| |||||| | ||d q |s|durt.|tr|j/|fi|}|t|7}|j0|gfdd|DRi|q |t|7}|j0|gfdd|DRi|q |1|| |||||S)N)alias_path_argsrrrrr)rdestrmetavarrrrr)helprrrr T)is_strip_annotatedrrrrractionrX)rc3(|]}dt||VqdSrrrrrrIrJrA&z5CliSettingsSource._add_parser_args..c3rrrrrrIrJrAr)2r3rr5 issubclassrrrrrjr_check_kebab_namerr`rrrwrcrrrrrrrr7rrrrsetrr!r CLI_SUPPRESS _help_format_metavar_formatrvrrr*_get_arg_namesrrkupper_convert_bool_flag_add_parser_submodelsrrr_add_parser_alias_paths)rurrr r r rr r subparsersrrrrris_alias_path_onlyrsubcommand_aliasrrsubcommand_helpis_append_actionis_parser_submodelrq arg_namesrIrrJrIs              /       ( z"CliSettingsSource._add_parser_argsrcCs|jr |ddS|S)Nrr)r|r)rurrIrIrJr!s z#CliSettingsSource._check_kebab_namecCs|ddkrEd}|jtur|j}|tur|}tjdks t|trG|js(t|jvrKt |jvrI|d=tjdkr7t n dt |  |d<dSdSdSdSdS)Nrrprstore_r) rr7rLrMrrprzrlrrmrryr)rurqrrrrIrIrJr(s     z$CliSettingsSource._convert_bool_flagrtuple[str, ...]c Csfg}|g|D])}|D]$}|||jkr||n ||dd|} | |vr/|| q q|S)NrXr)r!rrr) rur r r rr r1r@rarg_namerIrIrJr&s   z CliSettingsSource._get_arg_namesrrr1c st|tr tdd}i}|dd|d<| j|d<| d|d<tdd|D|d <|d r:t|d kr:td |jrUt|d krU|djdurLdnt|dj|d<| t urkt t | set t | rjt | | } n| jt urt| j} n| jdur|| j} | durd |jd }|ddurt|d|d|d<n||d<| d}|js||dd|dd| d<|j|fi|}|j|gfdd|DRi| |D]#}|j||||d||r|n|fdd| d dD| dqdS)Nz8cannot have nested models in a CliMutuallyExclusiveGrouprz optionsrrrcs|]}t|tVqdSr)r r|)rrrIrIrJrAs  z:CliSettingsSource._add_parser_submodels..rrz/cannot use union with CliMutuallyExclusiveGroupz default: z (undefined) zset z from JSON stringrc3s|] }|VqdSrrIrrrIrJrA;srcsg|] }|dqS)rrIr)r rIrJr Dsz;CliSettingsSource._add_parser_submodels..r )r r|rgrrCrrwrcrr7r3rr5rrrrsrurrrr)rurrrr r r rr1rqrrrr model_groupmodel_group_kwargs desc_headerrrI)r rrJr)sZ    $    &  z'CliSettingsSource._add_parser_submodelsrdict[str, str]c Cs|r|}|durt|tr|j|fi|n|}|d}|r%|ddn|}|D]_\} } |r3dn| } ||jkr@|| n ||dd| } i} t| d<d| d<|| | d<| d ksc|rhd | d <nd | d <d | d <| |vr|| |j ||j | fi| q+dSdS)NrrrXrrzpydantic alias pathrrrrrrr) rrrrrrrr#rrr) rurrr r r rcontextis_nested_alias_pathrrr4rqrIrIrJr*Hs4         z)CliSettingsSource._add_parser_alias_pathsobjcCs$|jst|Stddt|DS)NcSrrrrrIrIrJr orz8CliSettingsSource._get_modified_args..)rtr:rrur=rIrIrJ_get_modified_argsksz$CliSettingsSource._get_modified_args obj_qualnamecCsxd|vr|d|dddd||dddD}d|}|r.|d|dSt|dkr6|Sd|d S) NJSONrcSsg|]}|dkr|qS)rArIrrIrIrJr sr z=CliSettingsSource._metavar_format_choices..rrrrr)indexrr)rurr@rrIrIrJ_metavar_format_choicesqs 6 z)CliSettingsSource._metavar_format_choicescCsjt|}t|rd|jvr|jS|jS|durdSt|tr"t|St|tjr,t |St|t t t fs7|j }tt|rK|tt|j||St|tjtjfvrb|ttt ||St|trq|dd|DSt|t r|jtt|j||t|dr|jdSt |dS|t dur|jSt|rd St|t r|jSt|d d d d S) z\Pretty metavar representation of a type. Adapts logic from `pydantic._repr.display_as_type`.z.z...cSg|]}|jqSrIrrrrIrIrJr z=CliSettingsSource._metavar_format_recurse..rb)r@NrAztyping.rXztyping_extensions.)r _is_functionrbr`rr.reprtyping_extensions TypeAliasTyperyr1r/rrwr0r;rCrmap_metavar_format_recurser?Literaltypingr4rrrsr3rr>rIrIrJrMzs>       z)CliSettingsSource._metavar_format_recursecCs||ddS)Nrr)rMrr>rIrIrJr%rz!CliSettingsSource._metavar_formatcCsd|jr|jnd}|tkst|jvrtS|r;|tdfvr;t|jvr:|dur(dnd}||r3d|dnd|d7}ngd|jd}tt|sNt t|rXdt ||d}n?|tdfvrlt |rld| |d}n+|j tdfvrt|j|j }d|dur|j n|d}n|jdurd| |jd}||rd |n|7}tt|jtr|d d S|S) NrXzifdef: z (z required)(z (default: rz(default factory:  %z%%)rr#rrr7rkrsr3rr5rrHr%r_annotation_enum_val_to_namerrr rr r)rurrr_helpifdefr enum_namerIrIrJr$s(    zCliSettingsSource._help_format).rrrprrqrrrsrrtrrurrvrrwrrorrxrryrrzrr{rr|rrrr}rr~rrrrrrrrrrrrCrDr)rrrCr)rrrCr)rrrrrCrrQ)rrrCr)rrrryrCr)rrrryrCry)rryrrrrprCry)rryrrrCry)rryrrrrrCry)rrrryrr6rCr)rrrryrr6rCrD)rrrCr)rCr}) rrrryrrrqrrCr)rrrCr)r}r}r~rrrrrrrrrrrrCrD)rrrrr rr ryr ryrrr rrrrCr )rryrCry)rqrrr6rrrCrD) r ryr ryr rrr3r rrCr)rrrrrrr rr ryr ryrryr1rrqrrryrr6rr3rrrCrD)rrrr:r rr ryr ryrrrCrD)r=rrCr3r)rrr@rrCry)r=rrCry)rryrr6rrrCry)(r`rarbrcr add_argumentrr add_parseradd_subparsersr rsr%rr+rrrrrrrrrr}rrrrr!r(r&r)r*r?rCrMr%r$r{rIrIrvrJros/{      &  -       & #     E #  $roc@s$eZdZd ddZed d d Zd S)ConfigFileSourceMixinfilesrrCrcCsX|duriSt|ttjfr|g}i}|D]}t|}|r)|||q|Sr) rryr r rrrra _read_file)rur[rfiler[rIrIrJ _read_filess z!ConfigFileSourceMixin._read_filesrrcCrrrI)rurrIrIrJr\rz ConfigFileSourceMixin._read_fileN)r[rrCr)rrrCr)r`rarbr^rr\rIrIrIrJrZs  rZc<eZdZdZedfdfd d ZdddZdddZZS)JsonConfigSettingsSourcez> A source class that loads variables from a JSON file Nrr json_filerjson_file_encodingrcV|tkr|n|jd|_|dur|n|jd|_||j|_t||jdS)Nrarb) rZrrjson_file_pathrbr^ json_datarrrs)rurrarbrvrIrJrs z!JsonConfigSettingsSource.__init__r[rrCrcCs<t||jd }t|WdS1swYdSNr])openrbrload)rur[rarIrIrJr\s$z#JsonConfigSettingsSource._read_filerycCr)Nz (json_file=r)rwr`rdrrIrIrJrrz!JsonConfigSettingsSource.__repr__)rrrarrbrr[rrCrr r`rarbrcrZrsr\rr{rIrIrvrJr`s r`cs:eZdZdZefdfdd Zdd d ZdddZZS)TomlConfigSettingsSourcez> A source class that loads variables from a TOML file rr toml_filercs<|tkr|n|jd|_||j|_t||jdS)Nrm)rZrrtoml_file_pathr^ toml_datarrrs)rurrmrvrIrJrssz!TomlConfigSettingsSource.__init__r[rrCrcCsbtt|dd}tjdkrt|WdSt|WdS1s*wYdS)Nrb)moder=)rPrhrLrMrNrirO)rur[rmrIrIrJr\s $z#TomlConfigSettingsSource._read_filerycCr)Nz (toml_file=r)rwr`rnrrIrIrJrrz!TomlConfigSettingsSource.__repr__)rrrmrrjrrkrIrIrvrJrls  rlcs4eZdZdZ ddfd d ZedddZZS)!PyprojectTomlConfigSettingsSourcezK A source class that loads variables from a `pyproject.toml` file. NrrrmrrCrDcsj|||jdd|_|jdd|_||j|_|jD] }|j|i|_qtt| ||jdS)Npyproject_toml_depthrpyproject_toml_table_header)toolzpydantic-settings) _pick_pyproject_toml_filerrrntoml_table_headerr^rorrrlrs)rurrmrrvrIrJrss z*PyprojectTomlConfigSettingsSource.__init__provideddepthintrcCsz|r|Std}d}|s;|jjd}||kr;|r"|St|j|jkr- |S|jjd}|d7}||ks|S)zPick a `pyproject.toml` file path to use. Args: provided: Explicit path provided when instantiating this class. depth: Number of directories up the tree to check of a pyproject.toml. zpyproject.tomlrr)resolvercwdrparentryroot)rxryrvrchildrIrIrJrvs     z;PyprojectTomlConfigSettingsSource._pick_pyproject_toml_filer)rrrmrrCrD)rxrryrzrCr)r`rarbrcrsrnrvr{rIrIrvrJrrs rrcr_)YamlConfigSettingsSourcez> A source class that loads variables from a yaml file Nrr yaml_fileryaml_file_encodingrcrc)Nrr) rZrryaml_file_pathrr^ yaml_datarrrs)rurrrrvrIrJrs3rfz!YamlConfigSettingsSource.__init__r[rrCrcCsFtt||jd}t|piWdS1swYdSrg)rKrhrrE safe_load)rur[rrIrIrJr\Bs $z#YamlConfigSettingsSource._read_filerycCr)Nz (yaml_file=r)rwr`rrrIrIrJrGrz!YamlConfigSettingsSource.__repr__)rrrrrrrjrrkrIrIrvrJr.s rc@sNeZdZUded<ded<ded<dd d ZdddZdddZdddZdS)AzureKeyVaultMappingzdict[str, str | None]_loaded_secretsrS_secret_clientr _secret_names secret_clientrCrDcCs&i|_||_dd|jD|_dS)NcSrDrIrE)rsecretrIrIrJr VrGz1AzureKeyVaultMapping.__init__..)rrlist_properties_of_secretsrrurrIrIrJrsPszAzureKeyVaultMapping.__init__rryrcCsD||jvrz |j|j|j|<Wn tyt|w|j|Sr)rr get_secretrrKeyError)rurrIrIrJ __getitem__Xs   z AzureKeyVaultMapping.__getitem__rzcC t|jSr)rrrrIrIrJ__len__a zAzureKeyVaultMapping.__len__ Iterator[str]cCrr)iterrrrIrIrJ__iter__drzAzureKeyVaultMapping.__iter__N)rrSrCrD)rryrCr)rCrz)rCr)r`rarb__annotations__rsrrrrIrIrIrJrKs    rcsTeZdZUded<ded<ded<   ddfdd ZdddZdddZZS)AzureKeyVaultSettingsSourcery_urlrQ _credentialrSrNrrurl credentialrrrrrrCrDc s0t||_||_tj|d|dd||ddS)NTz--F)rrr)rrr)rWrrrrrs)rurrrrrrrvrIrJrsms  z$AzureKeyVaultSettingsSource.__init__Mapping[str, Optional[str]]cCst|j|jd}t|S)N) vault_urlr)rSrrrrrIrIrJr+sz*AzureKeyVaultSettingsSource._load_env_varscCrO)Nz(url=rlr)rwr`rr)rrIrIrJrsz$AzureKeyVaultSettingsSource.__repr__)NNN)rrrryrrQrrrrrrrCrD)rCrr)r`rarbrrsr+rr{rIrIrvrJrhs  rFrryrcCs|r|S|Srr)rrrIrIrJ_get_env_var_keysrrrrZstr | None | EnvNoneTypecCs||kr|dus |St|Sr)r)rrZrIrIrJ_parse_env_none_strsrr,r-rYcsfdd|DS)Ncs.i|]\}}r |dkst|t|qS)rX)rr)rkvrrYrZrIrJrsz"parse_env_vars..)r)r,rrYrZrIrrJr.sr.rWr[rrXcCs tdttj|||||dS)Nzmread_env_file will be removed in the next version, use DotEnvSettingsSource._static_read_env_file if you mustrW)rrDeprecationWarningrRr_)r[rXrrYrZrIrIrJ read_env_filesrrrr list[Any]cCsz|durt|tr|jdd}|dur|dkr|}Wn ty%Ynwtdd|Dr1dSt|trAt|^}}t ||St |}|t urKdSt |p\t |p\t |dp\t |dS)Nr~RootModelRootTypecsr5r)rr*)rmdrIrIrJrArBz)_annotation_is_complex..F__pydantic_core_schema____get_pydantic_core_schema__)r r+rr TypeErrorrCrr9r:rr;r,_annotation_is_complex_innerr)rrroot_annotationinnermetaoriginrIrIrJrs0    rc Cs2t|ttfr dSt|ttttttt fpt |Sr) r4rybytesr)rr!rr" frozensetrrrrIrIrJrs rcstfddt|DS)Nc3s|]}t|VqdSr)rrrrIrJrArBz$_union_is_complex..)rCr:)rrrIrrJrsrtypestuple[Any, ...]rrcCsP|rt|}|durt||vrdSt|D] }t||d|dr#dSq||vS)NT)rr)rr;r:r)rrrrr<rIrIrJrs rrcCs(t|tkrt|d}t|tks|Sr)r;r8r:rrIrIrJrs   r Optional[str]cCsN|t|gt|RD]}t|tr$|tdd|Dvr$||jSq dS)Ncs|]}|jVqdSr)rrFrIrIrJrAz/_annotation_enum_val_to_name..)r;r:r4rrr)rrr<rIrIrJrSs  rSrcCsL|t|gt|RD]}t|tr#|tdd|Dvr#||Sq dS)NcsrrrErFrIrIrJrA rz/_annotation_enum_name_to_val..)r;r:r4rr)rrr<rIrIrJr2 s   r2r type[Any]dict[str, FieldInfo]cCs8t|r t|dr |jSt|r|jStd|jd)N__pydantic_fields__zError: z? is not subclass of BaseModel or pydantic.dataclasses.dataclass)r5rrr3rrgr`)rrIrIrJr s rrrr6rr:tuple[tuple[str, ...], bool]c Cs"g}d}t|j|jfs||g7}d}nkg}|j|jfD]7}|dur#qt|tr0||d}qt|trN|jD]}t|trG||d}q8||q8q||q|D](} tt| j d}|sf| n|}t | j dkrqdnd||<|s~|r~||qV|sdd|D}t t ||fS) NTFrrrrcSsg|]}|qSrIr)r alias_namerIrIrJr 2 sz$_get_alias_names..)rCrrrryrr'choicesr$rrrrrfromkeys) rrrrrr,new_alias_pathsrr alias_pathrIrIrJr s<          rr=cCst|ttfSr)rrr)r=rIrIrJrH6 srH)rCrD)TN)rrArrprorrCr)F)rryrrprCryr)rrrZrrCr)FFN) r,r-rrprYrprZrrCr-rm)rrrrrCrp)rrrCrp)TF) rrrrrrprrprCrp)rrrCr)rrrrrCr)rrrrrCr)rrrCr) rryrr6rr:rrprCr)r=rrCrp) __future__r _annotationsrr rrrLrOrabcrrrMargparserr r r r r collectionsrr dataclassesrrenumrpathlibrtextwraprrrrrrrrrrrrrr r!r"r#r$r%rJdotenvr&pydanticr'r(r)r*r+r,r-pydantic._internal._reprr. pydantic._internal._typing_extrar/r0r1pydantic._internal._utilsr2r3r4pydantic.dataclassesr5pydantic.fieldsr6 pydantic_corer7r8r9r:r;pydantic_settings.utilsr<rOrNrEpydantic._internal._dataclassesr?pydantic_settings.mainr@rArKrPrWryr[rYrZrr\r^rerrgrjrkrlrmrnr|r} CliSubCommandCliPositionalArgrpr~rrr# CliSuppressrrrrrrrr(rRrorZr`rlrrrrrrrr.rrrrrrrSr2rrrHrIrIrIrJs     @ $                  1m("<kht!. #    "      $