Skip to content

Commit f4f446e

Browse files
committed
Rename interceptor input dataclasses
1 parent 48face5 commit f4f446e

File tree

1 file changed

+9
-13
lines changed

1 file changed

+9
-13
lines changed

temporalio/client.py

Lines changed: 9 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -975,7 +975,7 @@ async def _start_update_with_start(
975975
else:
976976
update_name = str(update)
977977

978-
update_input = StartWorkflowUpdateWithStartUpdateWorkflowInput(
978+
update_input = UpdateWithStartUpdateWorkflowInput(
979979
id=start_workflow_operation._start_workflow_input.id,
980980
update_id=id,
981981
update=update_name,
@@ -2476,7 +2476,7 @@ def __init__(
24762476

24772477
self._id = id
24782478
self._workflow_handle: Future[WorkflowHandle[SelfType, ReturnType]] = Future()
2479-
self._start_workflow_input = StartWorkflowUpdateWithStartStartWorkflowInput(
2479+
self._start_workflow_input = UpdateWithStartStartWorkflowInput(
24802480
workflow=name,
24812481
args=temporalio.common._arg_or_args(arg, args),
24822482
id=id,
@@ -5073,7 +5073,7 @@ class StartWorkflowUpdateInput:
50735073

50745074

50755075
@dataclass
5076-
class StartWorkflowUpdateWithStartUpdateWorkflowInput:
5076+
class UpdateWithStartUpdateWorkflowInput:
50775077
"""Update input for :py:meth:`OutboundInterceptor.start_workflow_update_with_start`."""
50785078

50795079
id: str
@@ -5089,7 +5089,7 @@ class StartWorkflowUpdateWithStartUpdateWorkflowInput:
50895089

50905090
# TODO: name
50915091
@dataclass
5092-
class StartWorkflowUpdateWithStartStartWorkflowInput:
5092+
class UpdateWithStartStartWorkflowInput:
50935093
"""StartWorkflow input for :py:meth:`OutboundInterceptor.start_workflow_update_with_start`."""
50945094

50955095
# Similar to StartWorkflowInput but without e.g. run_id, start_signal,
@@ -5124,8 +5124,8 @@ class StartWorkflowUpdateWithStartStartWorkflowInput:
51245124
class StartWorkflowUpdateWithStartInput:
51255125
"""Input for :py:meth:`OutboundInterceptor.start_workflow_update_with_start`."""
51265126

5127-
update_workflow_input: StartWorkflowUpdateWithStartUpdateWorkflowInput
5128-
start_workflow_input: StartWorkflowUpdateWithStartStartWorkflowInput
5127+
update_workflow_input: UpdateWithStartUpdateWorkflowInput
5128+
start_workflow_input: UpdateWithStartStartWorkflowInput
51295129

51305130

51315131
@dataclass
@@ -5565,7 +5565,7 @@ async def _build_signal_with_start_workflow_execution_request(
55655565
return req
55665566

55675567
async def _build_update_with_start_start_workflow_execution_request(
5568-
self, input: StartWorkflowUpdateWithStartStartWorkflowInput
5568+
self, input: UpdateWithStartStartWorkflowInput
55695569
) -> temporalio.api.workflowservice.v1.StartWorkflowExecutionRequest:
55705570
req = temporalio.api.workflowservice.v1.StartWorkflowExecutionRequest()
55715571
await self._populate_start_workflow_execution_request(req, input)
@@ -5577,9 +5577,7 @@ async def _populate_start_workflow_execution_request(
55775577
temporalio.api.workflowservice.v1.StartWorkflowExecutionRequest,
55785578
temporalio.api.workflowservice.v1.SignalWithStartWorkflowExecutionRequest,
55795579
],
5580-
input: Union[
5581-
StartWorkflowInput, StartWorkflowUpdateWithStartStartWorkflowInput
5582-
],
5580+
input: Union[StartWorkflowInput, UpdateWithStartStartWorkflowInput],
55835581
) -> None:
55845582
req.namespace = self._client.namespace
55855583
req.workflow_id = input.id
@@ -5822,9 +5820,7 @@ async def start_workflow_update(
58225820

58235821
async def _build_update_workflow_execution_request(
58245822
self,
5825-
input: Union[
5826-
StartWorkflowUpdateInput, StartWorkflowUpdateWithStartUpdateWorkflowInput
5827-
],
5823+
input: Union[StartWorkflowUpdateInput, UpdateWithStartUpdateWorkflowInput],
58285824
) -> temporalio.api.workflowservice.v1.UpdateWorkflowExecutionRequest:
58295825
run_id = input.run_id if isinstance(input, StartWorkflowUpdateInput) else None
58305826
req = temporalio.api.workflowservice.v1.UpdateWorkflowExecutionRequest(

0 commit comments

Comments
 (0)