flyte 0.0.1b0__py3-none-any.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Potentially problematic release.
This version of flyte might be problematic. Click here for more details.
- flyte/__init__.py +62 -0
- flyte/_api_commons.py +3 -0
- flyte/_bin/__init__.py +0 -0
- flyte/_bin/runtime.py +126 -0
- flyte/_build.py +25 -0
- flyte/_cache/__init__.py +12 -0
- flyte/_cache/cache.py +146 -0
- flyte/_cache/defaults.py +9 -0
- flyte/_cache/policy_function_body.py +42 -0
- flyte/_cli/__init__.py +0 -0
- flyte/_cli/_common.py +287 -0
- flyte/_cli/_create.py +42 -0
- flyte/_cli/_delete.py +23 -0
- flyte/_cli/_deploy.py +140 -0
- flyte/_cli/_get.py +235 -0
- flyte/_cli/_run.py +152 -0
- flyte/_cli/main.py +72 -0
- flyte/_code_bundle/__init__.py +8 -0
- flyte/_code_bundle/_ignore.py +113 -0
- flyte/_code_bundle/_packaging.py +187 -0
- flyte/_code_bundle/_utils.py +339 -0
- flyte/_code_bundle/bundle.py +178 -0
- flyte/_context.py +146 -0
- flyte/_datastructures.py +342 -0
- flyte/_deploy.py +202 -0
- flyte/_doc.py +29 -0
- flyte/_docstring.py +32 -0
- flyte/_environment.py +43 -0
- flyte/_group.py +31 -0
- flyte/_hash.py +23 -0
- flyte/_image.py +760 -0
- flyte/_initialize.py +634 -0
- flyte/_interface.py +84 -0
- flyte/_internal/__init__.py +3 -0
- flyte/_internal/controllers/__init__.py +115 -0
- flyte/_internal/controllers/_local_controller.py +118 -0
- flyte/_internal/controllers/_trace.py +40 -0
- flyte/_internal/controllers/pbhash.py +39 -0
- flyte/_internal/controllers/remote/__init__.py +40 -0
- flyte/_internal/controllers/remote/_action.py +141 -0
- flyte/_internal/controllers/remote/_client.py +43 -0
- flyte/_internal/controllers/remote/_controller.py +361 -0
- flyte/_internal/controllers/remote/_core.py +402 -0
- flyte/_internal/controllers/remote/_informer.py +361 -0
- flyte/_internal/controllers/remote/_service_protocol.py +50 -0
- flyte/_internal/imagebuild/__init__.py +11 -0
- flyte/_internal/imagebuild/docker_builder.py +416 -0
- flyte/_internal/imagebuild/image_builder.py +241 -0
- flyte/_internal/imagebuild/remote_builder.py +0 -0
- flyte/_internal/resolvers/__init__.py +0 -0
- flyte/_internal/resolvers/_task_module.py +54 -0
- flyte/_internal/resolvers/common.py +31 -0
- flyte/_internal/resolvers/default.py +28 -0
- flyte/_internal/runtime/__init__.py +0 -0
- flyte/_internal/runtime/convert.py +199 -0
- flyte/_internal/runtime/entrypoints.py +135 -0
- flyte/_internal/runtime/io.py +136 -0
- flyte/_internal/runtime/resources_serde.py +138 -0
- flyte/_internal/runtime/task_serde.py +210 -0
- flyte/_internal/runtime/taskrunner.py +190 -0
- flyte/_internal/runtime/types_serde.py +54 -0
- flyte/_logging.py +124 -0
- flyte/_protos/__init__.py +0 -0
- flyte/_protos/common/authorization_pb2.py +66 -0
- flyte/_protos/common/authorization_pb2.pyi +108 -0
- flyte/_protos/common/authorization_pb2_grpc.py +4 -0
- flyte/_protos/common/identifier_pb2.py +71 -0
- flyte/_protos/common/identifier_pb2.pyi +82 -0
- flyte/_protos/common/identifier_pb2_grpc.py +4 -0
- flyte/_protos/common/identity_pb2.py +48 -0
- flyte/_protos/common/identity_pb2.pyi +72 -0
- flyte/_protos/common/identity_pb2_grpc.py +4 -0
- flyte/_protos/common/list_pb2.py +36 -0
- flyte/_protos/common/list_pb2.pyi +69 -0
- flyte/_protos/common/list_pb2_grpc.py +4 -0
- flyte/_protos/common/policy_pb2.py +37 -0
- flyte/_protos/common/policy_pb2.pyi +27 -0
- flyte/_protos/common/policy_pb2_grpc.py +4 -0
- flyte/_protos/common/role_pb2.py +37 -0
- flyte/_protos/common/role_pb2.pyi +53 -0
- flyte/_protos/common/role_pb2_grpc.py +4 -0
- flyte/_protos/common/runtime_version_pb2.py +28 -0
- flyte/_protos/common/runtime_version_pb2.pyi +24 -0
- flyte/_protos/common/runtime_version_pb2_grpc.py +4 -0
- flyte/_protos/logs/dataplane/payload_pb2.py +96 -0
- flyte/_protos/logs/dataplane/payload_pb2.pyi +168 -0
- flyte/_protos/logs/dataplane/payload_pb2_grpc.py +4 -0
- flyte/_protos/secret/definition_pb2.py +49 -0
- flyte/_protos/secret/definition_pb2.pyi +93 -0
- flyte/_protos/secret/definition_pb2_grpc.py +4 -0
- flyte/_protos/secret/payload_pb2.py +62 -0
- flyte/_protos/secret/payload_pb2.pyi +94 -0
- flyte/_protos/secret/payload_pb2_grpc.py +4 -0
- flyte/_protos/secret/secret_pb2.py +38 -0
- flyte/_protos/secret/secret_pb2.pyi +6 -0
- flyte/_protos/secret/secret_pb2_grpc.py +198 -0
- flyte/_protos/secret/secret_pb2_grpc_grpc.py +198 -0
- flyte/_protos/validate/validate/validate_pb2.py +76 -0
- flyte/_protos/workflow/node_execution_service_pb2.py +26 -0
- flyte/_protos/workflow/node_execution_service_pb2.pyi +4 -0
- flyte/_protos/workflow/node_execution_service_pb2_grpc.py +32 -0
- flyte/_protos/workflow/queue_service_pb2.py +106 -0
- flyte/_protos/workflow/queue_service_pb2.pyi +141 -0
- flyte/_protos/workflow/queue_service_pb2_grpc.py +172 -0
- flyte/_protos/workflow/run_definition_pb2.py +128 -0
- flyte/_protos/workflow/run_definition_pb2.pyi +310 -0
- flyte/_protos/workflow/run_definition_pb2_grpc.py +4 -0
- flyte/_protos/workflow/run_logs_service_pb2.py +41 -0
- flyte/_protos/workflow/run_logs_service_pb2.pyi +28 -0
- flyte/_protos/workflow/run_logs_service_pb2_grpc.py +69 -0
- flyte/_protos/workflow/run_service_pb2.py +133 -0
- flyte/_protos/workflow/run_service_pb2.pyi +175 -0
- flyte/_protos/workflow/run_service_pb2_grpc.py +412 -0
- flyte/_protos/workflow/state_service_pb2.py +58 -0
- flyte/_protos/workflow/state_service_pb2.pyi +71 -0
- flyte/_protos/workflow/state_service_pb2_grpc.py +138 -0
- flyte/_protos/workflow/task_definition_pb2.py +72 -0
- flyte/_protos/workflow/task_definition_pb2.pyi +65 -0
- flyte/_protos/workflow/task_definition_pb2_grpc.py +4 -0
- flyte/_protos/workflow/task_service_pb2.py +44 -0
- flyte/_protos/workflow/task_service_pb2.pyi +31 -0
- flyte/_protos/workflow/task_service_pb2_grpc.py +104 -0
- flyte/_resources.py +226 -0
- flyte/_retry.py +32 -0
- flyte/_reusable_environment.py +25 -0
- flyte/_run.py +411 -0
- flyte/_secret.py +61 -0
- flyte/_task.py +367 -0
- flyte/_task_environment.py +200 -0
- flyte/_timeout.py +47 -0
- flyte/_tools.py +27 -0
- flyte/_trace.py +128 -0
- flyte/_utils/__init__.py +20 -0
- flyte/_utils/asyn.py +119 -0
- flyte/_utils/coro_management.py +25 -0
- flyte/_utils/file_handling.py +72 -0
- flyte/_utils/helpers.py +108 -0
- flyte/_utils/lazy_module.py +54 -0
- flyte/_utils/uv_script_parser.py +49 -0
- flyte/_version.py +21 -0
- flyte/connectors/__init__.py +0 -0
- flyte/errors.py +143 -0
- flyte/extras/__init__.py +5 -0
- flyte/extras/_container.py +273 -0
- flyte/io/__init__.py +11 -0
- flyte/io/_dataframe.py +0 -0
- flyte/io/_dir.py +448 -0
- flyte/io/_file.py +468 -0
- flyte/io/pickle/__init__.py +0 -0
- flyte/io/pickle/transformer.py +117 -0
- flyte/io/structured_dataset/__init__.py +129 -0
- flyte/io/structured_dataset/basic_dfs.py +219 -0
- flyte/io/structured_dataset/structured_dataset.py +1061 -0
- flyte/py.typed +0 -0
- flyte/remote/__init__.py +25 -0
- flyte/remote/_client/__init__.py +0 -0
- flyte/remote/_client/_protocols.py +131 -0
- flyte/remote/_client/auth/__init__.py +12 -0
- flyte/remote/_client/auth/_authenticators/__init__.py +0 -0
- flyte/remote/_client/auth/_authenticators/base.py +397 -0
- flyte/remote/_client/auth/_authenticators/client_credentials.py +73 -0
- flyte/remote/_client/auth/_authenticators/device_code.py +118 -0
- flyte/remote/_client/auth/_authenticators/external_command.py +79 -0
- flyte/remote/_client/auth/_authenticators/factory.py +200 -0
- flyte/remote/_client/auth/_authenticators/pkce.py +516 -0
- flyte/remote/_client/auth/_channel.py +184 -0
- flyte/remote/_client/auth/_client_config.py +83 -0
- flyte/remote/_client/auth/_default_html.py +32 -0
- flyte/remote/_client/auth/_grpc_utils/__init__.py +0 -0
- flyte/remote/_client/auth/_grpc_utils/auth_interceptor.py +288 -0
- flyte/remote/_client/auth/_grpc_utils/default_metadata_interceptor.py +151 -0
- flyte/remote/_client/auth/_keyring.py +143 -0
- flyte/remote/_client/auth/_token_client.py +260 -0
- flyte/remote/_client/auth/errors.py +16 -0
- flyte/remote/_client/controlplane.py +95 -0
- flyte/remote/_console.py +18 -0
- flyte/remote/_data.py +155 -0
- flyte/remote/_logs.py +116 -0
- flyte/remote/_project.py +86 -0
- flyte/remote/_run.py +873 -0
- flyte/remote/_secret.py +132 -0
- flyte/remote/_task.py +227 -0
- flyte/report/__init__.py +3 -0
- flyte/report/_report.py +178 -0
- flyte/report/_template.html +124 -0
- flyte/storage/__init__.py +24 -0
- flyte/storage/_remote_fs.py +34 -0
- flyte/storage/_storage.py +251 -0
- flyte/storage/_utils.py +5 -0
- flyte/types/__init__.py +13 -0
- flyte/types/_interface.py +25 -0
- flyte/types/_renderer.py +162 -0
- flyte/types/_string_literals.py +120 -0
- flyte/types/_type_engine.py +2210 -0
- flyte/types/_utils.py +80 -0
- flyte-0.0.1b0.dist-info/METADATA +179 -0
- flyte-0.0.1b0.dist-info/RECORD +390 -0
- flyte-0.0.1b0.dist-info/WHEEL +5 -0
- flyte-0.0.1b0.dist-info/entry_points.txt +3 -0
- flyte-0.0.1b0.dist-info/top_level.txt +1 -0
- union/__init__.py +54 -0
- union/_api_commons.py +3 -0
- union/_bin/__init__.py +0 -0
- union/_bin/runtime.py +113 -0
- union/_build.py +25 -0
- union/_cache/__init__.py +12 -0
- union/_cache/cache.py +141 -0
- union/_cache/defaults.py +9 -0
- union/_cache/policy_function_body.py +42 -0
- union/_cli/__init__.py +0 -0
- union/_cli/_common.py +263 -0
- union/_cli/_create.py +40 -0
- union/_cli/_delete.py +23 -0
- union/_cli/_deploy.py +120 -0
- union/_cli/_get.py +162 -0
- union/_cli/_params.py +579 -0
- union/_cli/_run.py +150 -0
- union/_cli/main.py +72 -0
- union/_code_bundle/__init__.py +8 -0
- union/_code_bundle/_ignore.py +113 -0
- union/_code_bundle/_packaging.py +187 -0
- union/_code_bundle/_utils.py +342 -0
- union/_code_bundle/bundle.py +176 -0
- union/_context.py +146 -0
- union/_datastructures.py +295 -0
- union/_deploy.py +185 -0
- union/_doc.py +29 -0
- union/_docstring.py +26 -0
- union/_environment.py +43 -0
- union/_group.py +31 -0
- union/_hash.py +23 -0
- union/_image.py +760 -0
- union/_initialize.py +585 -0
- union/_interface.py +84 -0
- union/_internal/__init__.py +3 -0
- union/_internal/controllers/__init__.py +77 -0
- union/_internal/controllers/_local_controller.py +77 -0
- union/_internal/controllers/pbhash.py +39 -0
- union/_internal/controllers/remote/__init__.py +40 -0
- union/_internal/controllers/remote/_action.py +131 -0
- union/_internal/controllers/remote/_client.py +43 -0
- union/_internal/controllers/remote/_controller.py +169 -0
- union/_internal/controllers/remote/_core.py +341 -0
- union/_internal/controllers/remote/_informer.py +260 -0
- union/_internal/controllers/remote/_service_protocol.py +44 -0
- union/_internal/imagebuild/__init__.py +11 -0
- union/_internal/imagebuild/docker_builder.py +416 -0
- union/_internal/imagebuild/image_builder.py +243 -0
- union/_internal/imagebuild/remote_builder.py +0 -0
- union/_internal/resolvers/__init__.py +0 -0
- union/_internal/resolvers/_task_module.py +31 -0
- union/_internal/resolvers/common.py +24 -0
- union/_internal/resolvers/default.py +27 -0
- union/_internal/runtime/__init__.py +0 -0
- union/_internal/runtime/convert.py +163 -0
- union/_internal/runtime/entrypoints.py +121 -0
- union/_internal/runtime/io.py +136 -0
- union/_internal/runtime/resources_serde.py +134 -0
- union/_internal/runtime/task_serde.py +202 -0
- union/_internal/runtime/taskrunner.py +179 -0
- union/_internal/runtime/types_serde.py +53 -0
- union/_logging.py +124 -0
- union/_protos/__init__.py +0 -0
- union/_protos/common/authorization_pb2.py +66 -0
- union/_protos/common/authorization_pb2.pyi +106 -0
- union/_protos/common/authorization_pb2_grpc.py +4 -0
- union/_protos/common/identifier_pb2.py +71 -0
- union/_protos/common/identifier_pb2.pyi +82 -0
- union/_protos/common/identifier_pb2_grpc.py +4 -0
- union/_protos/common/identity_pb2.py +48 -0
- union/_protos/common/identity_pb2.pyi +72 -0
- union/_protos/common/identity_pb2_grpc.py +4 -0
- union/_protos/common/list_pb2.py +36 -0
- union/_protos/common/list_pb2.pyi +69 -0
- union/_protos/common/list_pb2_grpc.py +4 -0
- union/_protos/common/policy_pb2.py +37 -0
- union/_protos/common/policy_pb2.pyi +27 -0
- union/_protos/common/policy_pb2_grpc.py +4 -0
- union/_protos/common/role_pb2.py +37 -0
- union/_protos/common/role_pb2.pyi +51 -0
- union/_protos/common/role_pb2_grpc.py +4 -0
- union/_protos/common/runtime_version_pb2.py +28 -0
- union/_protos/common/runtime_version_pb2.pyi +24 -0
- union/_protos/common/runtime_version_pb2_grpc.py +4 -0
- union/_protos/logs/dataplane/payload_pb2.py +96 -0
- union/_protos/logs/dataplane/payload_pb2.pyi +168 -0
- union/_protos/logs/dataplane/payload_pb2_grpc.py +4 -0
- union/_protos/secret/definition_pb2.py +49 -0
- union/_protos/secret/definition_pb2.pyi +93 -0
- union/_protos/secret/definition_pb2_grpc.py +4 -0
- union/_protos/secret/payload_pb2.py +62 -0
- union/_protos/secret/payload_pb2.pyi +94 -0
- union/_protos/secret/payload_pb2_grpc.py +4 -0
- union/_protos/secret/secret_pb2.py +38 -0
- union/_protos/secret/secret_pb2.pyi +6 -0
- union/_protos/secret/secret_pb2_grpc.py +198 -0
- union/_protos/validate/validate/validate_pb2.py +76 -0
- union/_protos/workflow/node_execution_service_pb2.py +26 -0
- union/_protos/workflow/node_execution_service_pb2.pyi +4 -0
- union/_protos/workflow/node_execution_service_pb2_grpc.py +32 -0
- union/_protos/workflow/queue_service_pb2.py +75 -0
- union/_protos/workflow/queue_service_pb2.pyi +103 -0
- union/_protos/workflow/queue_service_pb2_grpc.py +172 -0
- union/_protos/workflow/run_definition_pb2.py +100 -0
- union/_protos/workflow/run_definition_pb2.pyi +256 -0
- union/_protos/workflow/run_definition_pb2_grpc.py +4 -0
- union/_protos/workflow/run_logs_service_pb2.py +41 -0
- union/_protos/workflow/run_logs_service_pb2.pyi +28 -0
- union/_protos/workflow/run_logs_service_pb2_grpc.py +69 -0
- union/_protos/workflow/run_service_pb2.py +133 -0
- union/_protos/workflow/run_service_pb2.pyi +173 -0
- union/_protos/workflow/run_service_pb2_grpc.py +412 -0
- union/_protos/workflow/state_service_pb2.py +58 -0
- union/_protos/workflow/state_service_pb2.pyi +69 -0
- union/_protos/workflow/state_service_pb2_grpc.py +138 -0
- union/_protos/workflow/task_definition_pb2.py +72 -0
- union/_protos/workflow/task_definition_pb2.pyi +65 -0
- union/_protos/workflow/task_definition_pb2_grpc.py +4 -0
- union/_protos/workflow/task_service_pb2.py +44 -0
- union/_protos/workflow/task_service_pb2.pyi +31 -0
- union/_protos/workflow/task_service_pb2_grpc.py +104 -0
- union/_resources.py +226 -0
- union/_retry.py +32 -0
- union/_reusable_environment.py +25 -0
- union/_run.py +374 -0
- union/_secret.py +61 -0
- union/_task.py +354 -0
- union/_task_environment.py +186 -0
- union/_timeout.py +47 -0
- union/_tools.py +27 -0
- union/_utils/__init__.py +11 -0
- union/_utils/asyn.py +119 -0
- union/_utils/file_handling.py +71 -0
- union/_utils/helpers.py +46 -0
- union/_utils/lazy_module.py +54 -0
- union/_utils/uv_script_parser.py +49 -0
- union/_version.py +21 -0
- union/connectors/__init__.py +0 -0
- union/errors.py +128 -0
- union/extras/__init__.py +5 -0
- union/extras/_container.py +263 -0
- union/io/__init__.py +11 -0
- union/io/_dataframe.py +0 -0
- union/io/_dir.py +425 -0
- union/io/_file.py +418 -0
- union/io/pickle/__init__.py +0 -0
- union/io/pickle/transformer.py +117 -0
- union/io/structured_dataset/__init__.py +122 -0
- union/io/structured_dataset/basic_dfs.py +219 -0
- union/io/structured_dataset/structured_dataset.py +1057 -0
- union/py.typed +0 -0
- union/remote/__init__.py +23 -0
- union/remote/_client/__init__.py +0 -0
- union/remote/_client/_protocols.py +129 -0
- union/remote/_client/auth/__init__.py +12 -0
- union/remote/_client/auth/_authenticators/__init__.py +0 -0
- union/remote/_client/auth/_authenticators/base.py +391 -0
- union/remote/_client/auth/_authenticators/client_credentials.py +73 -0
- union/remote/_client/auth/_authenticators/device_code.py +120 -0
- union/remote/_client/auth/_authenticators/external_command.py +77 -0
- union/remote/_client/auth/_authenticators/factory.py +200 -0
- union/remote/_client/auth/_authenticators/pkce.py +515 -0
- union/remote/_client/auth/_channel.py +184 -0
- union/remote/_client/auth/_client_config.py +83 -0
- union/remote/_client/auth/_default_html.py +32 -0
- union/remote/_client/auth/_grpc_utils/__init__.py +0 -0
- union/remote/_client/auth/_grpc_utils/auth_interceptor.py +204 -0
- union/remote/_client/auth/_grpc_utils/default_metadata_interceptor.py +144 -0
- union/remote/_client/auth/_keyring.py +154 -0
- union/remote/_client/auth/_token_client.py +258 -0
- union/remote/_client/auth/errors.py +16 -0
- union/remote/_client/controlplane.py +86 -0
- union/remote/_data.py +149 -0
- union/remote/_logs.py +74 -0
- union/remote/_project.py +86 -0
- union/remote/_run.py +820 -0
- union/remote/_secret.py +132 -0
- union/remote/_task.py +193 -0
- union/report/__init__.py +3 -0
- union/report/_report.py +178 -0
- union/report/_template.html +124 -0
- union/storage/__init__.py +24 -0
- union/storage/_remote_fs.py +34 -0
- union/storage/_storage.py +247 -0
- union/storage/_utils.py +5 -0
- union/types/__init__.py +11 -0
- union/types/_renderer.py +162 -0
- union/types/_string_literals.py +120 -0
- union/types/_type_engine.py +2131 -0
- union/types/_utils.py +80 -0
|
@@ -0,0 +1,138 @@
|
|
|
1
|
+
# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
|
|
2
|
+
"""Client and server classes corresponding to protobuf-defined services."""
|
|
3
|
+
import grpc
|
|
4
|
+
|
|
5
|
+
from union._protos.workflow import state_service_pb2 as workflow_dot_state__service__pb2
|
|
6
|
+
|
|
7
|
+
|
|
8
|
+
class StateServiceStub(object):
|
|
9
|
+
"""provides an interface for managing the state of actions.
|
|
10
|
+
"""
|
|
11
|
+
|
|
12
|
+
def __init__(self, channel):
|
|
13
|
+
"""Constructor.
|
|
14
|
+
|
|
15
|
+
Args:
|
|
16
|
+
channel: A grpc.Channel.
|
|
17
|
+
"""
|
|
18
|
+
self.Store = channel.unary_unary(
|
|
19
|
+
'/cloudidl.workflow.StateService/Store',
|
|
20
|
+
request_serializer=workflow_dot_state__service__pb2.StoreRequest.SerializeToString,
|
|
21
|
+
response_deserializer=workflow_dot_state__service__pb2.StoreResponse.FromString,
|
|
22
|
+
)
|
|
23
|
+
self.Load = channel.unary_unary(
|
|
24
|
+
'/cloudidl.workflow.StateService/Load',
|
|
25
|
+
request_serializer=workflow_dot_state__service__pb2.LoadRequest.SerializeToString,
|
|
26
|
+
response_deserializer=workflow_dot_state__service__pb2.LoadResponse.FromString,
|
|
27
|
+
)
|
|
28
|
+
self.Watch = channel.unary_stream(
|
|
29
|
+
'/cloudidl.workflow.StateService/Watch',
|
|
30
|
+
request_serializer=workflow_dot_state__service__pb2.WatchRequest.SerializeToString,
|
|
31
|
+
response_deserializer=workflow_dot_state__service__pb2.WatchResponse.FromString,
|
|
32
|
+
)
|
|
33
|
+
|
|
34
|
+
|
|
35
|
+
class StateServiceServicer(object):
|
|
36
|
+
"""provides an interface for managing the state of actions.
|
|
37
|
+
"""
|
|
38
|
+
|
|
39
|
+
def Store(self, request, context):
|
|
40
|
+
"""store the state of an action.
|
|
41
|
+
"""
|
|
42
|
+
context.set_code(grpc.StatusCode.UNIMPLEMENTED)
|
|
43
|
+
context.set_details('Method not implemented!')
|
|
44
|
+
raise NotImplementedError('Method not implemented!')
|
|
45
|
+
|
|
46
|
+
def Load(self, request, context):
|
|
47
|
+
"""load the state of an action.
|
|
48
|
+
"""
|
|
49
|
+
context.set_code(grpc.StatusCode.UNIMPLEMENTED)
|
|
50
|
+
context.set_details('Method not implemented!')
|
|
51
|
+
raise NotImplementedError('Method not implemented!')
|
|
52
|
+
|
|
53
|
+
def Watch(self, request, context):
|
|
54
|
+
"""watch for updates to the state of actions. this api guarantees at-least-once delivery semantics.
|
|
55
|
+
"""
|
|
56
|
+
context.set_code(grpc.StatusCode.UNIMPLEMENTED)
|
|
57
|
+
context.set_details('Method not implemented!')
|
|
58
|
+
raise NotImplementedError('Method not implemented!')
|
|
59
|
+
|
|
60
|
+
|
|
61
|
+
def add_StateServiceServicer_to_server(servicer, server):
|
|
62
|
+
rpc_method_handlers = {
|
|
63
|
+
'Store': grpc.unary_unary_rpc_method_handler(
|
|
64
|
+
servicer.Store,
|
|
65
|
+
request_deserializer=workflow_dot_state__service__pb2.StoreRequest.FromString,
|
|
66
|
+
response_serializer=workflow_dot_state__service__pb2.StoreResponse.SerializeToString,
|
|
67
|
+
),
|
|
68
|
+
'Load': grpc.unary_unary_rpc_method_handler(
|
|
69
|
+
servicer.Load,
|
|
70
|
+
request_deserializer=workflow_dot_state__service__pb2.LoadRequest.FromString,
|
|
71
|
+
response_serializer=workflow_dot_state__service__pb2.LoadResponse.SerializeToString,
|
|
72
|
+
),
|
|
73
|
+
'Watch': grpc.unary_stream_rpc_method_handler(
|
|
74
|
+
servicer.Watch,
|
|
75
|
+
request_deserializer=workflow_dot_state__service__pb2.WatchRequest.FromString,
|
|
76
|
+
response_serializer=workflow_dot_state__service__pb2.WatchResponse.SerializeToString,
|
|
77
|
+
),
|
|
78
|
+
}
|
|
79
|
+
generic_handler = grpc.method_handlers_generic_handler(
|
|
80
|
+
'cloudidl.workflow.StateService', rpc_method_handlers)
|
|
81
|
+
server.add_generic_rpc_handlers((generic_handler,))
|
|
82
|
+
|
|
83
|
+
|
|
84
|
+
# This class is part of an EXPERIMENTAL API.
|
|
85
|
+
class StateService(object):
|
|
86
|
+
"""provides an interface for managing the state of actions.
|
|
87
|
+
"""
|
|
88
|
+
|
|
89
|
+
@staticmethod
|
|
90
|
+
def Store(request,
|
|
91
|
+
target,
|
|
92
|
+
options=(),
|
|
93
|
+
channel_credentials=None,
|
|
94
|
+
call_credentials=None,
|
|
95
|
+
insecure=False,
|
|
96
|
+
compression=None,
|
|
97
|
+
wait_for_ready=None,
|
|
98
|
+
timeout=None,
|
|
99
|
+
metadata=None):
|
|
100
|
+
return grpc.experimental.unary_unary(request, target, '/cloudidl.workflow.StateService/Store',
|
|
101
|
+
workflow_dot_state__service__pb2.StoreRequest.SerializeToString,
|
|
102
|
+
workflow_dot_state__service__pb2.StoreResponse.FromString,
|
|
103
|
+
options, channel_credentials,
|
|
104
|
+
insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
|
|
105
|
+
|
|
106
|
+
@staticmethod
|
|
107
|
+
def Load(request,
|
|
108
|
+
target,
|
|
109
|
+
options=(),
|
|
110
|
+
channel_credentials=None,
|
|
111
|
+
call_credentials=None,
|
|
112
|
+
insecure=False,
|
|
113
|
+
compression=None,
|
|
114
|
+
wait_for_ready=None,
|
|
115
|
+
timeout=None,
|
|
116
|
+
metadata=None):
|
|
117
|
+
return grpc.experimental.unary_unary(request, target, '/cloudidl.workflow.StateService/Load',
|
|
118
|
+
workflow_dot_state__service__pb2.LoadRequest.SerializeToString,
|
|
119
|
+
workflow_dot_state__service__pb2.LoadResponse.FromString,
|
|
120
|
+
options, channel_credentials,
|
|
121
|
+
insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
|
|
122
|
+
|
|
123
|
+
@staticmethod
|
|
124
|
+
def Watch(request,
|
|
125
|
+
target,
|
|
126
|
+
options=(),
|
|
127
|
+
channel_credentials=None,
|
|
128
|
+
call_credentials=None,
|
|
129
|
+
insecure=False,
|
|
130
|
+
compression=None,
|
|
131
|
+
wait_for_ready=None,
|
|
132
|
+
timeout=None,
|
|
133
|
+
metadata=None):
|
|
134
|
+
return grpc.experimental.unary_stream(request, target, '/cloudidl.workflow.StateService/Watch',
|
|
135
|
+
workflow_dot_state__service__pb2.WatchRequest.SerializeToString,
|
|
136
|
+
workflow_dot_state__service__pb2.WatchResponse.FromString,
|
|
137
|
+
options, channel_credentials,
|
|
138
|
+
insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
|
|
@@ -0,0 +1,72 @@
|
|
|
1
|
+
# -*- coding: utf-8 -*-
|
|
2
|
+
# Generated by the protocol buffer compiler. DO NOT EDIT!
|
|
3
|
+
# source: workflow/task_definition.proto
|
|
4
|
+
"""Generated protocol buffer code."""
|
|
5
|
+
from google.protobuf import descriptor as _descriptor
|
|
6
|
+
from google.protobuf import descriptor_pool as _descriptor_pool
|
|
7
|
+
from google.protobuf import symbol_database as _symbol_database
|
|
8
|
+
from google.protobuf.internal import builder as _builder
|
|
9
|
+
# @@protoc_insertion_point(imports)
|
|
10
|
+
|
|
11
|
+
_sym_db = _symbol_database.Default()
|
|
12
|
+
|
|
13
|
+
|
|
14
|
+
from union._protos.common import identifier_pb2 as common_dot_identifier__pb2
|
|
15
|
+
from union._protos.common import identity_pb2 as common_dot_identity__pb2
|
|
16
|
+
from flyteidl.core import tasks_pb2 as flyteidl_dot_core_dot_tasks__pb2
|
|
17
|
+
from union._protos.validate.validate import validate_pb2 as validate_dot_validate__pb2
|
|
18
|
+
|
|
19
|
+
|
|
20
|
+
DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x1eworkflow/task_definition.proto\x12\x11\x63loudidl.workflow\x1a\x17\x63ommon/identifier.proto\x1a\x15\x63ommon/identity.proto\x1a\x19\x66lyteidl/core/tasks.proto\x1a\x17validate/validate.proto\"\x88\x01\n\x08TaskName\x12\x19\n\x03org\x18\x01 \x01(\tB\x07\xfa\x42\x04r\x02\x10\x01R\x03org\x12!\n\x07project\x18\x02 \x01(\tB\x07\xfa\x42\x04r\x02\x10\x01R\x07project\x12\x1f\n\x06\x64omain\x18\x03 \x01(\tB\x07\xfa\x42\x04r\x02\x10\x01R\x06\x64omain\x12\x1d\n\x04name\x18\x04 \x01(\tB\t\xfa\x42\x06r\x04\x10\x01\x18\x1eR\x04name\"\xb1\x01\n\x0eTaskIdentifier\x12\x19\n\x03org\x18\x01 \x01(\tB\x07\xfa\x42\x04r\x02\x10\x01R\x03org\x12!\n\x07project\x18\x02 \x01(\tB\x07\xfa\x42\x04r\x02\x10\x01R\x07project\x12\x1f\n\x06\x64omain\x18\x03 \x01(\tB\x07\xfa\x42\x04r\x02\x10\x01R\x06\x64omain\x12\x1d\n\x04name\x18\x04 \x01(\tB\t\xfa\x42\x06r\x04\x10\x01\x18\x1eR\x04name\x12!\n\x07version\x18\x05 \x01(\tB\x07\xfa\x42\x04r\x02\x10\x01R\x07version\"\\\n\x0cTaskMetadata\x12L\n\x0b\x64\x65ployed_by\x18\x01 \x01(\x0b\x32!.cloudidl.common.EnrichedIdentityB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01R\ndeployedBy\"\x93\x01\n\x04Task\x12\x44\n\x07task_id\x18\x01 \x01(\x0b\x32!.cloudidl.workflow.TaskIdentifierB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01R\x06taskId\x12\x45\n\x08metadata\x18\x02 \x01(\x0b\x32\x1f.cloudidl.workflow.TaskMetadataB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01R\x08metadata\"V\n\x08TaskSpec\x12J\n\rtask_template\x18\x01 \x01(\x0b\x32\x1b.flyteidl.core.TaskTemplateB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01R\x0ctaskTemplate\"\xd5\x01\n\x0bTaskDetails\x12\x44\n\x07task_id\x18\x01 \x01(\x0b\x32!.cloudidl.workflow.TaskIdentifierB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01R\x06taskId\x12\x45\n\x08metadata\x18\x02 \x01(\x0b\x32\x1f.cloudidl.workflow.TaskMetadataB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01R\x08metadata\x12\x39\n\x04spec\x18\x03 \x01(\x0b\x32\x1b.cloudidl.workflow.TaskSpecB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01R\x04specB\xc0\x01\n\x15\x63om.cloudidl.workflowB\x13TaskDefinitionProtoH\x02P\x01Z+github.com/unionai/cloud/gen/pb-go/workflow\xa2\x02\x03\x43WX\xaa\x02\x11\x43loudidl.Workflow\xca\x02\x11\x43loudidl\\Workflow\xe2\x02\x1d\x43loudidl\\Workflow\\GPBMetadata\xea\x02\x12\x43loudidl::Workflowb\x06proto3')
|
|
21
|
+
|
|
22
|
+
_globals = globals()
|
|
23
|
+
_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
|
|
24
|
+
_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'workflow.task_definition_pb2', _globals)
|
|
25
|
+
if _descriptor._USE_C_DESCRIPTORS == False:
|
|
26
|
+
DESCRIPTOR._options = None
|
|
27
|
+
DESCRIPTOR._serialized_options = b'\n\025com.cloudidl.workflowB\023TaskDefinitionProtoH\002P\001Z+github.com/unionai/cloud/gen/pb-go/workflow\242\002\003CWX\252\002\021Cloudidl.Workflow\312\002\021Cloudidl\\Workflow\342\002\035Cloudidl\\Workflow\\GPBMetadata\352\002\022Cloudidl::Workflow'
|
|
28
|
+
_TASKNAME.fields_by_name['org']._options = None
|
|
29
|
+
_TASKNAME.fields_by_name['org']._serialized_options = b'\372B\004r\002\020\001'
|
|
30
|
+
_TASKNAME.fields_by_name['project']._options = None
|
|
31
|
+
_TASKNAME.fields_by_name['project']._serialized_options = b'\372B\004r\002\020\001'
|
|
32
|
+
_TASKNAME.fields_by_name['domain']._options = None
|
|
33
|
+
_TASKNAME.fields_by_name['domain']._serialized_options = b'\372B\004r\002\020\001'
|
|
34
|
+
_TASKNAME.fields_by_name['name']._options = None
|
|
35
|
+
_TASKNAME.fields_by_name['name']._serialized_options = b'\372B\006r\004\020\001\030\036'
|
|
36
|
+
_TASKIDENTIFIER.fields_by_name['org']._options = None
|
|
37
|
+
_TASKIDENTIFIER.fields_by_name['org']._serialized_options = b'\372B\004r\002\020\001'
|
|
38
|
+
_TASKIDENTIFIER.fields_by_name['project']._options = None
|
|
39
|
+
_TASKIDENTIFIER.fields_by_name['project']._serialized_options = b'\372B\004r\002\020\001'
|
|
40
|
+
_TASKIDENTIFIER.fields_by_name['domain']._options = None
|
|
41
|
+
_TASKIDENTIFIER.fields_by_name['domain']._serialized_options = b'\372B\004r\002\020\001'
|
|
42
|
+
_TASKIDENTIFIER.fields_by_name['name']._options = None
|
|
43
|
+
_TASKIDENTIFIER.fields_by_name['name']._serialized_options = b'\372B\006r\004\020\001\030\036'
|
|
44
|
+
_TASKIDENTIFIER.fields_by_name['version']._options = None
|
|
45
|
+
_TASKIDENTIFIER.fields_by_name['version']._serialized_options = b'\372B\004r\002\020\001'
|
|
46
|
+
_TASKMETADATA.fields_by_name['deployed_by']._options = None
|
|
47
|
+
_TASKMETADATA.fields_by_name['deployed_by']._serialized_options = b'\372B\005\212\001\002\020\001'
|
|
48
|
+
_TASK.fields_by_name['task_id']._options = None
|
|
49
|
+
_TASK.fields_by_name['task_id']._serialized_options = b'\372B\005\212\001\002\020\001'
|
|
50
|
+
_TASK.fields_by_name['metadata']._options = None
|
|
51
|
+
_TASK.fields_by_name['metadata']._serialized_options = b'\372B\005\212\001\002\020\001'
|
|
52
|
+
_TASKSPEC.fields_by_name['task_template']._options = None
|
|
53
|
+
_TASKSPEC.fields_by_name['task_template']._serialized_options = b'\372B\005\212\001\002\020\001'
|
|
54
|
+
_TASKDETAILS.fields_by_name['task_id']._options = None
|
|
55
|
+
_TASKDETAILS.fields_by_name['task_id']._serialized_options = b'\372B\005\212\001\002\020\001'
|
|
56
|
+
_TASKDETAILS.fields_by_name['metadata']._options = None
|
|
57
|
+
_TASKDETAILS.fields_by_name['metadata']._serialized_options = b'\372B\005\212\001\002\020\001'
|
|
58
|
+
_TASKDETAILS.fields_by_name['spec']._options = None
|
|
59
|
+
_TASKDETAILS.fields_by_name['spec']._serialized_options = b'\372B\005\212\001\002\020\001'
|
|
60
|
+
_globals['_TASKNAME']._serialized_start=154
|
|
61
|
+
_globals['_TASKNAME']._serialized_end=290
|
|
62
|
+
_globals['_TASKIDENTIFIER']._serialized_start=293
|
|
63
|
+
_globals['_TASKIDENTIFIER']._serialized_end=470
|
|
64
|
+
_globals['_TASKMETADATA']._serialized_start=472
|
|
65
|
+
_globals['_TASKMETADATA']._serialized_end=564
|
|
66
|
+
_globals['_TASK']._serialized_start=567
|
|
67
|
+
_globals['_TASK']._serialized_end=714
|
|
68
|
+
_globals['_TASKSPEC']._serialized_start=716
|
|
69
|
+
_globals['_TASKSPEC']._serialized_end=802
|
|
70
|
+
_globals['_TASKDETAILS']._serialized_start=805
|
|
71
|
+
_globals['_TASKDETAILS']._serialized_end=1018
|
|
72
|
+
# @@protoc_insertion_point(module_scope)
|
|
@@ -0,0 +1,65 @@
|
|
|
1
|
+
from union._protos.common import identifier_pb2 as _identifier_pb2
|
|
2
|
+
from union._protos.common import identity_pb2 as _identity_pb2
|
|
3
|
+
from flyteidl.core import tasks_pb2 as _tasks_pb2
|
|
4
|
+
from union._protos.validate.validate import validate_pb2 as _validate_pb2
|
|
5
|
+
from google.protobuf import descriptor as _descriptor
|
|
6
|
+
from google.protobuf import message as _message
|
|
7
|
+
from typing import ClassVar as _ClassVar, Mapping as _Mapping, Optional as _Optional, Union as _Union
|
|
8
|
+
|
|
9
|
+
DESCRIPTOR: _descriptor.FileDescriptor
|
|
10
|
+
|
|
11
|
+
class TaskName(_message.Message):
|
|
12
|
+
__slots__ = ["org", "project", "domain", "name"]
|
|
13
|
+
ORG_FIELD_NUMBER: _ClassVar[int]
|
|
14
|
+
PROJECT_FIELD_NUMBER: _ClassVar[int]
|
|
15
|
+
DOMAIN_FIELD_NUMBER: _ClassVar[int]
|
|
16
|
+
NAME_FIELD_NUMBER: _ClassVar[int]
|
|
17
|
+
org: str
|
|
18
|
+
project: str
|
|
19
|
+
domain: str
|
|
20
|
+
name: str
|
|
21
|
+
def __init__(self, org: _Optional[str] = ..., project: _Optional[str] = ..., domain: _Optional[str] = ..., name: _Optional[str] = ...) -> None: ...
|
|
22
|
+
|
|
23
|
+
class TaskIdentifier(_message.Message):
|
|
24
|
+
__slots__ = ["org", "project", "domain", "name", "version"]
|
|
25
|
+
ORG_FIELD_NUMBER: _ClassVar[int]
|
|
26
|
+
PROJECT_FIELD_NUMBER: _ClassVar[int]
|
|
27
|
+
DOMAIN_FIELD_NUMBER: _ClassVar[int]
|
|
28
|
+
NAME_FIELD_NUMBER: _ClassVar[int]
|
|
29
|
+
VERSION_FIELD_NUMBER: _ClassVar[int]
|
|
30
|
+
org: str
|
|
31
|
+
project: str
|
|
32
|
+
domain: str
|
|
33
|
+
name: str
|
|
34
|
+
version: str
|
|
35
|
+
def __init__(self, org: _Optional[str] = ..., project: _Optional[str] = ..., domain: _Optional[str] = ..., name: _Optional[str] = ..., version: _Optional[str] = ...) -> None: ...
|
|
36
|
+
|
|
37
|
+
class TaskMetadata(_message.Message):
|
|
38
|
+
__slots__ = ["deployed_by"]
|
|
39
|
+
DEPLOYED_BY_FIELD_NUMBER: _ClassVar[int]
|
|
40
|
+
deployed_by: _identity_pb2.EnrichedIdentity
|
|
41
|
+
def __init__(self, deployed_by: _Optional[_Union[_identity_pb2.EnrichedIdentity, _Mapping]] = ...) -> None: ...
|
|
42
|
+
|
|
43
|
+
class Task(_message.Message):
|
|
44
|
+
__slots__ = ["task_id", "metadata"]
|
|
45
|
+
TASK_ID_FIELD_NUMBER: _ClassVar[int]
|
|
46
|
+
METADATA_FIELD_NUMBER: _ClassVar[int]
|
|
47
|
+
task_id: TaskIdentifier
|
|
48
|
+
metadata: TaskMetadata
|
|
49
|
+
def __init__(self, task_id: _Optional[_Union[TaskIdentifier, _Mapping]] = ..., metadata: _Optional[_Union[TaskMetadata, _Mapping]] = ...) -> None: ...
|
|
50
|
+
|
|
51
|
+
class TaskSpec(_message.Message):
|
|
52
|
+
__slots__ = ["task_template"]
|
|
53
|
+
TASK_TEMPLATE_FIELD_NUMBER: _ClassVar[int]
|
|
54
|
+
task_template: _tasks_pb2.TaskTemplate
|
|
55
|
+
def __init__(self, task_template: _Optional[_Union[_tasks_pb2.TaskTemplate, _Mapping]] = ...) -> None: ...
|
|
56
|
+
|
|
57
|
+
class TaskDetails(_message.Message):
|
|
58
|
+
__slots__ = ["task_id", "metadata", "spec"]
|
|
59
|
+
TASK_ID_FIELD_NUMBER: _ClassVar[int]
|
|
60
|
+
METADATA_FIELD_NUMBER: _ClassVar[int]
|
|
61
|
+
SPEC_FIELD_NUMBER: _ClassVar[int]
|
|
62
|
+
task_id: TaskIdentifier
|
|
63
|
+
metadata: TaskMetadata
|
|
64
|
+
spec: TaskSpec
|
|
65
|
+
def __init__(self, task_id: _Optional[_Union[TaskIdentifier, _Mapping]] = ..., metadata: _Optional[_Union[TaskMetadata, _Mapping]] = ..., spec: _Optional[_Union[TaskSpec, _Mapping]] = ...) -> None: ...
|
|
@@ -0,0 +1,44 @@
|
|
|
1
|
+
# -*- coding: utf-8 -*-
|
|
2
|
+
# Generated by the protocol buffer compiler. DO NOT EDIT!
|
|
3
|
+
# source: workflow/task_service.proto
|
|
4
|
+
"""Generated protocol buffer code."""
|
|
5
|
+
from google.protobuf import descriptor as _descriptor
|
|
6
|
+
from google.protobuf import descriptor_pool as _descriptor_pool
|
|
7
|
+
from google.protobuf import symbol_database as _symbol_database
|
|
8
|
+
from google.protobuf.internal import builder as _builder
|
|
9
|
+
# @@protoc_insertion_point(imports)
|
|
10
|
+
|
|
11
|
+
_sym_db = _symbol_database.Default()
|
|
12
|
+
|
|
13
|
+
|
|
14
|
+
from union._protos.validate.validate import validate_pb2 as validate_dot_validate__pb2
|
|
15
|
+
from union._protos.workflow import task_definition_pb2 as workflow_dot_task__definition__pb2
|
|
16
|
+
|
|
17
|
+
|
|
18
|
+
DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x1bworkflow/task_service.proto\x12\x11\x63loudidl.workflow\x1a\x17validate/validate.proto\x1a\x1eworkflow/task_definition.proto\"\x94\x01\n\x11\x44\x65ployTaskRequest\x12\x44\n\x07task_id\x18\x01 \x01(\x0b\x32!.cloudidl.workflow.TaskIdentifierB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01R\x06taskId\x12\x39\n\x04spec\x18\x02 \x01(\x0b\x32\x1b.cloudidl.workflow.TaskSpecB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01R\x04spec\"\x14\n\x12\x44\x65ployTaskResponse\"]\n\x15GetTaskDetailsRequest\x12\x44\n\x07task_id\x18\x01 \x01(\x0b\x32!.cloudidl.workflow.TaskIdentifierB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01R\x06taskId\"R\n\x16GetTaskDetailsResponse\x12\x38\n\x07\x64\x65tails\x18\x01 \x01(\x0b\x32\x1e.cloudidl.workflow.TaskDetailsR\x07\x64\x65tails2\xd6\x01\n\x0bTaskService\x12[\n\nDeployTask\x12$.cloudidl.workflow.DeployTaskRequest\x1a%.cloudidl.workflow.DeployTaskResponse\"\x00\x12j\n\x0eGetTaskDetails\x12(.cloudidl.workflow.GetTaskDetailsRequest\x1a).cloudidl.workflow.GetTaskDetailsResponse\"\x03\x90\x02\x01\x42\xbd\x01\n\x15\x63om.cloudidl.workflowB\x10TaskServiceProtoH\x02P\x01Z+github.com/unionai/cloud/gen/pb-go/workflow\xa2\x02\x03\x43WX\xaa\x02\x11\x43loudidl.Workflow\xca\x02\x11\x43loudidl\\Workflow\xe2\x02\x1d\x43loudidl\\Workflow\\GPBMetadata\xea\x02\x12\x43loudidl::Workflowb\x06proto3')
|
|
19
|
+
|
|
20
|
+
_globals = globals()
|
|
21
|
+
_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
|
|
22
|
+
_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'workflow.task_service_pb2', _globals)
|
|
23
|
+
if _descriptor._USE_C_DESCRIPTORS == False:
|
|
24
|
+
DESCRIPTOR._options = None
|
|
25
|
+
DESCRIPTOR._serialized_options = b'\n\025com.cloudidl.workflowB\020TaskServiceProtoH\002P\001Z+github.com/unionai/cloud/gen/pb-go/workflow\242\002\003CWX\252\002\021Cloudidl.Workflow\312\002\021Cloudidl\\Workflow\342\002\035Cloudidl\\Workflow\\GPBMetadata\352\002\022Cloudidl::Workflow'
|
|
26
|
+
_DEPLOYTASKREQUEST.fields_by_name['task_id']._options = None
|
|
27
|
+
_DEPLOYTASKREQUEST.fields_by_name['task_id']._serialized_options = b'\372B\005\212\001\002\020\001'
|
|
28
|
+
_DEPLOYTASKREQUEST.fields_by_name['spec']._options = None
|
|
29
|
+
_DEPLOYTASKREQUEST.fields_by_name['spec']._serialized_options = b'\372B\005\212\001\002\020\001'
|
|
30
|
+
_GETTASKDETAILSREQUEST.fields_by_name['task_id']._options = None
|
|
31
|
+
_GETTASKDETAILSREQUEST.fields_by_name['task_id']._serialized_options = b'\372B\005\212\001\002\020\001'
|
|
32
|
+
_TASKSERVICE.methods_by_name['GetTaskDetails']._options = None
|
|
33
|
+
_TASKSERVICE.methods_by_name['GetTaskDetails']._serialized_options = b'\220\002\001'
|
|
34
|
+
_globals['_DEPLOYTASKREQUEST']._serialized_start=108
|
|
35
|
+
_globals['_DEPLOYTASKREQUEST']._serialized_end=256
|
|
36
|
+
_globals['_DEPLOYTASKRESPONSE']._serialized_start=258
|
|
37
|
+
_globals['_DEPLOYTASKRESPONSE']._serialized_end=278
|
|
38
|
+
_globals['_GETTASKDETAILSREQUEST']._serialized_start=280
|
|
39
|
+
_globals['_GETTASKDETAILSREQUEST']._serialized_end=373
|
|
40
|
+
_globals['_GETTASKDETAILSRESPONSE']._serialized_start=375
|
|
41
|
+
_globals['_GETTASKDETAILSRESPONSE']._serialized_end=457
|
|
42
|
+
_globals['_TASKSERVICE']._serialized_start=460
|
|
43
|
+
_globals['_TASKSERVICE']._serialized_end=674
|
|
44
|
+
# @@protoc_insertion_point(module_scope)
|
|
@@ -0,0 +1,31 @@
|
|
|
1
|
+
from union._protos.validate.validate import validate_pb2 as _validate_pb2
|
|
2
|
+
from union._protos.workflow import task_definition_pb2 as _task_definition_pb2
|
|
3
|
+
from google.protobuf import descriptor as _descriptor
|
|
4
|
+
from google.protobuf import message as _message
|
|
5
|
+
from typing import ClassVar as _ClassVar, Mapping as _Mapping, Optional as _Optional, Union as _Union
|
|
6
|
+
|
|
7
|
+
DESCRIPTOR: _descriptor.FileDescriptor
|
|
8
|
+
|
|
9
|
+
class DeployTaskRequest(_message.Message):
|
|
10
|
+
__slots__ = ["task_id", "spec"]
|
|
11
|
+
TASK_ID_FIELD_NUMBER: _ClassVar[int]
|
|
12
|
+
SPEC_FIELD_NUMBER: _ClassVar[int]
|
|
13
|
+
task_id: _task_definition_pb2.TaskIdentifier
|
|
14
|
+
spec: _task_definition_pb2.TaskSpec
|
|
15
|
+
def __init__(self, task_id: _Optional[_Union[_task_definition_pb2.TaskIdentifier, _Mapping]] = ..., spec: _Optional[_Union[_task_definition_pb2.TaskSpec, _Mapping]] = ...) -> None: ...
|
|
16
|
+
|
|
17
|
+
class DeployTaskResponse(_message.Message):
|
|
18
|
+
__slots__ = []
|
|
19
|
+
def __init__(self) -> None: ...
|
|
20
|
+
|
|
21
|
+
class GetTaskDetailsRequest(_message.Message):
|
|
22
|
+
__slots__ = ["task_id"]
|
|
23
|
+
TASK_ID_FIELD_NUMBER: _ClassVar[int]
|
|
24
|
+
task_id: _task_definition_pb2.TaskIdentifier
|
|
25
|
+
def __init__(self, task_id: _Optional[_Union[_task_definition_pb2.TaskIdentifier, _Mapping]] = ...) -> None: ...
|
|
26
|
+
|
|
27
|
+
class GetTaskDetailsResponse(_message.Message):
|
|
28
|
+
__slots__ = ["details"]
|
|
29
|
+
DETAILS_FIELD_NUMBER: _ClassVar[int]
|
|
30
|
+
details: _task_definition_pb2.TaskDetails
|
|
31
|
+
def __init__(self, details: _Optional[_Union[_task_definition_pb2.TaskDetails, _Mapping]] = ...) -> None: ...
|
|
@@ -0,0 +1,104 @@
|
|
|
1
|
+
# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
|
|
2
|
+
"""Client and server classes corresponding to protobuf-defined services."""
|
|
3
|
+
import grpc
|
|
4
|
+
|
|
5
|
+
from union._protos.workflow import task_service_pb2 as workflow_dot_task__service__pb2
|
|
6
|
+
|
|
7
|
+
|
|
8
|
+
class TaskServiceStub(object):
|
|
9
|
+
"""TaskService provides an interface for managing tasks.
|
|
10
|
+
"""
|
|
11
|
+
|
|
12
|
+
def __init__(self, channel):
|
|
13
|
+
"""Constructor.
|
|
14
|
+
|
|
15
|
+
Args:
|
|
16
|
+
channel: A grpc.Channel.
|
|
17
|
+
"""
|
|
18
|
+
self.DeployTask = channel.unary_unary(
|
|
19
|
+
'/cloudidl.workflow.TaskService/DeployTask',
|
|
20
|
+
request_serializer=workflow_dot_task__service__pb2.DeployTaskRequest.SerializeToString,
|
|
21
|
+
response_deserializer=workflow_dot_task__service__pb2.DeployTaskResponse.FromString,
|
|
22
|
+
)
|
|
23
|
+
self.GetTaskDetails = channel.unary_unary(
|
|
24
|
+
'/cloudidl.workflow.TaskService/GetTaskDetails',
|
|
25
|
+
request_serializer=workflow_dot_task__service__pb2.GetTaskDetailsRequest.SerializeToString,
|
|
26
|
+
response_deserializer=workflow_dot_task__service__pb2.GetTaskDetailsResponse.FromString,
|
|
27
|
+
)
|
|
28
|
+
|
|
29
|
+
|
|
30
|
+
class TaskServiceServicer(object):
|
|
31
|
+
"""TaskService provides an interface for managing tasks.
|
|
32
|
+
"""
|
|
33
|
+
|
|
34
|
+
def DeployTask(self, request, context):
|
|
35
|
+
"""Deploy a task.
|
|
36
|
+
"""
|
|
37
|
+
context.set_code(grpc.StatusCode.UNIMPLEMENTED)
|
|
38
|
+
context.set_details('Method not implemented!')
|
|
39
|
+
raise NotImplementedError('Method not implemented!')
|
|
40
|
+
|
|
41
|
+
def GetTaskDetails(self, request, context):
|
|
42
|
+
"""Get detailed information about a task.
|
|
43
|
+
"""
|
|
44
|
+
context.set_code(grpc.StatusCode.UNIMPLEMENTED)
|
|
45
|
+
context.set_details('Method not implemented!')
|
|
46
|
+
raise NotImplementedError('Method not implemented!')
|
|
47
|
+
|
|
48
|
+
|
|
49
|
+
def add_TaskServiceServicer_to_server(servicer, server):
|
|
50
|
+
rpc_method_handlers = {
|
|
51
|
+
'DeployTask': grpc.unary_unary_rpc_method_handler(
|
|
52
|
+
servicer.DeployTask,
|
|
53
|
+
request_deserializer=workflow_dot_task__service__pb2.DeployTaskRequest.FromString,
|
|
54
|
+
response_serializer=workflow_dot_task__service__pb2.DeployTaskResponse.SerializeToString,
|
|
55
|
+
),
|
|
56
|
+
'GetTaskDetails': grpc.unary_unary_rpc_method_handler(
|
|
57
|
+
servicer.GetTaskDetails,
|
|
58
|
+
request_deserializer=workflow_dot_task__service__pb2.GetTaskDetailsRequest.FromString,
|
|
59
|
+
response_serializer=workflow_dot_task__service__pb2.GetTaskDetailsResponse.SerializeToString,
|
|
60
|
+
),
|
|
61
|
+
}
|
|
62
|
+
generic_handler = grpc.method_handlers_generic_handler(
|
|
63
|
+
'cloudidl.workflow.TaskService', rpc_method_handlers)
|
|
64
|
+
server.add_generic_rpc_handlers((generic_handler,))
|
|
65
|
+
|
|
66
|
+
|
|
67
|
+
# This class is part of an EXPERIMENTAL API.
|
|
68
|
+
class TaskService(object):
|
|
69
|
+
"""TaskService provides an interface for managing tasks.
|
|
70
|
+
"""
|
|
71
|
+
|
|
72
|
+
@staticmethod
|
|
73
|
+
def DeployTask(request,
|
|
74
|
+
target,
|
|
75
|
+
options=(),
|
|
76
|
+
channel_credentials=None,
|
|
77
|
+
call_credentials=None,
|
|
78
|
+
insecure=False,
|
|
79
|
+
compression=None,
|
|
80
|
+
wait_for_ready=None,
|
|
81
|
+
timeout=None,
|
|
82
|
+
metadata=None):
|
|
83
|
+
return grpc.experimental.unary_unary(request, target, '/cloudidl.workflow.TaskService/DeployTask',
|
|
84
|
+
workflow_dot_task__service__pb2.DeployTaskRequest.SerializeToString,
|
|
85
|
+
workflow_dot_task__service__pb2.DeployTaskResponse.FromString,
|
|
86
|
+
options, channel_credentials,
|
|
87
|
+
insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
|
|
88
|
+
|
|
89
|
+
@staticmethod
|
|
90
|
+
def GetTaskDetails(request,
|
|
91
|
+
target,
|
|
92
|
+
options=(),
|
|
93
|
+
channel_credentials=None,
|
|
94
|
+
call_credentials=None,
|
|
95
|
+
insecure=False,
|
|
96
|
+
compression=None,
|
|
97
|
+
wait_for_ready=None,
|
|
98
|
+
timeout=None,
|
|
99
|
+
metadata=None):
|
|
100
|
+
return grpc.experimental.unary_unary(request, target, '/cloudidl.workflow.TaskService/GetTaskDetails',
|
|
101
|
+
workflow_dot_task__service__pb2.GetTaskDetailsRequest.SerializeToString,
|
|
102
|
+
workflow_dot_task__service__pb2.GetTaskDetailsResponse.FromString,
|
|
103
|
+
options, channel_credentials,
|
|
104
|
+
insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
|