o _j,$@sddlZddlZddlmZddlmZddlmZddlm Z m Z m Z m Z ddl mZddlmZd d lmZd d lmZmZd d lmZd dlmZe rZddlmZddlmZdZGdddeZde de fddZdS)N)Iterator)reduce)Path) TYPE_CHECKINGAnyLiteralOptional) SettingsError)path_type_label)PydanticBaseSettingsSource) InitStateparse_env_vars)EnvSettingsSource)SecretsSettingsSource) BaseSettings)PathTypeicseZdZ          ddeeBdeddeddBdedBdedBd e dBd e dBd edBd edBd e dBde dBddffdd Z de ddfddZ ede dee fddZede dee e ffddZde fddZZS)NestedSecretsSettingsSourceNfile_secret_settings secrets_dirrsecrets_dir_missingokwarnerrorsecrets_dir_max_sizesecrets_case_sensitivesecrets_prefixsecrets_nested_delimitersecrets_nested_subdircase_sensitive env_prefix _init_statereturnc  st|d|} | j} tt|dd|| d_t|| dd_jdvr.tdjt|| dt_t|| d| | d d _ t|| d | | d d _ t|| d| d_ t|| dd _ j r|sw| dr{tdt j_ jdurg}ntjttfrjg}ntj}dd|D_jD]}|qtj| j j j d dd| dur| nt|dddd_tjsi_dStddfddjD}t|j jj_dS)N settings_clsrrrrz#invalid secrets_dir_missing value: rrr"Frr#r env_nested_delimiterr!zQOptions secrets_nested_delimiter and secrets_nested_subdir are mutually exclusivecSsg|] }t|qS)r expanduserresolve.0pr)r)m/home/dinkstrade/pdmp/venv/lib/python3.10/site-packages/pydantic_settings/sources/providers/nested_secrets.py jsz8NestedSecretsSettingsSource.__init__..Tr$)r"r#r(env_ignore_emptyenv_parse_enumsenv_parse_none_strr$cSstg||RSN)dictitems)d1d2r)r)r/sz6NestedSecretsSettingsSource.__init__..c3s|]}|VqdSr4) load_secretsr,selfr)r/ z7NestedSecretsSettingsSource.__init__..)getattr model_configfirst_not_nonegetrrr SECRETS_DIR_MAX_SIZErr"rr r!ossep isinstancerstrlist secrets_pathsvalidate_secrets_pathsuper__init__r3lenenv_varsrrr1)r<rrrrrrr r!r"r#r$r&confpathspathsecrets __class__r;r/rLs           z$NestedSecretsSettingsSource.__init__rQcCs|s,|jdkr dS|jdkrtjd|ddddS|jdkr*td|dt|s9tdt|td d | |D}||j krStd |j d dS) Nrrz directory "z" does not existr ) stacklevelrz.secrets_dir must reference a directory, not a css|]}|jVqdSr4)statst_size)r-fr)r)r/r=r>zDNestedSecretsSettingsSource.validate_secrets_path..zsecrets_dir size is above z bytes) existsrwarningsrr ValueErroris_dirr sum_iter_secret_filesr)r<rQsecrets_dir_sizer)r)r/rJs    z1NestedSecretsSettingsSource.validate_secrets_pathc#s8tdtdttffdd EdHdS)aYield the secret files contained in ``path``. ``path`` is expected to already be resolved. The directory tree is walked explicitly so that symbolic links are handled safely: * a file is only yielded if its real location stays within ``path``; entries that resolve outside of it (e.g. through a symbolic link) are skipped, so they neither contribute to the ``secrets_dir_max_size`` accounting nor get loaded; * each real directory is visited at most once, so cyclic or repeated symlinks cannot make the walk loop and inflate the size accounting or the number of loaded secrets. Because the size check and the loader share this iterator, they always see the same set of files. directoryr%c3s|}|vr dS|zt|}Wn ty"YdSw|D]'}|}|r@|ks8|jvr?|EdHq%|rL|jvrL|Vq%dSr4)r+addsortediterdirOSErrorr\parentsis_file)r` resolved_direntriesentryresolvedrQ seen_dirswalkr)r/rms(  z.walkN)setrrrQr)rkr/r^sz.NestedSecretsSettingsSource._iter_secret_filescsfdd|DS)Ncs(i|]}t||jddqS)zutf-8)encoding)rG relative_to read_textstripr,ror)r/ s(z.)r^)clsrQr)ror/r:sz(NestedSecretsSettingsSource.load_secretscCsd|jdS)Nz(NestedSecretsSettingsSource(secrets_dir=))rr;r)r)r/__repr__sz$NestedSecretsSettingsSource.__repr__) NNNNNNNNNN)__name__ __module__ __qualname__r rrrintboolrGrrLrrJ staticmethodrr^ classmethodr5r:rw __classcell__r)r)rSr/rsT     r/robjsr%cGsttdd|dS)NcSs|duSr4r))or)r)r/r9sz first_not_none..)nextfilter)rr)r)r/rAsrA) rDrZcollections.abcr functoolsrpathlibrtypingrrrr exceptionsr utilsr baser rrenvrrRrmainrsourcesrrCrrAr)r)r)r/s&          ?