o h`@sUddlmZddlZddlZddlZddlmZmZddl m Z ddl m Z ddl mZddlmZmZmZmZmZmZmZmZddlmZdd lmZmZmZmZdd lm Z dd l!m"Z"m#Z#dd l$m%Z%dd l&m'Z'm(Z(ddl)m*Z*er}ddl+m,Z,eee-eeee-feeee-fdffZ.edZ/de0d<Gddde1Z2GdddeZ3Gddde3Z4Gddde3Z5Gddde5Z6Gddde5Z7Gd d!d!e7Z8dd"d#d6d,d-Z9d7d2d3Z:d8d4d5Z;dS)9) annotationsN)ABCabstractmethod)deque) is_dataclass)Path) TYPE_CHECKINGAnyListMappingSequenceTupleUnioncast) dotenv_values) AliasChoices AliasPath BaseModelJson)origin_is_union) deep_updatelenient_issubclass) FieldInfo)get_args get_origin)path_type_label) BaseSettings. DotenvTypeENV_FILE_SENTINELc@s eZdZdS) SettingsErrorN)__name__ __module__ __qualname__r$r$O/var/www/html/ai/venv/lib/python3.10/site-packages/pydantic_settings/sources.pyr !sr c@sTeZdZdZdddZedd d ZdddZdddZd ddZ ed!ddZ dS)"PydanticBaseSettingsSourcezi Abstract base class for settings sources, every settings source classes should inherit from it. settings_clstype[BaseSettings]cCs||_|j|_dSN)r' model_configconfig)selfr'r$r$r%__init__*s z#PydanticBaseSettingsSource.__init__fieldr field_namestrreturntuple[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. Nr$r,r.r/r$r$r%get_field_value.sz*PydanticBaseSettingsSource.get_field_valueboolcCst|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 annotationmetadata)r,r.r$r$r%field_is_complex>s z+PydanticBaseSettingsSource.field_is_complexvaluer value_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)r:decode_complex_value)r,r/r.r;r<r$r$r%prepare_field_valueJs z.PydanticBaseSettingsSource.prepare_field_valuecCs t|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 )jsonloads)r,r/r.r;r$r$r%r=[s z/PydanticBaseSettingsSource.decode_complex_valuedict[str, Any]cCsdSr)r$r,r$r$r%__call__iz#PydanticBaseSettingsSource.__call__N)r'r(r.rr/r0r1r2)r.rr1r6 r/r0r.rr;r r<r6r1r )r/r0r.rr;r r1r r1rA) r!r"r#__doc__r-rr5r:r>r=rCr$r$r$r%r&%s    r&cs@eZdZdZdfdd ZdddZdddZdddZZS)InitSettingsSourcezX Source class for loading values provided during settings class initialization. r'r( init_kwargsrAcs||_t|dSr))rJsuperr-)r,r'rJ __class__r$r%r-sszInitSettingsSource.__init__r.rr/r0r1r2cCr3)N)NrFr$r4r$r$r%r5wrDz"InitSettingsSource.get_field_valuecCs|jSr)rJrBr$r$r%rC{szInitSettingsSource.__call__cCd|jdS)NzInitSettingsSource(init_kwargs=)rNrBr$r$r%__repr__~zInitSettingsSource.__repr__)r'r(rJrArErGr1r0) r!r"r#rHr-r5rCrQ __classcell__r$r$rLr%rIns   rIcsJeZdZ ddfd d ZdddZdddZd ddZd!ddZZS)"PydanticBaseEnvSettingsSourceNr'r(case_sensitive bool | None env_prefix str | Noner1NonecsNt||dur |n|jdd|_|dur||_dS|jdd|_dS)NrVFrXr)rKr-r+getrVrX)r,r'rVrXrLr$r%r-s &z&PydanticBaseEnvSettingsSource.__init__r;r0cCs|js|S|Sr))rVlower)r,r;r$r$r%_apply_case_sensitivesz3PydanticBaseEnvSettingsSource._apply_case_sensitiver.rr/list[tuple[str, str, bool]]cCsg}t|jttfr|j}n|j}|rft|trY|D]:}t|tr7||||t |dkr2dndfqt|trVt t|d}||||t |dkrRdndfq|S||||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. TFr) isinstancevalidation_aliasrrconvert_to_aliaseslistr0appendr]lenrrX)r,r.r/ field_infov_aliasalias first_argr$r$r%_extract_field_infos(   ( z1PydanticBaseEnvSettingsSource._extract_field_info field_valuesrAc Csi}|D]M\}}d}|jrt|jds|||<q|jjD]\}}|js3||kr3|}nq |s;|||<qt|jtrOt|t rO| ||||<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'}}} N model_fields) itemsr8hasattrrlrar\rrr`dict'_replace_field_names_case_insensitively) r,r.rkvaluesnamer;sub_model_fieldsub_model_field_namefr$r$r%rps$ zEPydanticBaseEnvSettingsSource._replace_field_names_case_insensitivelyc Csi}|jjD]n\}}z |||\}}}Wnty1}ztd|d|jjd|d}~wwz |||||}Wnt yV}ztd|d|jjd|d}~ww|durv|j srt |j t rrt|trr|||||<q|||<q|S)Nzerror getting value for field "z" from source ""zerror parsing value for field ")r'rlrmr5 Exceptionr rMr!r> ValueErrorrVrr8rr`rorp)r,datar/r. field_value field_keyr<er$r$r%rCsB z&PydanticBaseEnvSettingsSource.__call__)NN)r'r(rVrWrXrYr1rZ)r;r0r1r0)r.rr/r0r1r^)r.rrkrAr1rArG) r!r"r#r-r]rjrprCrTr$r$rLr%rUs   (;rUcsZeZdZdZ   d"d#fd d Zd$fdd Zed%ddZd&ddZd'd d!Z Z S)(SecretsSettingsSourcezE Source class for loading settings values from secret files. Nr'r( secrets_dirstr | Path | NonerVrWrXrYr1rZcs4t||||dur||_dS|jd|_dS)Nr~)rKr-r+r[r~)r,r'r~rVrXrLr$r%r-s$zSecretsSettingsSource.__init__rAcsli}|jdur |St|j|_|js"td|jd|S|js1tdt |jt S)z4 Build fields from "secrets" files. Nz directory "z" does not existz.secrets_dir must reference a directory, not a ) r~r expanduser secrets_pathexistswarningswarnis_dirr rrKrC)r,secretsrLr$r%rCs    zSecretsSettingsSource.__call__dir_pathr file_namer0r6 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)iterdirrrr\)clsrrrVrur$r$r%find_case_path2s z$SecretsSettingsSource.find_case_pathr.rr/r2cCsx|||D]0\}}}||j||j}|sq|r&|||fStjd|dt |dddqd||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. zattempted to load secret file "z" but found a z instead.) stacklevelN) rjrrrVis_file read_textstriprrr)r,r.r/r{env_namer<pathr$r$r%r5Fs  z%SecretsSettingsSource.get_field_valuecCrO)Nz"SecretsSettingsSource(secrets_dir=rP)r~rBr$r$r%rQcrRzSecretsSettingsSource.__repr__NNN) r'r(r~rrVrWrXrYr1rZrG)rrrr0rVr6r1rrErS) r!r"r#rHr-rC classmethodrr5rQrTr$r$rLr%r}s   r}cs~eZdZdZ   d1d2fd d Zd3ddZd4ddZd5ddZd6d"d#Zd7d%d&Z e d8d)d*Z d9d-d.Z d:d/d0Z ZS);EnvSettingsSourcezN Source class for loading settings values from environment variables. Nr'r(rVrWrXrYenv_nested_delimiterr1rZcsDt||||dur|n|jd|_t|j|_||_ dS)Nr) rKr-r+r[rrerXenv_prefix_len_load_env_varsenv_vars)r,r'rVrXrrLr$r%r-ls  zEnvSettingsSource.__init__Mapping[str, str | None]cCs |jrtjSddtjDS)NcSi|] \}}||qSr$r\.0kvr$r$r% ~z4EnvSettingsSource._load_env_vars..)rVosenvironrmrBr$r$r%r{sz EnvSettingsSource._load_env_varsr.rr/r0r2cCs>d}|||D]\}}}|j|}|durnq|||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. N)rjrr[)r,r.r/env_valr{rr<r$r$r%r5s   z!EnvSettingsSource.get_field_valuer;r r<r6c Cs||\}}|s |rO|dur||||j}|r|SdSz ||||}Wnty<}z |s2|WYd}~nd}~wwt|trMt|||||jS|S|durU|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_complexexplode_env_varsrr=rxr`ror) r,r/r.r;r< is_complexallow_parse_failure env_val_builtr|r$r$r%r>s*  z%EnvSettingsSource.prepare_field_valuer8type[Any] | Noner9 list[Any]cstfddt|DS)Nc3s|]}t|VqdSr))r7)rargr9r$r% z6EnvSettingsSource._union_is_complex..)anyr)r,r8r9r$rr%_union_is_complexsz#EnvSettingsSource._union_is_complextuple[bool, bool]cCsD||r d}d|fStt|jr ||j|jr d}d|fSdS)za Find out if a field is complex, and if so whether JSON errors should be ignored FT)FF)r:rrr8rr9)r,r.rr$r$r%rs z#EnvSettingsSource._field_is_complexFieldInfo | NonekeycCsF|r tt|jr dS|jr!t|jdr!|jj|r!|jj|SdS)a 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). Returns: Field if it finds the next field otherwise `None`. Nrl)rrr8rnrlr[)r.rr$r$r% next_fields   zEnvSettingsSource.next_fieldrrAc sfdd||D}i}|D]m\}tfdd|Ds#qjd}|j^}} } |} |} | D]} | | } | | i} q:| | } | r||r|| \}}|r|z  | | |}Wnt y{}z |sq|WYd}~nd}~ww|| | <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 dictionaty contains extracted values from nested env values. cs g|] \}}}|jqSr$)r)r_rrBr$r% sz6EnvSettingsSource.explode_env_vars..c3s|]}|VqdSr)) startswith)rprefix)rr$r%rrz5EnvSettingsSource.explode_env_vars..N) rjrmrrsplitrr setdefaultrr=rx)r,r/r.rprefixesresultrenv_name_without_prefixrkeyslast_keyenv_var target_fieldrrallow_json_failurer|r$)rr,r%rs8      z"EnvSettingsSource.explode_env_varscCsd|jd|jdS)Nz'EnvSettingsSource(env_nested_delimiter=, env_prefix_len=rP)rrrBr$r$r%rQ!s zEnvSettingsSource.__repr__r) r'r(rVrWrXrYrrYr1rZr1rrErFr8rr9rr1r6)r.rr1r)r.rrr0r1r)r/r0r.rrrr1rArS)r!r"r#rHr-rr5r>rr staticmethodrrrQrTr$r$rLr%rgs    (   %.rcsZeZdZdZeddddfdfdd ZdddZdddZd fdd Zd!ddZ Z S)"DotEnvSettingsSourcezB Source class for loading settings values from env files. Nr'r(env_fileDotenvType | Noneenv_file_encodingrYrVrWrXrr1rZcsJ|tkr|n|jd|_|dur|n|jd|_t||||dS)Nrr)rr*r[rrrKr-)r,r'rrrVrXrrLr$r%r--s zDotEnvSettingsSource.__init__rcCs ||jSr))_read_env_filesrVrBr$r$r%r<s z#DotEnvSettingsSource._load_env_varsr6cCsd|j}|dur iSt|ttjfr|g}i}|D]}t|}|r/|t ||j |dq|S)NencodingrV) rr`r0rPathLikerrrupdate read_env_filer)r,rV env_files dotenv_varsrenv_pathr$r$r%r?s z$DotEnvSettingsSource._read_env_filesrAcst}g}|jsdd|D}|jD]-\}}||jrE|durE||jd}| |j ^}}|r;||vsA|sE||vrE|||<q|S)NcSsg|]}|qSr$r)rxr$r$r%rVsz1DotEnvSettingsSource.__call__..) rKrCrVrrrmrrXrrr)r,rydata_lower_keysr env_valuer first_keyrrLr$r%rCQs  zDotEnvSettingsSource.__call__r0c Cs&d|jd|jd|jd|jd S)NzDotEnvSettingsSource(env_file=z, env_file_encoding=z, env_nested_delimiter=rrP)rrrrrBr$r$r%rQfs zDotEnvSettingsSource.__repr__)r'r(rrrrYrVrWrXrYrrYr1rZr)rVr6r1rrGrS) r!r"r#rHrr-rrrCrQrTr$r$rLr%r(s  rFr file_pathrrrYrVr6r1rcCs*t||pdd}|sdd|DS|S)Nutf8)rcSrr$rrr$r$r%rrrz!read_env_file..)rrm)rrrV file_varsr$r$r%rmsrr8rr9rcCsBtdd|Dr dSt|}t|p t|p t|dp t|dS)Ncss|]}t|tVqdSr))r`r)rmdr$r$r%rxrz)_annotation_is_complex..F__pydantic_core_schema____get_pydantic_core_schema__)rr_annotation_is_complex_innerrn)r8r9originr$r$r%r7wsr7c Cs2t|ttfr dSt|ttttttt fpt |S)NF) rr0bytesrr r tupleset frozensetrr)r8r$r$r%rs r)rrrrYrVr6r1rr)r8rr1r6)< __future__r _annotationsr?rrabcrr collectionsr dataclassesrpathlibrtypingrr r r r r rrdotenvrpydanticrrrr pydantic._internal._typing_extrarpydantic._internal._utilsrrpydantic.fieldsrtyping_extensionsrrpydantic_settings.utilsrpydantic_settings.mainrr0rr__annotations__rxr r&rIrUr}rrrr7rr$r$r$r%sB   (     ,IXBF