o iR@s UddlmZddlZddlZddlmZmZmZmZddl m Z ddl m Z ddlm Z ddlmZddlmZmZmZmZmZmZmZdd lmZmZdd lmZddlZddlZddl Zdd l!m"Z"dd l#m$Z$dd l%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+erddl,Ze dZ-de.d<e dZ/de.d<e dZ0de.d<e ddGdddZ1d:ddZ2d;dd Z3dd(d)Z6e d?d+d,Z7d:d-d.Z8e Gd/d0d0Z9Gd1d2d2eZ:e ddGd3d4d4Z;e ddGd5d6d6Zd9dZ?dS)@) annotationsN) AwaitableMappingMutableMappingSequence)contextmanager) ContextVar) dataclass) timedelta) TYPE_CHECKINGAnyCallable GeneratorOptionalUnionoverload)CancelOperationContextStartOperationContext) Concatenate)_link_conversion)WorkflowHandle)MethodAsyncNoParamMethodAsyncSingleParamMultiParamSpec ParamType ReturnTypeSelfTypez temporal-start-operation-contextz*ContextVar[_TemporalStartOperationContext]!_temporal_start_operation_contextz!temporal-cancel-operation-contextz+ContextVar[_TemporalCancelOperationContext]"_temporal_cancel_operation_contextz-temporal-nexus-backing-workflow-start-contextzContextVar[bool]._temporal_nexus_backing_workflow_start_contextT)frozenc@seZdZUdZded<dS)InfozInformation about the running Nexus operation. .. warning:: This API is experimental and unstable. Retrieved inside a Nexus operation handler via :py:func:`info`. str task_queueN__name__ __module__ __qualname____doc____annotations__r*r*i/var/www/html/karishye-ai-python/venv/lib/python3.10/site-packages/temporalio/nexus/_operation_context.pyr!=s r!returnboolcCs tduS)z5Whether the current code is inside a Nexus operation.N)_try_temporal_contextr*r*r*r+ in_operationK r/cCs tS)z,Get the current Nexus operation information.)_temporal_contextinfor*r*r*r+r2Pr0r2temporalio.client.ClientcCstjS)zPGet the Temporal client used by the worker handling the current Nexus operation.)r1clientr*r*r*r+r4Usr4FUnion[_TemporalStartOperationContext, _TemporalCancelOperationContext]cCst}|dur td|SNzNot in Nexus operation context.)r. RuntimeErrorctxr*r*r+r1Zsr1POptional[Union[_TemporalStartOperationContext, _TemporalCancelOperationContext]]cCs,td}td}|r|rtd|p|S)Nz6Cannot be in both start and cancel operation contexts.)rgetrr7) start_ctx cancel_ctxr*r*r+r.cs  r.Generator[None, None, None]c cs0td}z dVWt|dSt|w)NT)rsetreset)tokenr*r*r+%_nexus_backing_workflow_start_contextms  rBcCs tdS)NF)rr;r*r*r*r+(_in_nexus_backing_workflow_start_contextvs rCc@sfeZdZUdZded< ded< ded< edd d Zdd d ZdddZdddZ dddZ dS)_TemporalStartOperationContextzMContext for a Nexus start operation being handled by a Temporal Nexus Worker.r nexus_contextCallable[[], Info]r2r3r4r,cCtd}|dur td|Sr6)rr;r7clsr9r*r*r+r; z"_TemporalStartOperationContext.getNonecCt|dSN)rr?selfr*r*r+r?z"_TemporalStartOperationContext.set list[temporalio.client.Callback]cCs"|j}|jrt|j|jdgSgS)N)urlheaders)rE callback_url NexusCallbackcallback_headers)rOr9r*r*r+_get_callbackssz-_TemporalStartOperationContext._get_callbacks1list[temporalio.api.common.v1.Link.WorkflowEvent]cCs.g}|jjD]}t|}r||q|SrM)rE inbound_linksrnexus_link_to_workflow_eventappend)rO event_links inbound_linklinkr*r*r+_get_workflow_event_linkss   z8_TemporalStartOperationContext._get_workflow_event_linksworkflow_handle*temporalio.client.WorkflowHandle[Any, Any]c Csg}z8t|jtjjjjr$|jdr$|jj}r$|dr$| |j |s,t |g}|j jdd|DW|StyX}ztd|d|WYd}~|Sd}~ww)Nr^workflow_eventcss|]}t|VqdSrM)rworkflow_event_to_nexus_link).0r^r*r*r+ s  zE_TemporalStartOperationContext._add_outbound_links..zCFailed to create WorkflowExecutionStarted event links for workflow z: ) isinstance_start_workflow_response temporalioapiworkflowservicev1StartWorkflowExecutionResponseHasFieldr^r[rbr:workflow_execution_started_event_link_from_workflow_handlerEoutbound_linksextend Exceptionloggerwarning)rOr`wf_event_linksr^er*r*r+_add_outbound_linkss4      z2_TemporalStartOperationContext._add_outbound_linksN)r,rDr,rK)r,rQ)r,rX)r`ra) r%r&r'r(r) classmethodr;r?rWr_rvr*r*r*r+rDzs      rDcseZdZdZdEfdd ZedFd d Zed d d d ej j j ej j j d dd d d d d d gid dej jjd ddGd6d7Zed d d d ej j j ej j j d dd d d d d d gid dej jjd ddHd;d7Zed d d d ej j j ej j j d dd d d d d d gid dej jjd ddId=d7Zeej jfgd d d d d ej j j ej j j d dd d d d d d gid dej jjd d>dJdAd7Zej jfgd d d d d ej j j ej j j d dd d d d d d gid dej jjd d>dKdDd7ZZS)LWorkflowRunOperationContextzsContext received by a workflow run operation. .. warning:: This API is experimental and unstable. argsr kwargsr,rKcs tj|i|t|_dS)z.Initialize the workflow run operation context.N)super__init__rDr;r1)rOrzr{ __class__r*r+r}sz$WorkflowRunOperationContext.__init__r9rcs"|difddtDS)Ncsi|] }|jt|jqSr*)namegetattr)rdfr8r*r+ szMWorkflowRunOperationContext._from_start_operation_context..r*) dataclassesfieldsrHr*r8r+_from_start_operation_contextsz9WorkflowRunOperationContext._from_start_operation_contextNF)r#execution_timeout run_timeout task_timeoutid_reuse_policyid_conflict_policy retry_policy cron_schedulememosearch_attributesstatic_summarystatic_details start_delay start_signalstart_signal_args rpc_metadata rpc_timeoutrequest_eager_startpriorityversioning_overrideworkflow(MethodAsyncNoParam[SelfType, ReturnType]idr"r# Optional[str]rOptional[timedelta]rrr'temporalio.common.WorkflowIDReusePolicyr*temporalio.common.WorkflowIDConflictPolicyr'Optional[temporalio.common.RetryPolicy]rrOptional[Mapping[str, Any]]r\Optional[Union[temporalio.common.TypedSearchAttributes, temporalio.common.SearchAttributes]]rrrrr Sequence[Any]rMapping[str, Union[str, bytes]]rrr-rtemporalio.common.Priorityr.Optional[temporalio.common.VersioningOverride]WorkflowHandle[ReturnType]cdSrMr*)rOrrr#rrrrrrrrrrrrrrrrrrrr*r*r+start_workflowsz*WorkflowRunOperationContext.start_workflow7MethodAsyncSingleParam[SelfType, ParamType, ReturnType]argrcrrMr*)rOrrrr#rrrrrrrrrrrrrrrrrrrr*r*r+rs FCallable[Concatenate[SelfType, MultiParamSpec], Awaitable[ReturnType]]crrMr*)rOrrzrr#rrrrrrrrrrrrrrrrrrrr*r*r+r"")rzr# result_typerrrrrrrrrrrrrrrrrrrrOptional[type[ReturnType]]crrMr*)rOrrrzrr#rrrrrrrrrrrrrrrrrrrrr*r*r+rGr0Union[str, Callable[..., Awaitable[ReturnType]]]Optional[type]cstv|jjjdid|d|d|d|d|p|jjd|d|d|d | d | d | d | d | d|d|d|d|d|d|d|d|d|d|d|d|d|jd|jd|jjj IdH}Wdn1s}wY|j |t t  |S)aStart a workflow that will deliver the result of the Nexus operation. The workflow will be started in the same namespace as the Nexus worker, using the same client as the worker. If task queue is not specified, the worker's task queue will be used. See :py:meth:`temporalio.client.Client.start_workflow` for all arguments. The return value is :py:class:`temporalio.nexus.WorkflowHandle`. The workflow will be started as usual, with the following modifications: - On workflow completion, Temporal server will deliver the workflow result to the Nexus operation caller, using the callback from the Nexus operation start request. - The request ID from the Nexus operation start request will be used as the request ID for the start workflow request. - Inbound links to the caller that were submitted in the Nexus start operation request will be attached to the started workflow and, outbound links to the started workflow will be added to the Nexus start operation response. If the Nexus caller is itself a workflow, this means that the workflow in the caller namespace web UI will contain links to the started workflow, and vice versa. rrrzrr#rrrrrrrrrrrrrrrrrrrr callbacksworkflow_event_links request_idNr*)rBr1r4rr2r#rWr_rErrvrr#_unsafe_from_client_workflow_handle)rOrrrzrr#rrrrrrrrrrrrrrrrrrrr wf_handler*r*r+rks~D          )rzr r{r r,rK)r9rr,ry).rrrr"r#rrrrrrrrrrrrrrr"rrrrrrrrrrrrrrrrrrrr-rrrrr,r)0rrrrrr"r#rrrrrrrrrrrrrrr"rrrrrrrrrrrrrrrrrrrr-rrrrr,r)0rrrzrrr"r#rrrrrrrrrrrrrrr"rrrrrrrrrrrrrrrrrrrr-rrrrr,r)4rr"rr rzrrr"r#rrrrrrrrrrrrrrrrr"rrrrrrrrrrrrrrrrrrrr-rrrrr,r)4rrrr rzrrr"r#rrrrrrrrrrrrrrrrr"rrrrrrrrrrrrrrrrrrrr-rrrrr,r)r%r&r'r(r}rxrrrhcommonWorkflowIDReusePolicyALLOW_DUPLICATEWorkflowIDConflictPolicy UNSPECIFIEDPrioritydefaultr _arg_unset __classcell__r*r*r~r+rys !" $&ryc@s$eZdZUdZded< ded<dS)rUzNexus callback to attach to events such as workflow completion. .. warning:: This API is experimental and unstable. r"rRzMapping[str, str]rSNr$r*r*r*r+rUs rUc@sHeZdZUdZded< ded< ded< edd d Zdd d ZdS)_TemporalCancelOperationContextzNContext for a Nexus cancel operation being handled by a Temporal Nexus Worker.rrErFr2r3r4r,cCrG)Nz&Not in Nexus cancel operation context.)rr;r7rHr*r*r+r;rJz#_TemporalCancelOperationContext.getrKcCrLrM)rr?rNr*r*r+r?rPz#_TemporalCancelOperationContext.setN)r,rrw)r%r&r'r(r)rxr;r?r*r*r*r+rs  rcs,eZdZdZdfdd ZdddZZS) LoggerAdapterz=Logger adapter that adds Nexus operation context information.rrlogging.Loggerextrarcst||pidS)zInitialize the logger adapter.N)r|r})rOrrrr~r*r+r}szLoggerAdapter.__init__msgr r{MutableMapping[str, Any]r,$tuple[Any, MutableMapping[str, Any]]cCsZt|jpi}t}r|jj|d<|jj|d<|j|d<||diB|d<||fS)z3Process log records to add Nexus operation context.service operationr#r) dictrr.rErrr2r#r;)rOrr{rtctxr*r*r+processs   zLoggerAdapter.process)rrrrr)rr r{rr,r)r%r&r'r(r}rrr*r*r~r+rsrztemporalio.nexus)r,r-)r,r!)r,r3)r,r5)r,r:)r,r>)@ __future__rrloggingcollections.abcrrrr contextlibr contextvarsrr datetimer typingr r r rrrrnexusrpc.handlerrrtyping_extensionsrtemporalio.api.common.v1rh!temporalio.api.workflowservice.v1temporalio.commontemporalio.nexusrtemporalio.nexus._tokenrtemporalio.typesrrrrrrtemporalio.clientrr)rrr!r/r2r4r1r.rBrCrDryrUrr getLoggerrrr*r*r*r+s`    $              M