o i%@s.dZddlmZmZddlmZmZmZmZm Z m Z m Z m Z m Z mZddlZddlZddlZddlmZmZddlmZmZddlmZmZmZmZmZmZe dZe deee egeffZGd d d ej j!ej"j!Z#d e ed eed e efddZ$d e e ed ee ed e e efddZ%dS)zPlugin module for Temporal SDK. This module provides plugin functionality that allows customization of both client and worker behavior in the Temporal SDK through configurable parameters. )AbstractAsyncContextManagerasynccontextmanager) Any AsyncIterator AwaitableCallableOptionalSequenceTypeTypeVarUnioncastN) ClientConfigWorkflowHistory) ConnectConfig ServiceClient)ReplayerReplayerConfigWorker WorkerConfigWorkflowReplayResultWorkflowRunnerTc@seZdZdZdddddddddd dedeejjdee ej j dee e dee e d ee ed eed ee ejj d ee eed ee gedfddfddZdefddZdedefddZdede egeefdejjfddZdedefddZdedefddZ de!de e!gedfddfddZ"e#d e$d!e%e&de e$e%e&gee%e'fde%e%e'fd"d#Z(dS)$ SimplePluginzA simple plugin definition which has a limited set of configurations but makes it easier to produce a plugin which needs to configure them. N) data_converterclient_interceptors activitiesnexus_service_handlers workflowsworkflow_runnerworker_interceptors workflow_failure_exception_types run_contextnamerrrrrrr r!r"returnc Cs@||_||_||_||_||_||_||_||_| |_| |_ dS)aCreate a simple plugin with configurable parameters. Each of the parameters will be applied to any component for which they are applicable. All arguments are optional, and all but run_context can also be callables for more complex modification. See the type PluginParameter above. For details on each argument, see below. Args: name: The name of the plugin. data_converter: Data converter for serialization, or callable to customize existing one. Applied to the Client and Replayer. client_interceptors: Client interceptors to append, or callable to customize existing ones. Applied to the Client. Note, if the provided interceptor is also a worker.Interceptor, it will be added to any worker which uses that client. activities: Activity functions to append, or callable to customize existing ones. Applied to the Worker. nexus_service_handlers: Nexus service handlers to append, or callable to customize existing ones. Applied to the Worker. workflows: Workflow classes to append, or callable to customize existing ones. Applied to the Worker and Replayer. workflow_runner: Workflow runner, or callable to customize existing one. Applied to the Worker and Replayer. worker_interceptors: Worker interceptors to append, or callable to customize existing ones. Applied to the Worker and Replayer. workflow_failure_exception_types: Exception types for workflow failures to append, or callable to customize existing ones. Applied to the Worker and Replayer. run_context: A place to run custom code to wrap around the Worker (or Replayer) execution. Specifically, it's an async context manager producer. Applied to the Worker and Replayer. Returns: A configured Plugin instance. N) _namerrrrrrr r!r") selfr#rrrrrrr r!r"r'W/var/www/html/karishye-ai-python/venv/lib/python3.10/site-packages/temporalio/plugin.py__init__-s2 zSimplePlugin.__init__cCs|jS)See base class.)r%)r&r'r'r(r#jszSimplePlugin.nameconfigcCsDt|d|j}|r||d<t|d|j}|dur ||d<|S)r*r interceptorsN)_resolve_parametergetr_resolve_append_parameterr)r&r+rr,r'r'r(configure_clientns  zSimplePlugin.configure_clientnextcs||IdHSr*Nr')r&r+r1r'r'r(connect_service_client~sz#SimplePlugin.connect_service_clientcCst|d|j}|r||d<t|d|j}|dur ||d<t|d|j}|dur1||d<t|d|j}|r@||d<t|d|j}|durQ||d<t|d|j}|durb||d<|S)r*rrNrrr,r!) r/r.rrrr-rr r!)r&r+rrrrr,failure_exception_typesr'r'r(configure_workers<    zSimplePlugin.configure_workercCst|d|j}|r||d<t|d|j}|dur ||d<t|d|j}|r/||d<t|d|j}|dur@||d<t|d|j}|durQ||d<|S)r*rrNrr,r!)r-r.rr/rrr r!)r&r+rrrr,r4r'r'r(configure_replayers2   zSimplePlugin.configure_replayerworkerc sj|jr,|4IdH||IdHWdIdHdS1IdHs%wYdS||IdHdSr2r")r&r7r1r'r'r( run_workers .zSimplePlugin.run_workerreplayer historiesc Cs|jrG|4IdH.|||4IdH}|VWdIdHn 1IdHs*wYWdIdHdS1IdHs@wYdS|||4IdH}|VWdIdHdS1IdHsewYdSr2r8)r&r:r;r1resultsr'r'r( run_replayers *..zSimplePlugin.run_replayer))__name__ __module__ __qualname____doc__strPluginParameter temporalio converter DataConverterr client Interceptorrrr rr7 BaseExceptionrrr)r#rr0rrrservicer3rr5rr6rr9rrrrrr=r'r'r'r(r(s         = '!     rexisting parameterr$cCs8|dur|St|rttttgttf||S|SN)callabler rrrrKrLr'r'r(r-s  r-cCsP|dur|St|rtttttgtttf||St|p"gt|SrM)rNr rrr rlistrOr'r'r(r/sr/)&rA contextlibrrtypingrrrrrr r r r r temporalio.clientrDtemporalio.convertertemporalio.workerrrtemporalio.servicerrrrrrrrrrCrGPluginr7rr-r/r'r'r'r(s60   E