o i<@s:UdZddlmZddlmZmZddlmZddlm Z m Z m Z m Z m Z mZmZddlmZmZmZddlZddlmZeeeefZd ed <Gd d d ed dZGddded dZd+ddZd,ddZ d-ddZ!d.dd Z"ed!d"Gd#d$d$Z#Gd%d&d&ed dZ$ed!d"Gd'd(d(Z%eGd)d*d*Z&dS)/zEnvironment and file-based configuration for Temporal clients. This module provides utilities to load Temporal client configuration from TOML files and environment variables. ) annotations) dataclassfieldPath)AnyDictLiteralMappingOptionalUnioncast)Self TypeAlias TypedDictN) envconfigr DataSourcec@s:eZdZUdZded<ded<ded<ded<ded <d S) ClientConfigTLSDictz1Dictionary representation of TLS config for TOML.Optional[bool]disabledstr server_nameMapping[str, str]server_ca_cert client_cert client_keyN__name__ __module__ __qualname____doc____annotations__r"r"Z/var/www/html/karishye-ai-python/venv/lib/python3.10/site-packages/temporalio/envconfig.pyr  rF)totalc@:eZdZUdZded<ded<ded<ded<ded <d S) ClientConfigProfileDictz>Dictionary representation of a client config profile for TOML.raddress namespaceapi_keyrtlsr grpc_metaNrr"r"r"r#r'"r$r'dOptional[Mapping[str, Any]]returnOptional[DataSource]cCs0|sdSd|vr |dSd|vrt|dSdS)Ndatapathrr-r"r"r#_from_dict_to_source,s r4sourceOptional[Mapping[str, str]]cCsDt|tr dt|iSt|trd|iSt|tr d|diSdS)Nr2r1utf-8) isinstancerrbytesdecode)r5r"r"r#_source_to_dict6s    r;%tuple[Optional[str], Optional[bytes]]cCsxd}d}t|trt|}||fSt|tr|d}||fSt|tr*|}||fS|dur8tdt|j||fS)Nr7zHconfig_source must be one of pathlib.Path, str, bytes, or None, but got )r8rrencoder9 TypeErrortyper)r5r2r1r"r"r#_source_to_path_and_dataBs$   r@Optional[bytes]cCs~|durdSt|tr$t|d }|WdS1swYt|tr.|dSt|tr5|Stdt|j )Nrbr7z;Source must be one of pathlib.Path, str, or bytes, but got ) r8ropenreadrr=r9r>r?r)r5fr"r"r# _read_sourceUs     rFT)frozenc@szeZdZUdZdZded< dZded< dZded< dZded < dZ ded < dd dZ dddZ e dddZ dS)ClientConfigTLSzoTLS configuration as specified as part of client configuration .. warning:: Experimental API. Nrr Optional[str]rr0server_root_ca_certrclient_private_keyr/rcsbi|jdur |jd<|jdur|jd<d fdd }|d |j|d |j|d |jS)@Convert to a dictionary that can be used for TOML serialization.Nrrkey6Literal['server_ca_cert', 'client_cert', 'client_key']r5r0cs(|durt|}r||<dSdSdSN)r;)rMr5valr3r"r# set_sources z+ClientConfigTLS.to_dict..set_sourcerrr)rMrNr5r0rrrJrrK)selfrQr"r3r#to_dictws       zClientConfigTLS.to_dict)Union[bool, temporalio.service.TLSConfig]cCs6|jdurdStjj|jt|jt|jt|jdS)z:Create a `temporalio.service.TLSConfig` from this profile.TF)domainrJrrK) r temporalioservice TLSConfigrrFrJrrKrSr"r"r#to_connect_tls_configs z%ClientConfigTLS.to_connect_tls_configr-Optional[ClientConfigTLSDict]Optional[Self]c CsD|sdS||d|dt|dt|dt|ddS)z+Create a ClientConfigTLS from a dictionary.NrrrrrrR)getr4clsr-r"r"r# from_dicts   zClientConfigTLS.from_dict)r/r)r/rU)r-r\r/r])rrrr rr!rrJrrKrTr[ classmethodrar"r"r"r#rHds         rHc@r&) ClientConnectConfigzArguments for `temporalio.client.Client.connect` that are configurable via environment configuration. .. warning:: Experimental API. r target_hostr)r*rUr+r rpc_metadataNrr"r"r"r#rcs  rcc@seZdZUdZdZded< dZded< dZded< dZded< e e d Z d ed < e d$ddZ d%ddZd&ddZe d'ddddddd(d"d#ZdS))ClientConfigProfilea"Represents a client configuration profile. This class holds the configuration as loaded from a file or environment. See `to_connect_config` to transform the profile to `ClientConnectConfig`, which can be used to create a client. .. warning:: Experimental API. NrIr(r)r*zOptional[ClientConfigTLS]r+)default_factoryrr,r-r'r/rc Cs:||d|d|dt|d|dpidS)z/Create a ClientConfigProfile from a dictionary.r(r)r*r+r,)r(r)r*r+r,)r^rHrar_r"r"r#ras zClientConfigProfile.from_dictcCspi}|jdur |j|d<|jdur|j|d<|jdur |j|d<|jr.|j}r.||d<|jr6|j|d<|S)rLNr(r)r*r+r,)r(r)r*r+rTr,)rSr-tls_dictr"r"r#rTs       zClientConfigProfile.to_dictrccCsvi}|jr |j|d<|jdur|j|d<|jdur"|j|d<d|d<|jdur.|j|d<|jr6|j|d<tt|S)z1Create a `ClientConnectConfig` from this profile.rdNr)r*Tr+re)r(r)r*r+r[r,r rc)rSconfigr"r"r#to_client_connect_configs        z,ClientConfigProfile.to_client_connect_configF) config_source disable_file disable_envconfig_file_strictoverride_env_varsprofilerkr0rlboolrmrnror6c Cs.t|\}}tj|||||||d}t|S)a!Load a single client profile from given sources, applying env overrides. To get a :py:class:`ClientConnectConfig`, use the :py:meth:`to_client_connect_config` method on the returned profile. Args: profile: Profile to load from the config. config_source: If present, this is used as the configuration source instead of default file locations. This can be a path to the file or the string/byte contents of the file. disable_file: If true, file loading is disabled. This is only used when ``config_source`` is not present. disable_env: If true, environment variable loading and overriding is disabled. This takes precedence over the ``override_env_vars`` parameter. config_file_strict: If true, will error on unrecognized keys. override_env_vars: The environment to use for loading and overrides. If not provided, the current process's environment is used. To use a specific set of environment variables, provide them here. To disable environment variable loading, set ``disable_env`` to true. Returns: The client configuration profile. )rpr2r1rlrmrnenv_vars)r@_bridge_envconfigload_client_connect_configrfra) rprkrlrmrnror2r1 raw_profiler"r"r#loads $ zClientConfigProfile.load)r-r'r/r)r/r')r/rcrO)rprIrkr0rlrqrmrqrnrqror6r/rf)rrrr r(r!r)r*r+rdictr,rbrarTrj staticmethodrvr"r"r"r#rfs2       rfc@sleZdZUdZded< d!ddZed"d d Zed dd dd#ddZ e d$d dddd dd%dd Z d S)& ClientConfigaKClient configuration loaded from TOML and environment variables. This contains a mapping of profile names to client profiles. Use `ClientConfigProfile.to_connect_config` to create a `ClientConnectConfig` from a profile. See `load_profile` to load an individual profile. .. warning:: Experimental API. z!Mapping[str, ClientConfigProfile]profilesr/%Mapping[str, ClientConfigProfileDict]cCsdd|jDS)rLcSsi|] \}}||qSr")rT.0kvr"r"r# ?sz(ClientConfig.to_dict..)rzitemsrZr"r"r#rT=szClientConfig.to_dictr-Mapping[str, Mapping[str, Any]]rcCs|dd|DdS)z(Create a ClientConfig from a dictionary.cSs"i|] \}}|ttt|qSr")rfrar r'r|r"r"r#rJsz*ClientConfig.from_dict..)rz)rr_r"r"r#raAs zClientConfig.from_dictNF)rkrnrorkr0rnrqror6cCs(t|\}}tj||||d}t|S)aLoad all client profiles from given sources. This does not apply environment variable overrides to the profiles, it only uses an environment variable to find the default config file path (``TEMPORAL_CONFIG_FILE``). To get a single profile with environment variables applied, use :py:meth:`ClientConfigProfile.load`. Args: config_source: If present, this is used as the configuration source instead of default file locations. This can be a path to the file or the string/byte contents of the file. config_file_strict: If true, will TOML file parsing will error on unrecognized keys. override_env_vars: The environment variables to use for locating the default config file. If not provided, the current process's environment is used to check for ``TEMPORAL_CONFIG_FILE``. To use a specific set of environment variables, provide them here. To disable environment variable loading, set ``disable_file`` to true or pass an empty dictionary for this parameter. )r2r1rnrr)r@rsload_client_configryra)rkrnror2r1loaded_profilesr"r"r#rvPs  zClientConfig.load) config_filerlrmrnrorprIrrlrmrccCs2d}|r |s t|}tj||||||d}|S)a3Load a single client profile and convert to connect config. This is a convenience function that combines loading a profile and converting it to a connect config dictionary. This will use the current process's environment for overrides unless disabled. Args: profile: The profile to load from the config. Defaults to "default". config_file: Path to a specific TOML config file. If not provided, default file locations are used. This is ignored if ``disable_file`` is true. disable_file: If true, file loading is disabled. disable_env: If true, environment variable loading and overriding is disabled. config_file_strict: If true, will error on unrecognized keys in the TOML file. override_env_vars: A dictionary of environment variables to use for loading and overrides. If not provided, the current process's environment is used. To use a specific set of environment variables, provide them here. To disable environment variable loading, set ``disable_env`` to true. Returns: TypedDict of keyword arguments for :py:meth:`temporalio.client.Client.connect`. N)rprkrlrmrnro)rrfrvrj)rprrlrmrnrorkprofr"r"r#rtus$z'ClientConfig.load_client_connect_config)r/r{)r-rr/r)rkr0rnrqror6r/ryrO)rprIrrIrlrqrmrqrnrqror6r/rc) rrrr r!rTrbrarxrvrtr"r"r"r#ry.s*    $ry)r-r.r/r0)r5r0r/r6)r5r0r/r<)r5r0r/rA)'r __future__r dataclassesrrpathlibrtypingrrr r r r r typing_extensionsrrrtemporalio.servicerW%temporalio.bridge.temporal_sdk_bridgerrsrr9rr!rr'r4r;r@rFrHrcrfryr"r"r"r#s0  $   Bw