U ={fD@spUddlmZddlZddlZddlZddlZddlZddlZddl Z ddl m Z m Z ddl mZmZmZmZmZddlmZddlmZddlmZddlmZdd lmZdd lmZmZmZm Z m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(ddl)Z)dd l*m+Z+dd l,m-Z-m.Z.m/Z/m0Z0dd l1m2Z2ddl3m4Z4m5Z5m6Z6ddl7m8Z8m9Z9m:Z:ddl;mZ>ddl?m@Z@ddl)mAZAmBZBmCZCmDZDddlEmFZFerejGdkrddlHaHndaHddlIaIddlJaJddlKmLZLn daJdaHdaIddddZMddddZNe&eeOe!e&eeOfe$e&eeOfdffZPe&eeOe!e&eeOfe$e&eeOfdffZQedZRdeSd <edZTd!eSd"<Gd#d$d$ZUGd%d&d&ZVGd'd(d(eZWe%d)ZXeAe&eXdfeUfZYeAeXeVfZZGd*d+d+eOZ[Gd,d-d-e\Z]Gd.d/d/e Z^Gd0d1d1e^Z_Gd2d3d3e^Z`Gd4d5d5e`ZaGd6d7d7e`ZbGd8d9d9ebZcGd:d;d;ebe eXZdGdd?d?e_eeZfGd@dAdAe_eeZgGdBdCdCegZhGdDdEdEe_eeZidldGdHdGdIdJdKZjdmdLdLdMdNdOdPZkdndQdHdHdLdQdRdSdTZlddFdFddUdVdLdHdHdLdQdWdXdYZmdZd[dHd\d]d^ZndZdHd_d`daZodZd[dHd\dbdcZpdodZdedHdHdHdfdgdhZqdidid_djdkZrdS)p) annotationsN)ABCabstractmethod)SUPPRESSArgumentParser HelpFormatter Namespace_SubParsersAction)deque) is_dataclass)Enum)Path) FunctionType) TYPE_CHECKINGAnyCallableGenericListMappingSequenceTupleTypeVarUnioncastoverload) dotenv_values) AliasChoices AliasPath BaseModelJson)Representation) WithArgsTypesorigin_is_union typing_base) deep_updateis_model_classlenient_issubclass)is_pydantic_dataclass) FieldInfo)PydanticUndefined) Annotated_AnnotatedAliasget_args get_origin)path_type_label ) BaseSettingsNonereturnc CsJtdk r dSz ddlaWn,tk rD}ztd|W5d}~XYnXdS)NrzBPyYAML is not installed, run `pip install pydantic-settings[yaml]`)yaml ImportErrorer:X/var/www/html/Darija-Ai-API/env/lib/python3.8/site-packages/pydantic_settings/sources.py import_yaml<s  r<c CsjtjdkrRtdk rdSz ddlaWqftk rN}ztd|W5d}~XYqfXntdk r^dSddladS)Nr/rzAtomli is not installed, run `pip install pydantic-settings[toml]`)sys version_infotomlir7tomllibr8r:r:r; import_tomlFs  rA.PathType DEFAULT_PATH DotenvTypeENV_FILE_SENTINELc@s eZdZdS)_CliSubCommandN__name__ __module__ __qualname__r:r:r:r;rG`srGc@s eZdZdS)_CliPositionalArgNrHr:r:r:r;rLdsrLc@s eZdZdS)_CliInternalArgParserNrHr:r:r:r;rMhsrMTc@s eZdZdS) EnvNoneTypeNrHr:r:r:r;rOqsrOc@s eZdZdS) SettingsErrorNrHr:r:r:r;rPusrPc@seZdZdZddddZedddd d d Zdd d ddZdddd ddddZdddddddZ eddddZ dS)PydanticBaseSettingsSourcezi Abstract base class for settings sources, every settings source classes should inherit from it. type[BaseSettings]) settings_clscCs||_|j|_dSN)rS model_configconfig)selfrSr:r:r;__init__~sz#PydanticBaseSettingsSource.__init__r(strtuple[Any, str, bool]field field_namer5cCsdS)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:rWr\r]r:r:r;get_field_valuesz*PydanticBaseSettingsSource.get_field_valueboolr\r5cCst|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)rWr\r:r:r;field_is_complexs z+PydanticBaseSettingsSource.field_is_complexrr]r\valuevalue_is_complexr5cCs(|dk r$||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)redecode_complex_value)rWr]r\rgrhr:r:r;prepare_field_values z.PydanticBaseSettingsSource.prepare_field_value)r]r\rgr5cCs 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)rWr]r\rgr:r:r;ris z/PydanticBaseSettingsSource.decode_complex_valuedict[str, Any]r4cCsdSrTr:rWr:r:r;__call__sz#PydanticBaseSettingsSource.__call__N) rIrJrK__doc__rXrr_rerjriror:r:r:r;rQys rQcsVeZdZdZdddfdd Zddd d d d Zdd ddZdd ddZZS)InitSettingsSourcezX Source class for loading values provided during settings class initialization. rRrm)rS init_kwargscs||_t|dSrT)rrsuperrX)rWrSrr __class__r:r;rXszInitSettingsSource.__init__r(rYrZr[cCsdS)N)NrBFr:r^r:r:r;r_sz"InitSettingsSource.get_field_valuer4cCs|jSrTrrrnr:r:r;roszInitSettingsSource.__call__cCsd|jdS)NzInitSettingsSource(init_kwargs=)rvrnr:r:r;__repr__szInitSettingsSource.__repr__) rIrJrKrprXr_rorx __classcell__r:r:rtr;rqs rqc seZdZdddddddddfdd Zd d d d d Zd d ddddZd dddddZdddddZddddZZ S)PydanticBaseEnvSettingsSourceNrR bool | None str | Noner3)rScase_sensitive env_prefixenv_ignore_emptyenv_parse_none_strenv_parse_enumsr5cst||dk r|n |jdd|_|dk r4|n |jdd|_|dk rP|n |jdd|_|dk rl|n |jd|_|dk r|n |jd|_dS)Nr}Fr~rBrrr) rsrXrVgetr}r~rrr)rWrSr}r~rrrrtr:r;rXs z&PydanticBaseEnvSettingsSource.__init__rY)rgr5cCs|js|S|SrT)r}lower)rWrgr:r:r;_apply_case_sensitivesz3PydanticBaseEnvSettingsSource._apply_case_sensitiver(zlist[tuple[str, str, bool]]r[cCs&g}t|jttfr |j}n|j}|rt|tr|D]t}t|trn||||t |dkrddndfq8t|tr8t t|d}||||t |dkrdndfq8n||||dfnZt t |j rt|j |jr||||j|dfn||||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_aliasrrZconvert_to_aliaseslistrYappendrlenrr"r-rc_union_is_complexrdr~)rWr\r] field_infoZv_aliasaliasZ first_argr:r:r;_extract_field_infos&   (  z1PydanticBaseEnvSettingsSource._extract_field_inform)r\ field_valuesr5c Csi}|D]\}}d}|j}t|}tt|jrjt|dkrjtd|krj|D]} | tdkrP| }qjqP|rxt|ds|||<q |jD]&\} } | j s| | kr| }qq|s|||<q t |jt rt |tr||||| <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)itemsrcr,r"r-rtypehasattrrrrr&rrdict'_replace_field_names_case_insensitively) rWr\rvaluesnamergZsub_model_fieldrcargsargZsub_model_field_namefr:r:r;rs0&  zEPydanticBaseEnvSettingsSource._replace_field_names_case_insensitively) field_valuer5cCsJi}|D]8\}}t|tstk rn}z td|d|jjd|W5d}~XYnXz|||||}Wn>t k r}z td|d|jjd|W5d}~XYnX|dk r|j dk rt |t r| |}nt |trd}|js t |t r |||||<q|||<q|S)Nzerror getting value for field "z" from source ""zerror parsing value for field ")rSrrr_ ExceptionrPrurIrj ValueErrorrrrrrOr}r)rWdatar]r\r field_keyrhr9r:r:r;roks>     z&PydanticBaseEnvSettingsSource.__call__)NNNNN) rIrJrKrXrrrrroryr:r:rtr;rzs *Erzc seZdZdZddddddddddfd d Zd d fd d ZedddddddZddddddZdd ddZ Z S)SecretsSettingsSourcezE Source class for loading settings values from secret files. NrRzstr | Path | Noner{r|r3)rS secrets_dirr}r~rrrr5cs4t|||||||dk r"|n |jd|_dS)Nr)rsrXrVrr)rWrSrr}r~rrrrtr:r;rXs zSecretsSettingsSource.__init__rmr4csli}|jdkr|St|j|_|jsDtd|jd|S|jsbtdt |jt S)z4 Build fields from "secrets" files. Nz directory "z" does not existz.secrets_dir must reference a directory, not a ) rr expanduser secrets_pathexistswarningswarnis_dirrPr.rsro)rWZsecretsrtr:r;ros   zSecretsSettingsSource.__call__r rYr` Path | None)dir_path file_namer}r5cCsB|D]4}|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)iterdirrr)clsrrr}rr:r:r;find_case_paths   z$SecretsSettingsSource.find_case_pathr(rZr[cCsx|||D]`\}}}||j||j}|s.q |rL|||fStjd|dt |dddq d||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) rrrr}is_file read_textstriprrr.)rWr\r]renv_namerhpathr:r:r;r_s z%SecretsSettingsSource.get_field_valuecCsd|jdS)Nz"SecretsSettingsSource(secrets_dir=rw)rrnr:r:r;rxszSecretsSettingsSource.__repr__)NNNNNN) rIrJrKrprXro classmethodrr_rxryr:r:rtr;rs"rc seZdZdZd(dddddddddfdd Zd d d d ZddddddZddddddddZdddddZe d)dddddd d!Z ddd d"d#d$d%Z dd d&d'Z Z S)*EnvSettingsSourcezN Source class for loading settings values from environment variables. NrRr{r|r3)rSr}r~env_nested_delimiterrrrr5csJt|||||||dk r"|n |jd|_t|j|_||_ dS)Nr) rsrXrVrrrr~env_prefix_len_load_env_varsenv_vars)rWrSr}r~rrrrrtr:r;rXs  zEnvSettingsSource.__init__Mapping[str, str | None]r4cCsttj|j|j|jSrT)parse_env_varsosenvironr}rrrnr:r:r;rsz EnvSettingsSource._load_env_varsr(rYrZr[cCs>d}|||D]"\}}}|j|}|dk rq4q|||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)rrr)rWr\r]env_valrrrhr:r:r;r_s  z!EnvSettingsSource.get_field_valuerr`rfc Cs||\}}|jrBt|jtrB|tdd|jDkrB|j|}|sJ|rt|trX|S|dkrz||||j }|r|Sqz| |||}Wn*t k r}z |s|W5d}~XYnXt|t rt |||||j S|Sn |dk r|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. css|] }|jVqdSrTr.0valr:r:r; 0sz8EnvSettingsSource.prepare_field_value..N)_field_is_complexrr&rcr tuplerrOexplode_env_varsrrirrr$) rWr]r\rgrh is_complexallow_parse_failureZ env_val_builtr9r:r:r;rjs*   z%EnvSettingsSource.prepare_field_valueztuple[bool, bool]racCs>||rd}n&tt|jr2t|j|jr2d}ndSd|fS)za Find out if a field is complex, and if so whether JSON errors should be ignored FT)FF)rer"r-rcrrd)rWr\rr:r:r;rKs  z#EnvSettingsSource._field_is_complexzFieldInfo | Any | NonezFieldInfo | None)r\rr}r5c Cs|sdSt|tr|jn|}tt|s2t|tr\t|D]}t|||}|r:|Sq:nvt |slt |rt |rz|j n t t |j}|dks|r||r||S|s|D] \}}||kr|SqdS)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)rr(rcr"r-r!r,r next_fieldr%r'__pydantic_fields__rrrrrr) r\rr}rctype_Z type_has_keyfieldsr]rr:r:r;r[s(    zEnvSettingsSource.next_fieldrm)r]r\rr5c sptt|jt}fdd||D}i}|D]2\}tfdd|DsXq6jd}|j ^} } } |} |} | D]*} | |j } t | tr| |i} q | | j } | s|r6|r6| r| \}}nd\}}|r6z| | |}Wn.tk r4}z|s$|W5d}~XYnXt | tr6| | ksbt |trb| | ikr6|| | <q6|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)r_rrnr:r; sz6EnvSettingsSource.explode_env_vars..c3s|]}|VqdSrT) startswith)rprefix)rr:r;rsz5EnvSettingsSource.explode_env_vars..N)TT)r&r-rcrrranyrsplitrrr}r setdefaultrrirrO)rWr]r\rZis_dictprefixesresultrZenv_name_without_prefixrkeyslast_keyenv_varZ target_fieldrrZallow_json_failurer9r:)rrWr;rs>    " z"EnvSettingsSource.explode_env_varscCsd|jd|jdS)Nz'EnvSettingsSource(env_nested_delimiter=, env_prefix_len=rw)rrrnr:r:r;rxszEnvSettingsSource.__repr__)NNNNNN)N)rIrJrKrprXrr_rjr staticmethodrrrxryr:r:rtr;rs ".87rc seZdZdZedddddddfddddddddddd fd d Zd d d dZd d ddZdd fdd Zdd ddZ Z S)DotEnvSettingsSourcezB Source class for loading settings values from env files. NrRzDotenvType | Noner|r{r3) rSenv_fileenv_file_encodingr}r~rrrrr5c sP|tkr |n |jd|_|dk r&|n |jd|_t||||||| dS)Nrr)rFrUrrrrsrX) rWrSrrr}r~rrrrrtr:r;rXs zDotEnvSettingsSource.__init__rr4cCs|SrT)_read_env_filesrnr:r:r;rsz#DotEnvSettingsSource._load_env_varsc Csn|j}|dkriSt|ttjfr(|g}i}|D]8}t|}|r0|t ||j |j |j |j dq0|S)Nencodingr} ignore_emptyparse_none_str)rrrYrPathLiker rrupdate read_env_filerr}rr)rWZ env_filesZ dotenv_varsrZenv_pathr:r:r;rs&  z$DotEnvSettingsSource._read_env_filesrmc st}|jddk}|jD]\}}|s2q$d}|jjD]2\}}|||D]\}} }| | rVd}qBqVqB|s$|r| |j r|t |j d} ||| <q$|||<q$|S)NextraZforbidFT) rsrorVrrrrSrrrr~r) rWrZis_extra_allowedrZ env_valueZenv_usedr]r\rZfield_env_nameZnormalized_env_namertr:r;ro s"    zDotEnvSettingsSource.__call__rYc Cs&d|jd|jd|jd|jd S)NzDotEnvSettingsSource(env_file=z, env_file_encoding=z, env_nested_delimiter=rrw)rrrrrnr:r:r;rx$s$zDotEnvSettingsSource.__repr__) rIrJrKrprFrXrrrorxryr:r:rtr;rs&rcseZdZdZddddddddddejejejej ej e fdddddddddddd d d d d dd d fd d Z e ddddZe dddddZe dddddZdddddddfddZe ddd d!Ze dddd"d!Zdd#dd$dd%d!Zd&d'd'd(d)d*Zd'd&d+d'd,d-d.Zd'd&d'd/d0d1Zd'd&d2d'd3d4d5Zd6d'd7d8d9d:d;Zd'd7dd?d@Zd6dAdBdCdDZedEddFdGZd d'dddHdIdJdKZejejejej ej e fdEd d d d d dd dLdMdNZdd6d&d'd'dd&dOdPdQdRZd'd'd&dSd&dTdUdVZdd8d&d'd'd'd&dd7dSd dW dXdYZdd| dk r>tdtj|d |jd|j | d| dkrtt|j|jdn| } |j| | | ||||d|dkr|dkrtjdd}n"t|ttfstdt||j||j|ddS)NrrrFrTr3nullrrrrB.z'CLI settings source prefix is invalid: zGCase-insensitive matching is only supported on the internal root parser)rrrr~r})prog description)rrrrrrrNFrz>cli_parse_args must be List[str] or Tuple[str, ...], recieved  parsed_args)rUrr=argvrrrrrrrrendswithreplace isidentifierrPrsrXrMrp_connect_root_parserrrrrr _parse_argsr)rWrSrrrrrrrrr}rrrrrrrrtr:r;rXQsp   &    zCliSettingsSource.__init__rmr4cCsdSrTr:rnr:r:r;roszCliSettingsSource.__call__z"list[str] | tuple[str, ...] | boolzCliSettingsSource[T])rr5cCsdS)a Parse and load the command line arguments list into the CLI settings source. Args: args: The command line arguments to parse and load. Defaults to `None`, which means do not parse command line arguments. If set to `True`, defaults to sys.argv[1:]. If set to `False`, does not parse command line arguments. Returns: CliSettingsSource: The object instance itself. Nr:)rWrr:r:r;rosz&Namespace | dict[str, list[str] | str])rr5cCsdS)at Loads parsed command line arguments into the CLI settings source. Note: The parsed args must be in `argparse.Namespace` or vars dictionary (e.g., vars(argparse.Namespace)) format. Args: parsed_args: The parsed args to load. Returns: CliSettingsSource: The object instance itself. Nr:rWrr:r:r;ros)rrz)list[str] | tuple[str, ...] | bool | Nonez-Namespace | dict[str, list[str] | str] | Nonez%dict[str, Any] | CliSettingsSource[T])rrr5cs|dk r|dk rtdnf|dk rb|dkr6|jidS|dkrLtjdd}|j||j|dS|dk rv|j|dStSdS)Nz/`args` and `parsed_args` are mutually exclusiveFrTr)rPrr=rrrrsro)rWrrrtr:r;ros   rcCsdSrTr:rnr:r:r;rsz CliSettingsSource._load_env_varscCsdS)a Loads the parsed command line arguments into the CLI environment settings variables. Note: The parsed args must be in `argparse.Namespace` or vars dictionary (e.g., vars(argparse.Namespace)) format. Args: parsed_args: The parsed args to load. Returns: CliSettingsSource: The object instance itself. Nr:r r:r:r;rsrz/Mapping[str, str | None] | CliSettingsSource[T]cs|dkr iSt|trt|}g}|D]N\}}t|trN|||||<q*|dr*|dk r*||dd|q*|j D] }|D]}||kr|j ||<qqdd|D}|rt |t dtfdd|Dsd |<tttttf||j|j|j |_|S) N :subcommand:rcSs i|]\}}|ds||qS)r )r)rrrr:r:r; s z4CliSettingsSource._load_env_vars..)rc3s |]}d|kr|VqdS)rNr:)rr]Zlast_selected_subcommandr:r;r sz3CliSettingsSource._load_env_vars..z{})rrvarsrr_merge_parsed_listrrr_cli_subcommandsrrmaxrrrrrrrYr}rr)rWrZselected_subcommandsr]r subcommands subcommandr:r r;rs4   z list[str]rY) parsed_listr]r5c Cszg}d}|j|t}|tksDtt|rDtddt|DsJ|}n&|rlt|dksh|ddrltnt }|D]}|dr| dr|dd}|r<|d r| |||}d}q|d s|dr| ||}nXz| |||}WnDtk r4}z$||kr||}| |||}W5d}~XYnXd }q|st| |||}qt|t krd|dWS|tkrdd |dWSi} |D]} | t| qt| WSWn8tk r}ztd |d |W5d}~XYnXdS)NFcss.|]&}|tdk rt|ttfkr|VqdSrT)rr-rrrrr:r:r;rs z7CliSettingsSource._merge_parsed_list..rr[],{TzParsing error encountered for z: )_cli_dict_argsrrr"r-rr,rrrYr_consume_comma_consume_object_or_array_consume_string_or_numberrjoinrrkrldumpsrrP) rWrr] merged_listis_last_consumed_a_value merge_typeZ inferred_typerr9Z merged_dictitemr:r:r;rsX  $       z$CliSettingsSource._merge_parsed_listr`)r$r!r"r5cCs|s|d|ddS)Nz""r)r)rWr$r!r"r:r:r;rLs z CliSettingsSource._consume_comma)r$r!r5cCsd}|drdnd}tdt|D]l}||dkr>|d7}q$||dkr$|d8}|||kr$|dkr$||d|d||ddSq$td|d dS) Nrr}r)rr)r%rrzMissing end delimiter "r)rrangerrrP)rWr$r!countZ close_delimconsumedr:r:r;rQs   z*CliSettingsSource._consume_object_or_arraytype[Any] | None)r$r!r#r5c Csz|tk r dnt|}d}|t|krj||dkrN|dksH||ddkrN| }|s`||dkr`qj|d7}q|rvtd|d|}|ttfkrz t|Wn@tk r||jkrd}|d kr|dsd|d}YnX| |nd d | d dD\}}|drD| dsD|dsD| drDtd ||d|d}}| t ||i||dS)NrFrr\rzMismatched quotesr)truefalsercss|] }|VqdSrTr:)rkvr:r:r;rtsz>CliSettingsSource._consume_string_or_number..=z1Dictionary key=val parameter is a quoted string: )rYrrPrrfloatrrrrrrrkr ) rWr$r!r#r(Zis_find_end_quoteZ val_stringrrr:r:r;r^s4 $     0z+CliSettingsSource._consume_string_or_numberztype[BaseModel]r(zlist[type[BaseModel]])modelr]rr5cCst|js|jfnt|j}|jr4tdd|D}g}|D]l}t|tfddrhtd|jd|n&t|tfddrtd|jd|t |st |r<| |q<|S)NcSsg|]}|tdk r|qSrTrrr:r:r;rs z5CliSettingsSource._get_sub_models..F)is_include_originz.CliSubCommand is not outermost annotation for rz1CliPositionalArg is not outermost annotation for ) r,rcrr_annotation_contains_typesrGrPrIrLr%r'r)rWr0r]r field_types sub_modelsrr:r:r;_get_sub_models{s z!CliSettingsSource._get_sub_modelszdict[str, str]ztuple[tuple[str, ...], bool])r]ralias_path_argsr5c Cs*g}d}t|j|jfs(||g7}d}ng}|j|jfD]p}|dkrHq8q8t|trb||d}q8t|tr|jD](}t|tr||d}qr||qrq8||q8|D]R} tt| j d}|j s| n|}t | j dkrdnd||<|s|r||q|j sdd|D}t t||fS) NTFrrrrcSsg|] }|qSr:r)rZ resolved_namer:r:r;rsz9CliSettingsSource._get_resolved_names..)rrrrrYrrchoicesrrr}rrrrfromkeys) rWr]rr7resolved_namesis_alias_path_onlyZnew_alias_pathsrrZ alias_pathr:r:r;_get_resolved_namess8         z%CliSettingsSource._get_resolved_nameszlist[tuple[str, FieldInfo]])r0r5c Csggg}}}t|r|jn|j}|D]R\}}t|jkr|sdtd|jd|dnt |j |j frtd|jd|dnjddt |j D}t|dkrtd|jd|dn0t|d st|d std|jd|d |||fq,t|jkrr|s8td |jd|dn*t |j |j frbtd |jd|d|||fq,|||fq,|||S) Nzsubcommand argument rz has a default valuez has an aliascSsg|]}|tdk r|qSrTr1rr:r:r;rs z6CliSettingsSource._sort_arg_fields..rz has multiple typesrz is not derived from BaseModelzpositional argument )r'rrrrGrd is_requiredrPrIrrrr,rcrr%rrL) rWr0Zpositional_argsZsubcommand_argsZ optional_argsrr]rr4r:r:r;_sort_arg_fieldss0    z"CliSettingsSource._sort_arg_fieldsrNcCs|jS)z#The connected root parser instance.) _root_parserrnr:r:r;rszCliSettingsSource.root_parserzCallable[..., Any]) parser_method method_namerkwargsr5csndk rD|jdkrDdkrDt|jtrDd dddddfdd }|Sdkrfdddd fd d }|SSdS)NFparsed_args_methodrMz"list[str] | tuple[str, ...] | NonezNamespace | Noner)rr namespacer5cs`g}|rtt|ngD]6}td|}|rH|d|d}||q|||S)Nz^(--[^\s=]+)(.*)rr)shlexrrrematchgrouprr)rrrEZinsensitive_argsrZmatched)rAr:r;parse_args_insensitive_methods  zOCliSettingsSource._connect_parser_method..parse_args_insensitive_method)rrCr5cstdddS)Nz0cannot connect CLI settings source root parser: z. is set to `None` but is needed for connecting)rP)rrC)rBr:r;none_parser_methods zDCliSettingsSource._connect_parser_method..none_parser_method)NN)r}rr@rM)rWrArBrrCrJrKr:)rBrAr;_connect_parser_methods    z(CliSettingsSource._connect_parser_method)rrrrrrrr5c Cs||_||d|_||d|_||d|_||d|_||d|_||_i|_i|_ |j |j |j g|j |j dgddS)NrDrrrrparserr0 added_args arg_prefixsubcommand_prefixrIalias_prefixes)r@rLr _add_argument_add_argument_group _add_parser_add_subparsers_formatter_classrr_add_parser_argsrrSr~)rWrrrrrrrr:r:r;rs$ z&CliSettingsSource._connect_root_parserr)rNr0rOrPrQrIrRr5c sd}i} ||D]\} } ||| | } t| jkr |dkrt|j|d|d|jd}|| g|j|d<n|j|d|| t|drd |j|d} d| d|_ | d}|j |j || | j |j|jd |g|| d || d dgd q|| | | \}}d i}t|d <|| |d<||d|d<|| j|d<|jon| |d<|d|krqt| jtttttfddrd|d<t| jttfddr| j|j|d<|||||}t| jkr|d|d<|dg}|d=|d=d| rH| ddkrH|!|| |||||| | q|rRqq|dk rt"|trv|j#|f|}|t|7}|j$|ffdd|D|q|t|7}|j$|ffdd|D|q|%|| |||||S)Nrr )titledestrequiredmetavarrrr%r)helprrrrM--defaultr]rZr[T)is_strip_annotatedractionrBc3s|]}|VqdSrTr:rrarg_flagr:r;risz5CliSettingsSource._add_parser_args..c3s|]}|VqdSrTr:rbrcr:r;rls)&r?r6rGrdrVrrrrrr\rXrUrrWrpr=r _help_format_metavar_formatrcr>r3rsetrrrr_get_arg_namesrLupperr_add_parser_submodelsrrTrS_add_parser_alias_paths)rWrNr0rOrPrQrIrRZ subparsersr7r]rr5r\r;r<rC arg_namesr:rcr;rXs            " "z"CliSettingsSource._add_parser_argsztuple[str, ...])rPrQrRr;r5c CsTg}|g|D]@}|D]6}|||jkr4||n||dd|qq|S)NrBr)rr~r)rWrPrQrRr;rlrrr:r:r;rhqsz CliSettingsSource._get_arg_names) rNr5rOrPrQrdrlrCrr;r5c  sd} i} |dd| d<|jr6t|dkr6|djn| j| d<|js||dd|dd|d<|j|f| } |j| ffd d |D||D]H} |j|| || dd || r| n| fd d | ddDdqdS)Nrz optionsrYrrzset z from JSON stringr]c3s|]}|VqdSrTr:rbrcr:r;rsz:CliSettingsSource._add_parser_submodels..rcsg|]}|dqS)rr:rb)rPr:r;rsz;CliSettingsSource._add_parser_submodels..rM) rrrprrrrTrSrX)rWrNr5rOrPrQrdrlrCrr;Z model_groupZmodel_group_kwargsr0r:)rdrPr;rj~s0    z'CliSettingsSource._add_parser_submodels)rNr7rOrPrQrIr5c Cs |r|}|dk r.t|tr*|j|f|n|}|d}|rH|ddn|}|D]\} } |rddn| } ||jkr~|| n||dd| } i} t| d<d| d<|| | d<| d ks|rd | d <nd | d <d | d <| |krT|| |j |d| f| qTdS)NrrrBrr_zpydantic alias pathr]rZrr\rrarr^) rrrTrrr~rrrrS) rWrNr7rOrPrQrIcontextZis_nested_alias_pathrr\Zarg_namerCr:r:r;rks.      z)CliSettingsSource._add_parser_alias_paths)objr5cCs(|jst|Stddt|DSdS)NcSsg|]}|tdk r|qSrTr1rr:r:r;rs z8CliSettingsSource._get_modified_args..)rr,rrWrnr:r:r;_get_modified_argssz$CliSettingsSource._get_modified_args)r obj_qualnamer5cCs|d|kr>|d|dddd||dddD}d|}|r\|d|dSt|dkrl|Sd|d SdS) NJSONrcSsg|]}|dkr|qS)rrr:rrr:r:r;rsz=CliSettingsSource._metavar_format_choices..rrrrr%)indexrr)rWrrqr\r:r:r;_metavar_format_choicess 6 z)CliSettingsSource._metavar_format_choicescCsRt|}t|tr|jS|dkr$dSt|tr6t|St|tjrJt|St|t t t fs`|j }t t|r|tt|j||St|tjtjfkr|ttt||St|tr|dd|DSt|t r|jtt|j|||jdS|t dkr|jSt|r$dSt|t r6|jSt|dd d d SdS) z\Pretty metavar representation of a type. Adapts logic from `pydantic._repr.display_as_type`..z...cSsg|] }|jqSr:rrr:r:r;rsz=CliSettingsSource._metavar_format_recurse..)rqNrrztyping.rBztyping_extensions.)_strip_annotatedrrrIr reprtyping_extensionsZ TypeAliasTyperYr#r!rrur"r-rurmap_metavar_format_recurserpLiteraltypingr&r rKrr%rror:r:r;rzs:        z)CliSettingsSource._metavar_format_recursecCs||ddS)Nz, r)rzrror:r:r;rfsz!CliSettingsSource._metavar_format)rr5cCs|jr |jnd}|r4t|jkr||r,dnd7}nZd|jd}|jtdfkr`d|jd}n|jdk rxd|jd}||rd|n|7}tt |j t r| ddS|S) NrBz (required)z (required)z (default: rw %z%%) rr>rLrdrr_r)default_factory issubclassrr@rr)rWrZ_helpr_r:r:r;res  zCliSettingsSource._help_format)N)%rIrJrKrpr parse_args add_argumentadd_argument_groupr add_parseradd_subparsersrrXrrorrrrrr6r=r?propertyrrLrrXrhrjrkrprurzrfreryr:r:rtr;r+sv(6W$5 #(_ "%# "rc@s0eZdZdddddZeddddd Zd S) ConfigFileSourceMixinPathType | Nonerm)filesr5cCsX|dkr iSt|ttjfr"|g}i}|D](}t|}|r*|||q*|SrT) rrYrrr rrr _read_file)rWrrfile file_pathr:r:r; _read_files s z!ConfigFileSourceMixin._read_filesr )rr5cCsdSrTr:)rWrr:r:r;rsz ConfigFileSourceMixin._read_fileN)rIrJrKrrrr:r:r:r;r s rcs@eZdZdZedfddddfdd Zd d d d d ZZS)JsonConfigSettingsSourcez> A source class that loads variables from a JSON file NrRrr|)rS json_filejson_file_encodingcsV|tkr |n |jd|_|dk r&|n |jd|_||j|_t||jdS)Nrr) rDrUrZjson_file_pathrrZ json_datarsrX)rWrSrrrtr:r;rX s z!JsonConfigSettingsSource.__init__r rmrr5c Cs0t||jd}t|W5QRSQRXdSNr)openrrkload)rWrrr:r:r;r/sz#JsonConfigSettingsSource._read_filerIrJrKrprDrXrryr:r:rtr;rs rcs<eZdZdZefdddfdd Zddd d d ZZS) TomlConfigSettingsSourcez> A source class that loads variables from a TOML file rRr)rS toml_filecs<|tkr |n |jd|_||j|_t||jdS)Nr)rDrUrtoml_file_pathr toml_datarsrX)rWrSrrtr:r;rX9sz!TomlConfigSettingsSource.__init__r rmrc CsTtt|dd8}tjdkr4t|W5QRSt|W5QRSQRXdS)Nrb)moder/)rArr=r>r?rr@)rWrrr:r:r;rBs  z#TomlConfigSettingsSource._read_filerr:r:rtr;r4s rcsBeZdZdZdddddfdd Zedd d d d d ZZS)!PyprojectTomlConfigSettingsSourcezK A source class that loads variables from a `pyproject.toml` file. NrRrr3)rSrr5csj|||jdd|_|jdd|_||j|_|jD]}|j|i|_q A source class that loads variables from a yaml file NrRrr|)rS yaml_fileyaml_file_encodingcsV|tkr |n |jd|_|dk r&|n |jd|_||j|_t||jdS)Nrr) rDrUrZyaml_file_pathrrZ yaml_datarsrX)rWrSrrrtr:r;rX}s z!YamlConfigSettingsSource.__init__r rmrc Cs6tt||jd}t|W5QRSQRXdSr)r<rrr6Z safe_load)rWrrr:r:r;rsz#YamlConfigSettingsSource._read_filerr:r:rtr;rxs rFrYr`)rr}r5cCs|r|S|SrTr8)rr}r:r:r;_get_env_var_keysrr|zstr | None | EnvNoneType)rgrr5cCs||kr|dk s|St|SrT)rO)rgrr:r:r;_parse_env_none_strsrr)rr}rrr5csfdd|DS)Ncs.i|]&\}}r|dkst|t|qS)rB)rr)rkvr}rrr:r;r s z"parse_env_vars..)r)rr}rrr:rr;rsrrr )rrr}rrr5cCst||p dd}t||||S)Nutf8r)rr)rrr}rrZ file_varsr:r:r;rsrr)z list[Any])rcrdr5cCsbtdd|DrdSt|tr6t|^}}t||St|}t|p`t|p`t|dp`t|dS)Ncss|]}t|tVqdSrT)rr)rmdr:r:r;rsz)_annotation_is_complex..FZ__pydantic_core_schema__Z__get_pydantic_core_schema__)rrr+r,rbr-_annotation_is_complex_innerr)rcrdinnermetaoriginr:r:r;rbs   rb)rcr5c Cs2t|ttfrdSt|ttttttt fp0t |Sr) r&rYbytesrrrrrg frozensetr r rcr:r:r;rs rcstfddt|DS)Nc3s|]}t|VqdSrT)rbrsrdr:r;rsz$_union_is_complex..)rr,)rcrdr:rr;rsrTztuple[Any, ...])rctypesr2r`r5cCsP|r t|}|dkr$t||kr$dSt|D]}t||d|dr,dSq,||kS)NT)r2r`)rvr-r,r3)rcrr2r`rr:r:r;r3s r3rcCst|tkrt|d}q|S)Nr)r-r*r,rr:r:r;rvs rv)F)N)FFN)TF)s __future__rZ _annotationsrkrrGrFr=r|rabcrrargparserrrrr collectionsr dataclassesr enumr pathlibr rrrrrrrrrrrrrrrxZdotenvrZpydanticrrrrZpydantic._internal._reprr Z pydantic._internal._typing_extrar!r"r#Zpydantic._internal._utilsr$r%r&Zpydantic.dataclassesr'Zpydantic.fieldsr(Z pydantic_corer)r*r+r,r-Zpydantic_settings.utilsr.r>r@r?r6Zpydantic_settings.mainr2r<rArYrErCrD__annotations__rFrGrLrMrNZ CliSubCommandZCliPositionalArgrOrrPrQrqrzrrrrrrrrrrrrrrbrrr3rvr:r:r:r;s      8         ,, I:]gYc.