o j4@sddlmZddlZddlZddlmZddlmZm Z m Z m Z ddl m Z mZmZddlmZmZddlmZddlmZdd lmZd d lmZd d lmZd dlmZmZd dlm Z m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'ertddl(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_annotation_is_complex_get_model_fields_literal_has_numeric_enum_union_has_strict_types_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) selfr r"r$r&r(r)r+r,r- __class__Z/home/kuhnn/.local/lib/python3.10/site-packages/pydantic_settings/sources/providers/env.pyr2*s    zEnvSettingsSource.__init__Mapping[str, str | None]cCsttj|j|j|jSN)rosenvironr"r+r,r:r=r=r>r8Jsz 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_infor9r4)r:rDrEenv_val field_keyenv_namevalue_is_complexr=r=r>get_field_valueMs   z!EnvSettingsSource.get_field_valuevaluerrLboolc 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_varsr9decode_complex_value ValueErrordictr _coerce_env_val_strict) r:rErDrNrL is_complexallow_parse_failureenum_val env_val_builter=r=r>prepare_field_valuebs4     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_complexrrrQrmetadata)r:rDrYr=r=r>rPs 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)rRrrQr next_fieldrrrVr r ritemsrHlower) r:rDrbr"rQtype_ type_has_keyfieldsrEf_rKr=r=r>res.       zEnvSettingsSource.next_fieldr9dict[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|rt|tr|\}}jrt|j|}|dur|n|}n|rt|g}d}nd\}}|rzt|tr|nd}| ||}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().0rlrKrCr=r> sz6EnvSettingsSource.explode_env_vars..c3s|] }|r|VqdSr@) startswith)rnprefix)rKr=r> sz5EnvSettingsSource.explode_env_vars..NT)TT)r(rQrVrrrHrfnext StopIterationr6splitr5rer"rR setdefaultrrPr-rrrTrUrrW)r:rErDr9annis_dictprefixesresultrIrqenv_name_without_prefixkeyslast_keyenv_var target_fieldrbrXallow_json_failurerZ field_infor\r=)rKr:r>rSsd           z"EnvSettingsSource.explode_env_varsc Csz}|jd}|st|trtt|jot|jpt|j}|rrt|t rut|trx||j kr3|WSt |jt fdds{z t |j|WWStyqzt|}Wn ttjfy_wt|t spt |j|YWSwW|SW|SW|SW|StyY|Sw)a Coerce environment string values based on field annotation if model config is `strict=True` or if the field annotation contains strict-annotated types (e.g. Optional[StrictBool]). Args: field: The field. value: The value to coerce. Returns: The coerced value if successful, otherwise the original value. strictT) is_instance)r3r4rRrrrrQrrrFr,rrr validate_pythonr jsonloadsrUJSONDecodeError)r:rDrN should_coercedecodedr=r=r>rWsF     z(EnvSettingsSource._coerce_env_val_strictcCs|jjd|jd|jdS)Nz(env_nested_delimiter=z, env_prefix_len=))r<__name__r(r7rCr=r=r>__repr__BszEnvSettingsSource.__repr__)NNNNNNNN)r r!r"r#r$r%r&r'r(r%r)r*r+r#r,r%r-r#r.r/)r.r?)rDrrErFr.rG) rErFrDrrNrrLrOr.r)rDrr.r^r@)rDrarbrFr"r#r.rc)rErFrDrr9r?r.rm)rDrcrNrr.r)r.rF)r __module__ __qualname____doc__r2r8rMr]rPrerSrWr __classcell__r=r=r;r>r%s(   . 8 D&r), __future__r _annotationsrrAcollections.abcrtypingrrrrpydanticrr r pydantic._internal._utilsr r pydantic.dataclassesr pydantic.fieldsrtyping_inspection.introspectionrutilsrbasertypesrrrrrrrrrrpydantic_settings.mainrr__all__r=r=r=r>s(       (  &