diff --git a/context_propagation/interceptor.py b/context_propagation/interceptor.py index b90585484..19f0d3c9b 100644 --- a/context_propagation/interceptor.py +++ b/context_propagation/interceptor.py @@ -113,7 +113,7 @@ async def start_workflow_update( self, input: temporalio.client.StartWorkflowUpdateInput ) -> temporalio.client.WorkflowUpdateHandle[Any]: set_header_from_context(input, self._payload_converter) - return await self.next.start_workflow_update(input) + return await super().start_workflow_update(input) class _ContextPropagationActivityInboundInterceptor( @@ -123,40 +123,40 @@ async def execute_activity( self, input: temporalio.worker.ExecuteActivityInput ) -> Any: with context_from_header(input, temporalio.activity.payload_converter()): - return await self.next.execute_activity(input) + return await super().execute_activity(input) class _ContextPropagationWorkflowInboundInterceptor( temporalio.worker.WorkflowInboundInterceptor ): def init(self, outbound: temporalio.worker.WorkflowOutboundInterceptor) -> None: - self.next.init(_ContextPropagationWorkflowOutboundInterceptor(outbound)) + super().init(_ContextPropagationWorkflowOutboundInterceptor(outbound)) async def execute_workflow( self, input: temporalio.worker.ExecuteWorkflowInput ) -> Any: with context_from_header(input, temporalio.workflow.payload_converter()): - return await self.next.execute_workflow(input) + return await super().execute_workflow(input) async def handle_signal(self, input: temporalio.worker.HandleSignalInput) -> None: with context_from_header(input, temporalio.workflow.payload_converter()): - return await self.next.handle_signal(input) + return await super().handle_signal(input) async def handle_query(self, input: temporalio.worker.HandleQueryInput) -> Any: with context_from_header(input, temporalio.workflow.payload_converter()): - return await self.next.handle_query(input) + return await super().handle_query(input) def handle_update_validator( self, input: temporalio.worker.HandleUpdateInput ) -> None: with context_from_header(input, temporalio.workflow.payload_converter()): - self.next.handle_update_validator(input) + super().handle_update_validator(input) async def handle_update_handler( self, input: temporalio.worker.HandleUpdateInput ) -> Any: with context_from_header(input, temporalio.workflow.payload_converter()): - return await self.next.handle_update_handler(input) + return await super().handle_update_handler(input) class _ContextPropagationWorkflowOutboundInterceptor( @@ -166,28 +166,28 @@ async def signal_child_workflow( self, input: temporalio.worker.SignalChildWorkflowInput ) -> None: set_header_from_context(input, temporalio.workflow.payload_converter()) - return await self.next.signal_child_workflow(input) + return await super().signal_child_workflow(input) async def signal_external_workflow( self, input: temporalio.worker.SignalExternalWorkflowInput ) -> None: set_header_from_context(input, temporalio.workflow.payload_converter()) - return await self.next.signal_external_workflow(input) + return await super().signal_external_workflow(input) def start_activity( self, input: temporalio.worker.StartActivityInput ) -> temporalio.workflow.ActivityHandle: set_header_from_context(input, temporalio.workflow.payload_converter()) - return self.next.start_activity(input) + return super().start_activity(input) async def start_child_workflow( self, input: temporalio.worker.StartChildWorkflowInput ) -> temporalio.workflow.ChildWorkflowHandle: set_header_from_context(input, temporalio.workflow.payload_converter()) - return await self.next.start_child_workflow(input) + return await super().start_child_workflow(input) def start_local_activity( self, input: temporalio.worker.StartLocalActivityInput ) -> temporalio.workflow.ActivityHandle: set_header_from_context(input, temporalio.workflow.payload_converter()) - return self.next.start_local_activity(input) + return super().start_local_activity(input)