Skip to content

Commit cfb9169

Browse files
author
Andrei Neagu
committed
fixed interface
1 parent d217db4 commit cfb9169

File tree

1 file changed

+18
-9
lines changed

1 file changed

+18
-9
lines changed

packages/service-library/src/servicelib/rabbitmq/rpc_interfaces/dynamic_sidecar/containers_long_running_tasks.py

Lines changed: 18 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,8 @@ async def pull_user_services_docker_images_task(
2828
),
2929
lrt_namespace=lrt_namespace,
3030
)
31-
return TaskId(result)
31+
assert isinstance(result, TaskId) # nosec
32+
return result
3233

3334

3435
@log_decorator(_logger, level=logging.DEBUG)
@@ -46,7 +47,8 @@ async def create_service_containers_task(
4647
lrt_namespace=lrt_namespace,
4748
containers_create=containers_create,
4849
)
49-
return TaskId(result)
50+
assert isinstance(result, TaskId) # nosec
51+
return result
5052

5153

5254
@log_decorator(_logger, level=logging.DEBUG)
@@ -62,7 +64,8 @@ async def runs_docker_compose_down_task(
6264
TypeAdapter(RPCMethodName).validate_python("runs_docker_compose_down_task"),
6365
lrt_namespace=lrt_namespace,
6466
)
65-
return TaskId(result)
67+
assert isinstance(result, TaskId) # nosec
68+
return result
6669

6770

6871
@log_decorator(_logger, level=logging.DEBUG)
@@ -78,7 +81,8 @@ async def state_restore_task(
7881
TypeAdapter(RPCMethodName).validate_python("state_restore_task"),
7982
lrt_namespace=lrt_namespace,
8083
)
81-
return TaskId(result)
84+
assert isinstance(result, TaskId) # nosec
85+
return result
8286

8387

8488
@log_decorator(_logger, level=logging.DEBUG)
@@ -94,7 +98,8 @@ async def state_save_task(
9498
TypeAdapter(RPCMethodName).validate_python("state_save_task"),
9599
lrt_namespace=lrt_namespace,
96100
)
97-
return TaskId(result)
101+
assert isinstance(result, TaskId) # nosec
102+
return result
98103

99104

100105
@log_decorator(_logger, level=logging.DEBUG)
@@ -112,7 +117,8 @@ async def ports_inputs_pull_task(
112117
lrt_namespace=lrt_namespace,
113118
port_keys=port_keys,
114119
)
115-
return TaskId(result)
120+
assert isinstance(result, TaskId) # nosec
121+
return result
116122

117123

118124
@log_decorator(_logger, level=logging.DEBUG)
@@ -130,7 +136,8 @@ async def ports_outputs_pull_task(
130136
lrt_namespace=lrt_namespace,
131137
port_keys=port_keys,
132138
)
133-
return TaskId(result)
139+
assert isinstance(result, TaskId) # nosec
140+
return result
134141

135142

136143
@log_decorator(_logger, level=logging.DEBUG)
@@ -146,7 +153,8 @@ async def ports_outputs_push_task(
146153
TypeAdapter(RPCMethodName).validate_python("ports_outputs_push_task"),
147154
lrt_namespace=lrt_namespace,
148155
)
149-
return TaskId(result)
156+
assert isinstance(result, TaskId) # nosec
157+
return result
150158

151159

152160
@log_decorator(_logger, level=logging.DEBUG)
@@ -162,4 +170,5 @@ async def containers_restart_task(
162170
TypeAdapter(RPCMethodName).validate_python("containers_restart_task"),
163171
lrt_namespace=lrt_namespace,
164172
)
165-
return TaskId(result)
173+
assert isinstance(result, TaskId) # nosec
174+
return result

0 commit comments

Comments
 (0)