bivdZddlmZddlZddlZddlmZddlm Z m Z ddl m Z ddl mZdd lmZd d lmZd d lmZmZe rdd lmZGddeZy)zSecrets file settings source.) annotationsN)Path) TYPE_CHECKINGAny) FieldInfo)path_type_label) SettingsError)PydanticBaseEnvSettingsSource)EnvPrefixTargetPathType) BaseSettingsceZdZdZ d dfd Zd fd Zed dZd dZd dZ xZ S) SecretsSettingsSourcezE Source class for loading settings values from secret files. c t |||||||||||_y|jjd|_y)N secrets_dir)super__init__configgetr) self settings_clsrcase_sensitive env_prefixenv_prefix_targetenv_ignore_emptyenv_parse_none_strenv_parse_enums __class__s s/home/jang/Projects/ai-researcher/.venv/lib/python3.12/site-packages/pydantic_settings/sources/providers/secrets.pyrzSecretsSettingsSource.__init__sK         +6*A;t{{WdGec`i}|j|St|jttjfr |jgn |j}|Dcgc]}t |j }}g|_|D]G}|jstjd|d-|jj|It|js|S|jD])}|jrtdt|t |EScc}w)z4 Build fields from "secrets" files. z directory "z" does not existz.secrets_dir must reference a directory, not a )r isinstancestrosPathLiker expanduser secrets_pathsexistswarningswarnappendlenis_dirr rr__call__)rsecrets secrets_dirspr)pathr s r!r0zSecretsSettingsSource.__call__4s*,    #N-78H8H3PRP[P[J\-](()cgcscs 7CD!a++-D D! 0D;;= D61ABC""))$/  0 4%%&N&& nD;;=#&TUdeiUjTk$lmm nw!!!Es D+c|jD]G}|j|k(r|cS|r|jj|jk(sE|cSy)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)iterdirnamelower)clsdir_path file_namerfs r!find_case_pathz$SecretsSettingsSource.find_case_pathPsR!!# Avv"# )//:K(K   r"c n|j||D]\}}}t|jD]~}|j|||j}|s#|j r%|j j||fccStjd|dt|dddfS)al 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 that contains the value (`None` if the file does not exist), key, and a flag to determine whether value is complex. zattempted to load secret file "z" but found a z instead.) stacklevelN) _extract_field_inforeversedr)r=ris_file read_textstripr+r,r)rfield field_name field_keyenv_namevalue_is_complex secrets_pathr4s r!get_field_valuez%SecretsSettingsSource.get_field_valueds6:5M5MeU_5`  1Ix!1 ();); <  **<4CVCVW<<>>>+113Y@PPPMM9$~o^bNcMddmn#$   Y 000r"cN|jjd|jdS)Nz (secrets_dir=))r __name__r)rs r!__repr__zSecretsSettingsSource.__repr__s'..))*-8H8H7K1MMr")NNNNNNN)rztype[BaseSettings]rzPathType | Noner bool | Noner str | NonerzEnvPrefixTarget | NonerrQrrRrrQreturnNone)rSzdict[str, Any])r:rr;r%rboolrSz Path | None)rFrrGr%rSztuple[Any, str, bool])rSr%) rO __module__ __qualname____doc__rr0 classmethodr=rLrP __classcell__)r s@r!rrs(,&*!%48(,)-'+f(f%f$ f  f 2 f&f'f%f f,"8&1>Nr"r)rX __future__r _annotationsr&r+pathlibrtypingrrpydantic.fieldsrpydantic_settings.utilsr exceptionsr baser typesr rpydantic_settings.mainrrr"r!rfsA#2  &3'0-3kN9kNr"