o 찓i-/@sddlmZddlZddlmZddlmZmZm Z m Z ddl m Z m Z mZddlmZmZddlmZddlmZdd lmZd d lmZd d lmZd dlmZmZd dlmZm Z m!Z!m"Z"m#Z#erjddl$m%Z%GdddeZ&dgZ'dS)) annotationsN)Mapping) TYPE_CHECKINGAnyget_args get_origin)Json TypeAdapterValidationError) deep_updateis_model_class)is_pydantic_dataclass) FieldInfo)is_union_origin)_lenient_issubclass)PydanticBaseEnvSettingsSource) EnvNoneTypeEnvPrefixTarget)_annotation_contains_types_annotation_enum_name_to_val_get_model_fields_union_is_complexparse_env_vars) BaseSettingscseZdZdZ        d5d6fdd Zd7ddZd8ddZd9d#d$Zd:d&d'Z d;dd1d2Z d?d3d4Z Z S)@EnvSettingsSourcezN Source class for loading settings values from environment variables. N settings_clstype[BaseSettings]case_sensitive bool | None env_prefix str | Noneenv_prefix_targetEnvPrefixTarget | Noneenv_nested_delimiterenv_nested_max_split int | Noneenv_ignore_emptyenv_parse_none_strenv_parse_enumsreturnNonec svt||||||| |dur|n|jd|_|dur|n|jd|_|jp*dd|_t|j|_ | |_ dS)Nr%r&r) super__init__configgetr%r&maxsplitlenr!env_prefix_len_load_env_varsenv_vars) selfrrr!r#r%r&r(r)r* __class__[/home/ubuntu/.local/lib/python3.10/site-packages/pydantic_settings/sources/providers/env.pyr/&s    zEnvSettingsSource.__init__Mapping[str, str | None]cCsttj|j|j|jSN)rosenvironrr(r)r7r:r:r;r5Fsz EnvSettingsSource._load_env_varsfieldr field_namestrtuple[Any, str, bool]cCs>d}|||D]\}}}|j|}|durnq|||fS)ah 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 that contains the value (`None` if not found), key, and a flag to determine whether value is complex. N)_extract_field_infor6r1)r7rArBenv_val field_keyenv_namevalue_is_complexr:r:r;get_field_valueIs   z!EnvSettingsSource.get_field_valuevaluerrIboolc Cs||\}}|jrt|j|}|dur|n|}|s|rgt|tr#|S|dur5||||j}|r3|SdSz ||||}Wnt yT} z |sJ| WYd} ~ nd} ~ wwt|t ret |||||jS|S|durq| ||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*r annotation isinstancerexplode_env_varsr6decode_complex_value ValueErrordictr _coerce_env_val_strict) r7rBrArKrI is_complexallow_parse_failureenum_val env_val_builter:r:r;prepare_field_value^s4     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)field_is_complexrrrNrmetadata)r7rArVr:r:r;rMs z#EnvSettingsSource._field_is_complexFieldInfo | Any | NonekeyFieldInfo | Nonec Cs|sdSt|tr |jn|}t|D]}||||}|r!|Sqtt|tr/t|dSt|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)rOrrNr next_fieldrrrSr r ritemsrElower) r7rAr_rrNtype_ type_has_keyfieldsrBf_rHr:r:r;rbs.       zEnvSettingsSource.next_fieldr6dict[str, Any]c sjsiS|j}|tuptt|t}fdd||D}i}|D]\}z tfdd|D} Wn ty@Yq&wt | d} | jj ^} } |} |}| D]} ||j }t| tro| |i} qZ || j }|s||r|r|r|\}}jrt|j|}|dur|n|}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|] \}}}|jqSr:)r%).0rirHr@r:r; sz6EnvSettingsSource.explode_env_vars..c3s|] }|r|VqdSr=) startswith)rkprefix)rHr:r; sz5EnvSettingsSource.explode_env_vars..N)TT)r%rNrSrrrErcnext StopIterationr3splitr2rbrrO setdefaultrMr*rrQrRrrT)r7rBrAr6annis_dictprefixesresultrFrnenv_name_without_prefixkeyslast_keyenv_var target_fieldr_rUallow_json_failurerWrYr:)rHr7r;rPs\         z"EnvSettingsSource.explode_env_varscCsz5|jdr*t|tr-|dur0||jkr|WSt|jtfdds3t|j |WSW|SW|SW|SW|St y?Y|Sw)a  Coerce environment string values based on field annotation if model config is `strict=True`. Args: field: The field. value: The value to coerce. Returns: The coerced value if successful, otherwise the original value. strictNT) is_instance) r0r1rOrCr)rrNrr validate_pythonr )r7rArKr:r:r;rTs$   z(EnvSettingsSource._coerce_env_val_strictcCs|jjd|jd|jdS)Nz(env_nested_delimiter=z, env_prefix_len=))r9__name__r%r4r@r:r:r;__repr__)szEnvSettingsSource.__repr__)NNNNNNNN)rrrr r!r"r#r$r%r"r&r'r(r r)r"r*r r+r,)r+r<)rArrBrCr+rD) rBrCrArrKrrIrLr+r)rArr+r[r=)rAr^r_rCrr r+r`)rBrCrArr6r<r+rj)rAr`rKrr+r)r+rC)r __module__ __qualname____doc__r/r5rJrZrMrbrPrTr __classcell__r:r:r8r;r!s(   . 8 ?r)( __future__r _annotationsr>collections.abcrtypingrrrrpydanticrr r pydantic._internal._utilsr r pydantic.dataclassesr pydantic.fieldsrtyping_inspection.introspectionrutilsrbasertypesrrrrrrrpydantic_settings.mainrr__all__r:r:r:r;s&