o iy@sddlmZddlmZmZmZddlmZmZm Z m Z m Z ddl m Z mZmZmZmZmZddlmZddlmZddlmZGd d d ee e fZdddZdS)) annotations)Any AwaitableCallable) HandlerErrorHandlerErrorTypeInputT OperationInfoOutputT)CancelOperationContextFetchOperationInfoContextFetchOperationResultContextOperationHandlerStartOperationContextStartOperationResultAsync)"_temporal_cancel_operation_context)WorkflowHandle)is_async_callablec@sBeZdZdZdddZdd dZdddZd ddZd!ddZdS)"WorkflowRunOperationHandleraJOperation handler for Nexus operations that start a workflow. Use this class to create an operation handler that starts a workflow by passing your ``start`` method to the constructor. Your ``start`` method must use :py:func:`temporalio.nexus.WorkflowRunOperationContext.start_workflow` to start the workflow. startMCallable[[StartOperationContext, InputT], Awaitable[WorkflowHandle[OutputT]]]returnNonecCsHt|s t|d||_|jr t|jdd}r"|j|_dSdSdS)z.Initialize the workflow run operation handler.zp is not an `async def` method. WorkflowRunOperationHandler must be initialized with an `async def` start method.__func__N)r RuntimeError_start__doc__getattrr)selfr start_funcr!j/var/www/html/karishye-ai-python/venv/lib/python3.10/site-packages/temporalio/nexus/_operation_handlers.py__init__,s z$WorkflowRunOperationHandler.__init__ctxrinputrrcsD|||IdH}t|tstd|dt|dt|S)zJStart the operation, by starting a workflow and completing asynchronously.Nz Expected z' to be a nexus.WorkflowHandle, but got aJ. When using @workflow_run_operation you must use WorkflowRunOperationContext.start_workflow() to start a workflow that will deliver the result of the Nexus operation, and you must return the nexus.WorkflowHandle that it returns. It is not possible to use client.Client.start_workflow() and client.WorkflowHandle for this purpose.)r isinstancerrtyperto_token)rr$r%handler!r!r"r?s  z!WorkflowRunOperationHandler.startr tokenstrcst|IdHdS)z1Cancel the operation, by cancelling the workflow.N)_cancel_workflowrr$r*r!r!r"cancelPsz"WorkflowRunOperationHandler.cancelr r c td)zCFetch operation info (not supported for Temporal Nexus operations).zITemporal Nexus operation handlers do not support fetching operation info.NotImplementedErrorr-r!r!r" fetch_infoTz&WorkflowRunOperationHandler.fetch_infor r cr/)zEFetch operation result (not supported for Temporal Nexus operations).zOTemporal Nexus operation handlers do not support fetching the operation result.r0r-r!r!r" fetch_result\r3z(WorkflowRunOperationHandler.fetch_resultN)rrrr)r$rr%rrr)r$r r*r+rr)r$r r*r+rr )r$r r*r+rr ) __name__ __module__ __qualname__rr#rr.r2r4r!r!r!r"r#s    rr*r+kwargsrrrc sz tt|}Wnty}ztdtjd|d}~wwt}z| |j }Wnty?}ztdtjd|d}~ww|j di|IdHdS)aCancel a workflow that is backing a Nexus operation. This function is used by the Nexus worker to cancel a workflow that is backing a Nexus operation, i.e. started by a :py:func:`temporalio.nexus.workflow_run_operation`-decorated method. Args: token: The token of the workflow to cancel. kwargs: Additional keyword arguments to pass to the workflow cancel method. zsFailed to decode operation token as a workflow operation token. Canceling non-workflow operations is not supported.)r'NzAFailed to construct workflow handle from workflow operation tokenr!) rr from_token Exceptionrr NOT_FOUNDrget_to_client_workflow_handleclientr.)r*r8nexus_workflow_handleerrr$client_workflow_handler!r!r"r,es6r,N)r*r+r8rrr) __future__rtypingrrrnexusrpcrrrr r nexusrpc.handlerr r r rrr#temporalio.nexus._operation_contextrtemporalio.nexus._tokenr_utilrrr,r!r!r!r"s    B