raystack-proton 2025.11.5.35259__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.
Files changed (181) hide show
  1. google/__init__.py +0 -0
  2. google/api/__init__.py +0 -0
  3. google/api/annotations_pb2.py +37 -0
  4. google/api/annotations_pb2.pyi +8 -0
  5. google/api/field_behavior_pb2.py +38 -0
  6. google/api/field_behavior_pb2.pyi +27 -0
  7. google/api/http_pb2.py +41 -0
  8. google/api/http_pb2.pyi +47 -0
  9. google/api/httpbody_pb2.py +38 -0
  10. google/api/httpbody_pb2.pyi +18 -0
  11. protoc_gen_openapiv2/__init__.py +0 -0
  12. protoc_gen_openapiv2/options/__init__.py +0 -0
  13. protoc_gen_openapiv2/options/annotations_pb2.py +37 -0
  14. protoc_gen_openapiv2/options/annotations_pb2.pyi +16 -0
  15. protoc_gen_openapiv2/options/openapiv2_pb2.py +142 -0
  16. protoc_gen_openapiv2/options/openapiv2_pb2.pyi +463 -0
  17. raystack/__init__.py +0 -0
  18. raystack/assets/__init__.py +0 -0
  19. raystack/assets/v1beta1/__init__.py +0 -0
  20. raystack/assets/v1beta1/bucket_pb2.py +45 -0
  21. raystack/assets/v1beta1/bucket_pb2.pyi +59 -0
  22. raystack/assets/v1beta1/dashboard_pb2.py +45 -0
  23. raystack/assets/v1beta1/dashboard_pb2.pyi +65 -0
  24. raystack/assets/v1beta1/event_pb2.py +38 -0
  25. raystack/assets/v1beta1/event_pb2.pyi +19 -0
  26. raystack/assets/v1beta1/group_pb2.py +43 -0
  27. raystack/assets/v1beta1/group_pb2.pyi +35 -0
  28. raystack/assets/v1beta1/job_pb2.py +43 -0
  29. raystack/assets/v1beta1/job_pb2.pyi +28 -0
  30. raystack/assets/v1beta1/lineage_pb2.py +38 -0
  31. raystack/assets/v1beta1/lineage_pb2.pyi +16 -0
  32. raystack/assets/v1beta1/ownership_pb2.py +39 -0
  33. raystack/assets/v1beta1/ownership_pb2.pyi +25 -0
  34. raystack/assets/v1beta1/preview_pb2.py +38 -0
  35. raystack/assets/v1beta1/preview_pb2.pyi +16 -0
  36. raystack/assets/v1beta1/properties_pb2.py +42 -0
  37. raystack/assets/v1beta1/properties_pb2.pyi +25 -0
  38. raystack/assets/v1beta1/resource_pb2.py +37 -0
  39. raystack/assets/v1beta1/resource_pb2.pyi +21 -0
  40. raystack/assets/v1beta1/schema_pb2.py +44 -0
  41. raystack/assets/v1beta1/schema_pb2.pyi +58 -0
  42. raystack/assets/v1beta1/table_pb2.py +49 -0
  43. raystack/assets/v1beta1/table_pb2.pyi +63 -0
  44. raystack/assets/v1beta1/timestamp_pb2.py +40 -0
  45. raystack/assets/v1beta1/timestamp_pb2.pyi +25 -0
  46. raystack/assets/v1beta1/topic_pb2.py +46 -0
  47. raystack/assets/v1beta1/topic_pb2.pyi +41 -0
  48. raystack/assets/v1beta1/user_pb2.py +45 -0
  49. raystack/assets/v1beta1/user_pb2.pyi +63 -0
  50. raystack/assets/v1beta2/__init__.py +0 -0
  51. raystack/assets/v1beta2/application_pb2.py +39 -0
  52. raystack/assets/v1beta2/application_pb2.pyi +24 -0
  53. raystack/assets/v1beta2/asset_pb2.py +44 -0
  54. raystack/assets/v1beta2/asset_pb2.pyi +49 -0
  55. raystack/assets/v1beta2/bucket_pb2.py +42 -0
  56. raystack/assets/v1beta2/bucket_pb2.pyi +54 -0
  57. raystack/assets/v1beta2/common_pb2.py +44 -0
  58. raystack/assets/v1beta2/common_pb2.pyi +52 -0
  59. raystack/assets/v1beta2/dashboard_pb2.py +42 -0
  60. raystack/assets/v1beta2/dashboard_pb2.pyi +60 -0
  61. raystack/assets/v1beta2/experiment_pb2.py +41 -0
  62. raystack/assets/v1beta2/experiment_pb2.pyi +40 -0
  63. raystack/assets/v1beta2/feature_table_pb2.py +47 -0
  64. raystack/assets/v1beta2/feature_table_pb2.pyi +59 -0
  65. raystack/assets/v1beta2/group_pb2.py +40 -0
  66. raystack/assets/v1beta2/group_pb2.pyi +26 -0
  67. raystack/assets/v1beta2/job_pb2.py +39 -0
  68. raystack/assets/v1beta2/job_pb2.pyi +20 -0
  69. raystack/assets/v1beta2/metric_pb2.py +39 -0
  70. raystack/assets/v1beta2/metric_pb2.pyi +28 -0
  71. raystack/assets/v1beta2/model_pb2.py +49 -0
  72. raystack/assets/v1beta2/model_pb2.pyi +73 -0
  73. raystack/assets/v1beta2/table_pb2.py +47 -0
  74. raystack/assets/v1beta2/table_pb2.pyi +93 -0
  75. raystack/assets/v1beta2/topic_pb2.py +43 -0
  76. raystack/assets/v1beta2/topic_pb2.pyi +40 -0
  77. raystack/assets/v1beta2/user_pb2.py +43 -0
  78. raystack/assets/v1beta2/user_pb2.pyi +61 -0
  79. raystack/common/__init__.py +0 -0
  80. raystack/common/v1/__init__.py +0 -0
  81. raystack/common/v1/service_connect.py +119 -0
  82. raystack/common/v1/service_pb2.py +48 -0
  83. raystack/common/v1/service_pb2.pyi +39 -0
  84. raystack/compass/__init__.py +0 -0
  85. raystack/compass/v1beta1/__init__.py +0 -0
  86. raystack/compass/v1beta1/service_connect.py +2784 -0
  87. raystack/compass/v1beta1/service_pb2.py +581 -0
  88. raystack/compass/v1beta1/service_pb2.pyi +1055 -0
  89. raystack/entropy/__init__.py +0 -0
  90. raystack/entropy/v1beta1/__init__.py +0 -0
  91. raystack/entropy/v1beta1/module_connect.py +379 -0
  92. raystack/entropy/v1beta1/module_pb2.py +72 -0
  93. raystack/entropy/v1beta1/module_pb2.pyi +88 -0
  94. raystack/entropy/v1beta1/resource_connect.py +574 -0
  95. raystack/entropy/v1beta1/resource_pb2.py +134 -0
  96. raystack/entropy/v1beta1/resource_pb2.pyi +268 -0
  97. raystack/frontier/__init__.py +0 -0
  98. raystack/frontier/v1beta1/__init__.py +0 -0
  99. raystack/frontier/v1beta1/admin_connect.py +4021 -0
  100. raystack/frontier/v1beta1/admin_pb2.py +617 -0
  101. raystack/frontier/v1beta1/admin_pb2.pyi +1151 -0
  102. raystack/frontier/v1beta1/frontier_connect.py +10974 -0
  103. raystack/frontier/v1beta1/frontier_pb2.py +1612 -0
  104. raystack/frontier/v1beta1/frontier_pb2.pyi +2666 -0
  105. raystack/frontier/v1beta1/models_pb2.py +400 -0
  106. raystack/frontier/v1beta1/models_pb2.pyi +1160 -0
  107. raystack/guardian/__init__.py +0 -0
  108. raystack/guardian/v1beta1/__init__.py +0 -0
  109. raystack/guardian/v1beta1/guardian_connect.py +2654 -0
  110. raystack/guardian/v1beta1/guardian_pb2.py +527 -0
  111. raystack/guardian/v1beta1/guardian_pb2.pyi +1257 -0
  112. raystack/optimus/__init__.py +0 -0
  113. raystack/optimus/core/__init__.py +0 -0
  114. raystack/optimus/core/v1beta1/__init__.py +0 -0
  115. raystack/optimus/core/v1beta1/backup_connect.py +249 -0
  116. raystack/optimus/core/v1beta1/backup_pb2.py +70 -0
  117. raystack/optimus/core/v1beta1/backup_pb2.pyi +108 -0
  118. raystack/optimus/core/v1beta1/job_run_connect.py +314 -0
  119. raystack/optimus/core/v1beta1/job_run_pb2.py +88 -0
  120. raystack/optimus/core/v1beta1/job_run_pb2.pyi +164 -0
  121. raystack/optimus/core/v1beta1/job_spec_connect.py +1289 -0
  122. raystack/optimus/core/v1beta1/job_spec_pb2.py +229 -0
  123. raystack/optimus/core/v1beta1/job_spec_pb2.pyi +700 -0
  124. raystack/optimus/core/v1beta1/namespace_connect.py +249 -0
  125. raystack/optimus/core/v1beta1/namespace_pb2.py +63 -0
  126. raystack/optimus/core/v1beta1/namespace_pb2.pyi +66 -0
  127. raystack/optimus/core/v1beta1/project_connect.py +249 -0
  128. raystack/optimus/core/v1beta1/project_pb2.py +63 -0
  129. raystack/optimus/core/v1beta1/project_pb2.pyi +60 -0
  130. raystack/optimus/core/v1beta1/replay_connect.py +314 -0
  131. raystack/optimus/core/v1beta1/replay_pb2.py +72 -0
  132. raystack/optimus/core/v1beta1/replay_pb2.pyi +127 -0
  133. raystack/optimus/core/v1beta1/resource_connect.py +509 -0
  134. raystack/optimus/core/v1beta1/resource_pb2.py +93 -0
  135. raystack/optimus/core/v1beta1/resource_pb2.pyi +182 -0
  136. raystack/optimus/core/v1beta1/runtime_connect.py +119 -0
  137. raystack/optimus/core/v1beta1/runtime_pb2.py +45 -0
  138. raystack/optimus/core/v1beta1/runtime_pb2.pyi +19 -0
  139. raystack/optimus/core/v1beta1/secret_connect.py +314 -0
  140. raystack/optimus/core/v1beta1/secret_pb2.py +66 -0
  141. raystack/optimus/core/v1beta1/secret_pb2.pyi +81 -0
  142. raystack/optimus/core/v1beta1/status_pb2.py +39 -0
  143. raystack/optimus/core/v1beta1/status_pb2.pyi +31 -0
  144. raystack/optimus/integration/__init__.py +0 -0
  145. raystack/optimus/integration/v1beta1/__init__.py +0 -0
  146. raystack/optimus/integration/v1beta1/event_pb2.py +50 -0
  147. raystack/optimus/integration/v1beta1/event_pb2.pyi +94 -0
  148. raystack/optimus/plugins/__init__.py +0 -0
  149. raystack/optimus/plugins/v1beta1/__init__.py +0 -0
  150. raystack/optimus/plugins/v1beta1/dependency_resolver_connect.py +314 -0
  151. raystack/optimus/plugins/v1beta1/dependency_resolver_pb2.py +66 -0
  152. raystack/optimus/plugins/v1beta1/dependency_resolver_pb2.pyi +118 -0
  153. raystack/predator/__init__.py +0 -0
  154. raystack/predator/v1beta1/__init__.py +0 -0
  155. raystack/predator/v1beta1/metrics_log_pb2.py +46 -0
  156. raystack/predator/v1beta1/metrics_log_pb2.pyi +68 -0
  157. raystack/predator/v1beta1/result_log_pb2.py +45 -0
  158. raystack/predator/v1beta1/result_log_pb2.pyi +61 -0
  159. raystack/raccoon/__init__.py +0 -0
  160. raystack/raccoon/v1beta1/__init__.py +0 -0
  161. raystack/raccoon/v1beta1/raccoon_connect.py +119 -0
  162. raystack/raccoon/v1beta1/raccoon_pb2.py +52 -0
  163. raystack/raccoon/v1beta1/raccoon_pb2.pyi +74 -0
  164. raystack/siren/__init__.py +0 -0
  165. raystack/siren/v1beta1/__init__.py +0 -0
  166. raystack/siren/v1beta1/siren_connect.py +2329 -0
  167. raystack/siren/v1beta1/siren_pb2.py +415 -0
  168. raystack/siren/v1beta1/siren_pb2.pyi +890 -0
  169. raystack/stencil/__init__.py +0 -0
  170. raystack/stencil/v1beta1/__init__.py +0 -0
  171. raystack/stencil/v1beta1/stencil_connect.py +1094 -0
  172. raystack/stencil/v1beta1/stencil_pb2.py +158 -0
  173. raystack/stencil/v1beta1/stencil_pb2.pyi +357 -0
  174. raystack_proton-2025.11.5.35259.dist-info/METADATA +103 -0
  175. raystack_proton-2025.11.5.35259.dist-info/RECORD +181 -0
  176. raystack_proton-2025.11.5.35259.dist-info/WHEEL +5 -0
  177. raystack_proton-2025.11.5.35259.dist-info/licenses/LICENSE +201 -0
  178. raystack_proton-2025.11.5.35259.dist-info/top_level.txt +4 -0
  179. validate/__init__.py +0 -0
  180. validate/validate_pb2.py +86 -0
  181. validate/validate_pb2.pyi +497 -0
@@ -0,0 +1,314 @@
1
+ # -*- coding: utf-8 -*-
2
+ # Generated by https://github.com/connectrpc/connect-python. DO NOT EDIT!
3
+ # source: raystack/optimus/core/v1beta1/job_run.proto
4
+
5
+ from collections.abc import AsyncIterator, Iterable, Iterator, Mapping
6
+ from typing import Protocol
7
+
8
+ from connectrpc.client import ConnectClient, ConnectClientSync
9
+ from connectrpc.code import Code
10
+ from connectrpc.errors import ConnectError
11
+ from connectrpc.interceptor import Interceptor, InterceptorSync
12
+ from connectrpc.method import IdempotencyLevel, MethodInfo
13
+ from connectrpc.request import Headers, RequestContext
14
+ from connectrpc.server import ConnectASGIApplication, ConnectWSGIApplication, Endpoint, EndpointSync
15
+ import raystack.optimus.core.v1beta1.job_run_pb2 as raystack_dot_optimus_dot_core_dot_v1beta1_dot_job__run__pb2
16
+
17
+
18
+ class JobRunService(Protocol):
19
+ async def job_run_input(self, request: raystack_dot_optimus_dot_core_dot_v1beta1_dot_job__run__pb2.JobRunInputRequest, ctx: RequestContext) -> raystack_dot_optimus_dot_core_dot_v1beta1_dot_job__run__pb2.JobRunInputResponse:
20
+ raise ConnectError(Code.UNIMPLEMENTED, "Not implemented")
21
+
22
+ async def job_run(self, request: raystack_dot_optimus_dot_core_dot_v1beta1_dot_job__run__pb2.JobRunRequest, ctx: RequestContext) -> raystack_dot_optimus_dot_core_dot_v1beta1_dot_job__run__pb2.JobRunResponse:
23
+ raise ConnectError(Code.UNIMPLEMENTED, "Not implemented")
24
+
25
+ async def register_job_event(self, request: raystack_dot_optimus_dot_core_dot_v1beta1_dot_job__run__pb2.RegisterJobEventRequest, ctx: RequestContext) -> raystack_dot_optimus_dot_core_dot_v1beta1_dot_job__run__pb2.RegisterJobEventResponse:
26
+ raise ConnectError(Code.UNIMPLEMENTED, "Not implemented")
27
+
28
+ async def upload_to_scheduler(self, request: raystack_dot_optimus_dot_core_dot_v1beta1_dot_job__run__pb2.UploadToSchedulerRequest, ctx: RequestContext) -> raystack_dot_optimus_dot_core_dot_v1beta1_dot_job__run__pb2.UploadToSchedulerResponse:
29
+ raise ConnectError(Code.UNIMPLEMENTED, "Not implemented")
30
+
31
+
32
+ class JobRunServiceASGIApplication(ConnectASGIApplication):
33
+ def __init__(self, service: JobRunService, *, interceptors: Iterable[Interceptor]=(), read_max_bytes: int | None = None) -> None:
34
+ super().__init__(
35
+ endpoints={
36
+ "/raystack.optimus.core.v1beta1.JobRunService/JobRunInput": Endpoint.unary(
37
+ method=MethodInfo(
38
+ name="JobRunInput",
39
+ service_name="raystack.optimus.core.v1beta1.JobRunService",
40
+ input=raystack_dot_optimus_dot_core_dot_v1beta1_dot_job__run__pb2.JobRunInputRequest,
41
+ output=raystack_dot_optimus_dot_core_dot_v1beta1_dot_job__run__pb2.JobRunInputResponse,
42
+ idempotency_level=IdempotencyLevel.UNKNOWN,
43
+ ),
44
+ function=service.job_run_input,
45
+ ),
46
+ "/raystack.optimus.core.v1beta1.JobRunService/JobRun": Endpoint.unary(
47
+ method=MethodInfo(
48
+ name="JobRun",
49
+ service_name="raystack.optimus.core.v1beta1.JobRunService",
50
+ input=raystack_dot_optimus_dot_core_dot_v1beta1_dot_job__run__pb2.JobRunRequest,
51
+ output=raystack_dot_optimus_dot_core_dot_v1beta1_dot_job__run__pb2.JobRunResponse,
52
+ idempotency_level=IdempotencyLevel.UNKNOWN,
53
+ ),
54
+ function=service.job_run,
55
+ ),
56
+ "/raystack.optimus.core.v1beta1.JobRunService/RegisterJobEvent": Endpoint.unary(
57
+ method=MethodInfo(
58
+ name="RegisterJobEvent",
59
+ service_name="raystack.optimus.core.v1beta1.JobRunService",
60
+ input=raystack_dot_optimus_dot_core_dot_v1beta1_dot_job__run__pb2.RegisterJobEventRequest,
61
+ output=raystack_dot_optimus_dot_core_dot_v1beta1_dot_job__run__pb2.RegisterJobEventResponse,
62
+ idempotency_level=IdempotencyLevel.UNKNOWN,
63
+ ),
64
+ function=service.register_job_event,
65
+ ),
66
+ "/raystack.optimus.core.v1beta1.JobRunService/UploadToScheduler": Endpoint.unary(
67
+ method=MethodInfo(
68
+ name="UploadToScheduler",
69
+ service_name="raystack.optimus.core.v1beta1.JobRunService",
70
+ input=raystack_dot_optimus_dot_core_dot_v1beta1_dot_job__run__pb2.UploadToSchedulerRequest,
71
+ output=raystack_dot_optimus_dot_core_dot_v1beta1_dot_job__run__pb2.UploadToSchedulerResponse,
72
+ idempotency_level=IdempotencyLevel.UNKNOWN,
73
+ ),
74
+ function=service.upload_to_scheduler,
75
+ ),
76
+ },
77
+ interceptors=interceptors,
78
+ read_max_bytes=read_max_bytes,
79
+ )
80
+
81
+ @property
82
+ def path(self) -> str:
83
+ """Returns the URL path to mount the application to when serving multiple applications."""
84
+ return "/raystack.optimus.core.v1beta1.JobRunService"
85
+
86
+
87
+ class JobRunServiceClient(ConnectClient):
88
+ async def job_run_input(
89
+ self,
90
+ request: raystack_dot_optimus_dot_core_dot_v1beta1_dot_job__run__pb2.JobRunInputRequest,
91
+ *,
92
+ headers: Headers | Mapping[str, str] | None = None,
93
+ timeout_ms: int | None = None,
94
+ ) -> raystack_dot_optimus_dot_core_dot_v1beta1_dot_job__run__pb2.JobRunInputResponse:
95
+ return await self.execute_unary(
96
+ request=request,
97
+ method=MethodInfo(
98
+ name="JobRunInput",
99
+ service_name="raystack.optimus.core.v1beta1.JobRunService",
100
+ input=raystack_dot_optimus_dot_core_dot_v1beta1_dot_job__run__pb2.JobRunInputRequest,
101
+ output=raystack_dot_optimus_dot_core_dot_v1beta1_dot_job__run__pb2.JobRunInputResponse,
102
+ idempotency_level=IdempotencyLevel.UNKNOWN,
103
+ ),
104
+ headers=headers,
105
+ timeout_ms=timeout_ms,
106
+ )
107
+
108
+ async def job_run(
109
+ self,
110
+ request: raystack_dot_optimus_dot_core_dot_v1beta1_dot_job__run__pb2.JobRunRequest,
111
+ *,
112
+ headers: Headers | Mapping[str, str] | None = None,
113
+ timeout_ms: int | None = None,
114
+ ) -> raystack_dot_optimus_dot_core_dot_v1beta1_dot_job__run__pb2.JobRunResponse:
115
+ return await self.execute_unary(
116
+ request=request,
117
+ method=MethodInfo(
118
+ name="JobRun",
119
+ service_name="raystack.optimus.core.v1beta1.JobRunService",
120
+ input=raystack_dot_optimus_dot_core_dot_v1beta1_dot_job__run__pb2.JobRunRequest,
121
+ output=raystack_dot_optimus_dot_core_dot_v1beta1_dot_job__run__pb2.JobRunResponse,
122
+ idempotency_level=IdempotencyLevel.UNKNOWN,
123
+ ),
124
+ headers=headers,
125
+ timeout_ms=timeout_ms,
126
+ )
127
+
128
+ async def register_job_event(
129
+ self,
130
+ request: raystack_dot_optimus_dot_core_dot_v1beta1_dot_job__run__pb2.RegisterJobEventRequest,
131
+ *,
132
+ headers: Headers | Mapping[str, str] | None = None,
133
+ timeout_ms: int | None = None,
134
+ ) -> raystack_dot_optimus_dot_core_dot_v1beta1_dot_job__run__pb2.RegisterJobEventResponse:
135
+ return await self.execute_unary(
136
+ request=request,
137
+ method=MethodInfo(
138
+ name="RegisterJobEvent",
139
+ service_name="raystack.optimus.core.v1beta1.JobRunService",
140
+ input=raystack_dot_optimus_dot_core_dot_v1beta1_dot_job__run__pb2.RegisterJobEventRequest,
141
+ output=raystack_dot_optimus_dot_core_dot_v1beta1_dot_job__run__pb2.RegisterJobEventResponse,
142
+ idempotency_level=IdempotencyLevel.UNKNOWN,
143
+ ),
144
+ headers=headers,
145
+ timeout_ms=timeout_ms,
146
+ )
147
+
148
+ async def upload_to_scheduler(
149
+ self,
150
+ request: raystack_dot_optimus_dot_core_dot_v1beta1_dot_job__run__pb2.UploadToSchedulerRequest,
151
+ *,
152
+ headers: Headers | Mapping[str, str] | None = None,
153
+ timeout_ms: int | None = None,
154
+ ) -> raystack_dot_optimus_dot_core_dot_v1beta1_dot_job__run__pb2.UploadToSchedulerResponse:
155
+ return await self.execute_unary(
156
+ request=request,
157
+ method=MethodInfo(
158
+ name="UploadToScheduler",
159
+ service_name="raystack.optimus.core.v1beta1.JobRunService",
160
+ input=raystack_dot_optimus_dot_core_dot_v1beta1_dot_job__run__pb2.UploadToSchedulerRequest,
161
+ output=raystack_dot_optimus_dot_core_dot_v1beta1_dot_job__run__pb2.UploadToSchedulerResponse,
162
+ idempotency_level=IdempotencyLevel.UNKNOWN,
163
+ ),
164
+ headers=headers,
165
+ timeout_ms=timeout_ms,
166
+ )
167
+
168
+
169
+ class JobRunServiceSync(Protocol):
170
+ def job_run_input(self, request: raystack_dot_optimus_dot_core_dot_v1beta1_dot_job__run__pb2.JobRunInputRequest, ctx: RequestContext) -> raystack_dot_optimus_dot_core_dot_v1beta1_dot_job__run__pb2.JobRunInputResponse:
171
+ raise ConnectError(Code.UNIMPLEMENTED, "Not implemented")
172
+ def job_run(self, request: raystack_dot_optimus_dot_core_dot_v1beta1_dot_job__run__pb2.JobRunRequest, ctx: RequestContext) -> raystack_dot_optimus_dot_core_dot_v1beta1_dot_job__run__pb2.JobRunResponse:
173
+ raise ConnectError(Code.UNIMPLEMENTED, "Not implemented")
174
+ def register_job_event(self, request: raystack_dot_optimus_dot_core_dot_v1beta1_dot_job__run__pb2.RegisterJobEventRequest, ctx: RequestContext) -> raystack_dot_optimus_dot_core_dot_v1beta1_dot_job__run__pb2.RegisterJobEventResponse:
175
+ raise ConnectError(Code.UNIMPLEMENTED, "Not implemented")
176
+ def upload_to_scheduler(self, request: raystack_dot_optimus_dot_core_dot_v1beta1_dot_job__run__pb2.UploadToSchedulerRequest, ctx: RequestContext) -> raystack_dot_optimus_dot_core_dot_v1beta1_dot_job__run__pb2.UploadToSchedulerResponse:
177
+ raise ConnectError(Code.UNIMPLEMENTED, "Not implemented")
178
+
179
+
180
+ class JobRunServiceWSGIApplication(ConnectWSGIApplication):
181
+ def __init__(self, service: JobRunServiceSync, interceptors: Iterable[InterceptorSync]=(), read_max_bytes: int | None = None) -> None:
182
+ super().__init__(
183
+ endpoints={
184
+ "/raystack.optimus.core.v1beta1.JobRunService/JobRunInput": EndpointSync.unary(
185
+ method=MethodInfo(
186
+ name="JobRunInput",
187
+ service_name="raystack.optimus.core.v1beta1.JobRunService",
188
+ input=raystack_dot_optimus_dot_core_dot_v1beta1_dot_job__run__pb2.JobRunInputRequest,
189
+ output=raystack_dot_optimus_dot_core_dot_v1beta1_dot_job__run__pb2.JobRunInputResponse,
190
+ idempotency_level=IdempotencyLevel.UNKNOWN,
191
+ ),
192
+ function=service.job_run_input,
193
+ ),
194
+ "/raystack.optimus.core.v1beta1.JobRunService/JobRun": EndpointSync.unary(
195
+ method=MethodInfo(
196
+ name="JobRun",
197
+ service_name="raystack.optimus.core.v1beta1.JobRunService",
198
+ input=raystack_dot_optimus_dot_core_dot_v1beta1_dot_job__run__pb2.JobRunRequest,
199
+ output=raystack_dot_optimus_dot_core_dot_v1beta1_dot_job__run__pb2.JobRunResponse,
200
+ idempotency_level=IdempotencyLevel.UNKNOWN,
201
+ ),
202
+ function=service.job_run,
203
+ ),
204
+ "/raystack.optimus.core.v1beta1.JobRunService/RegisterJobEvent": EndpointSync.unary(
205
+ method=MethodInfo(
206
+ name="RegisterJobEvent",
207
+ service_name="raystack.optimus.core.v1beta1.JobRunService",
208
+ input=raystack_dot_optimus_dot_core_dot_v1beta1_dot_job__run__pb2.RegisterJobEventRequest,
209
+ output=raystack_dot_optimus_dot_core_dot_v1beta1_dot_job__run__pb2.RegisterJobEventResponse,
210
+ idempotency_level=IdempotencyLevel.UNKNOWN,
211
+ ),
212
+ function=service.register_job_event,
213
+ ),
214
+ "/raystack.optimus.core.v1beta1.JobRunService/UploadToScheduler": EndpointSync.unary(
215
+ method=MethodInfo(
216
+ name="UploadToScheduler",
217
+ service_name="raystack.optimus.core.v1beta1.JobRunService",
218
+ input=raystack_dot_optimus_dot_core_dot_v1beta1_dot_job__run__pb2.UploadToSchedulerRequest,
219
+ output=raystack_dot_optimus_dot_core_dot_v1beta1_dot_job__run__pb2.UploadToSchedulerResponse,
220
+ idempotency_level=IdempotencyLevel.UNKNOWN,
221
+ ),
222
+ function=service.upload_to_scheduler,
223
+ ),
224
+ },
225
+ interceptors=interceptors,
226
+ read_max_bytes=read_max_bytes,
227
+ )
228
+
229
+ @property
230
+ def path(self) -> str:
231
+ """Returns the URL path to mount the application to when serving multiple applications."""
232
+ return "/raystack.optimus.core.v1beta1.JobRunService"
233
+
234
+
235
+ class JobRunServiceClientSync(ConnectClientSync):
236
+ def job_run_input(
237
+ self,
238
+ request: raystack_dot_optimus_dot_core_dot_v1beta1_dot_job__run__pb2.JobRunInputRequest,
239
+ *,
240
+ headers: Headers | Mapping[str, str] | None = None,
241
+ timeout_ms: int | None = None,
242
+ ) -> raystack_dot_optimus_dot_core_dot_v1beta1_dot_job__run__pb2.JobRunInputResponse:
243
+ return self.execute_unary(
244
+ request=request,
245
+ method=MethodInfo(
246
+ name="JobRunInput",
247
+ service_name="raystack.optimus.core.v1beta1.JobRunService",
248
+ input=raystack_dot_optimus_dot_core_dot_v1beta1_dot_job__run__pb2.JobRunInputRequest,
249
+ output=raystack_dot_optimus_dot_core_dot_v1beta1_dot_job__run__pb2.JobRunInputResponse,
250
+ idempotency_level=IdempotencyLevel.UNKNOWN,
251
+ ),
252
+ headers=headers,
253
+ timeout_ms=timeout_ms,
254
+ )
255
+
256
+ def job_run(
257
+ self,
258
+ request: raystack_dot_optimus_dot_core_dot_v1beta1_dot_job__run__pb2.JobRunRequest,
259
+ *,
260
+ headers: Headers | Mapping[str, str] | None = None,
261
+ timeout_ms: int | None = None,
262
+ ) -> raystack_dot_optimus_dot_core_dot_v1beta1_dot_job__run__pb2.JobRunResponse:
263
+ return self.execute_unary(
264
+ request=request,
265
+ method=MethodInfo(
266
+ name="JobRun",
267
+ service_name="raystack.optimus.core.v1beta1.JobRunService",
268
+ input=raystack_dot_optimus_dot_core_dot_v1beta1_dot_job__run__pb2.JobRunRequest,
269
+ output=raystack_dot_optimus_dot_core_dot_v1beta1_dot_job__run__pb2.JobRunResponse,
270
+ idempotency_level=IdempotencyLevel.UNKNOWN,
271
+ ),
272
+ headers=headers,
273
+ timeout_ms=timeout_ms,
274
+ )
275
+
276
+ def register_job_event(
277
+ self,
278
+ request: raystack_dot_optimus_dot_core_dot_v1beta1_dot_job__run__pb2.RegisterJobEventRequest,
279
+ *,
280
+ headers: Headers | Mapping[str, str] | None = None,
281
+ timeout_ms: int | None = None,
282
+ ) -> raystack_dot_optimus_dot_core_dot_v1beta1_dot_job__run__pb2.RegisterJobEventResponse:
283
+ return self.execute_unary(
284
+ request=request,
285
+ method=MethodInfo(
286
+ name="RegisterJobEvent",
287
+ service_name="raystack.optimus.core.v1beta1.JobRunService",
288
+ input=raystack_dot_optimus_dot_core_dot_v1beta1_dot_job__run__pb2.RegisterJobEventRequest,
289
+ output=raystack_dot_optimus_dot_core_dot_v1beta1_dot_job__run__pb2.RegisterJobEventResponse,
290
+ idempotency_level=IdempotencyLevel.UNKNOWN,
291
+ ),
292
+ headers=headers,
293
+ timeout_ms=timeout_ms,
294
+ )
295
+
296
+ def upload_to_scheduler(
297
+ self,
298
+ request: raystack_dot_optimus_dot_core_dot_v1beta1_dot_job__run__pb2.UploadToSchedulerRequest,
299
+ *,
300
+ headers: Headers | Mapping[str, str] | None = None,
301
+ timeout_ms: int | None = None,
302
+ ) -> raystack_dot_optimus_dot_core_dot_v1beta1_dot_job__run__pb2.UploadToSchedulerResponse:
303
+ return self.execute_unary(
304
+ request=request,
305
+ method=MethodInfo(
306
+ name="UploadToScheduler",
307
+ service_name="raystack.optimus.core.v1beta1.JobRunService",
308
+ input=raystack_dot_optimus_dot_core_dot_v1beta1_dot_job__run__pb2.UploadToSchedulerRequest,
309
+ output=raystack_dot_optimus_dot_core_dot_v1beta1_dot_job__run__pb2.UploadToSchedulerResponse,
310
+ idempotency_level=IdempotencyLevel.UNKNOWN,
311
+ ),
312
+ headers=headers,
313
+ timeout_ms=timeout_ms,
314
+ )
@@ -0,0 +1,88 @@
1
+ # -*- coding: utf-8 -*-
2
+ # Generated by the protocol buffer compiler. DO NOT EDIT!
3
+ # NO CHECKED-IN PROTOBUF GENCODE
4
+ # source: raystack/optimus/core/v1beta1/job_run.proto
5
+ # Protobuf Python Version: 6.33.0
6
+ """Generated protocol buffer code."""
7
+ from google.protobuf import descriptor as _descriptor
8
+ from google.protobuf import descriptor_pool as _descriptor_pool
9
+ from google.protobuf import runtime_version as _runtime_version
10
+ from google.protobuf import symbol_database as _symbol_database
11
+ from google.protobuf.internal import builder as _builder
12
+ _runtime_version.ValidateProtobufRuntimeVersion(
13
+ _runtime_version.Domain.PUBLIC,
14
+ 6,
15
+ 33,
16
+ 0,
17
+ '',
18
+ 'raystack/optimus/core/v1beta1/job_run.proto'
19
+ )
20
+ # @@protoc_insertion_point(imports)
21
+
22
+ _sym_db = _symbol_database.Default()
23
+
24
+
25
+ from google.api import annotations_pb2 as google_dot_api_dot_annotations__pb2
26
+ from google.protobuf import duration_pb2 as google_dot_protobuf_dot_duration__pb2
27
+ from google.protobuf import timestamp_pb2 as google_dot_protobuf_dot_timestamp__pb2
28
+ from protoc_gen_openapiv2.options import annotations_pb2 as protoc__gen__openapiv2_dot_options_dot_annotations__pb2
29
+ from raystack.optimus.core.v1beta1 import job_spec_pb2 as raystack_dot_optimus_dot_core_dot_v1beta1_dot_job__spec__pb2
30
+
31
+
32
+ DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n+raystack/optimus/core/v1beta1/job_run.proto\x12\x1draystack.optimus.core.v1beta1\x1a\x1cgoogle/api/annotations.proto\x1a\x1egoogle/protobuf/duration.proto\x1a\x1fgoogle/protobuf/timestamp.proto\x1a.protoc-gen-openapiv2/options/annotations.proto\x1a,raystack/optimus/core/v1beta1/job_spec.proto\"|\n\x18UploadToSchedulerRequest\x12!\n\x0cproject_name\x18\x01 \x01(\tR\x0bprojectName\x12*\n\x0enamespace_name\x18\x02 \x01(\tH\x00R\rnamespaceName\x88\x01\x01\x42\x11\n\x0f_namespace_name\"X\n\x19UploadToSchedulerResponse\x12\x16\n\x06status\x18\x01 \x01(\x08R\x06status\x12#\n\rerror_message\x18\x02 \x01(\tR\x0c\x65rrorMessage\"\xbd\x01\n\x17RegisterJobEventRequest\x12!\n\x0cproject_name\x18\x01 \x01(\tR\x0bprojectName\x12\x19\n\x08job_name\x18\x02 \x01(\tR\x07jobName\x12%\n\x0enamespace_name\x18\x03 \x01(\tR\rnamespaceName\x12=\n\x05\x65vent\x18\x04 \x01(\x0b\x32\'.raystack.optimus.core.v1beta1.JobEventR\x05\x65vent\"\x1a\n\x18RegisterJobEventResponse\"\xb0\x02\n\x12JobRunInputRequest\x12!\n\x0cproject_name\x18\x01 \x01(\tR\x0bprojectName\x12\x19\n\x08job_name\x18\x02 \x01(\tR\x07jobName\x12=\n\x0cscheduled_at\x18\x04 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\x0bscheduledAt\x12#\n\rinstance_name\x18\x05 \x01(\tR\x0cinstanceName\x12U\n\rinstance_type\x18\x06 \x01(\x0e\x32\x30.raystack.optimus.core.v1beta1.InstanceSpec.TypeR\x0cinstanceType\x12\x1b\n\tjobrun_id\x18\x07 \x01(\tR\x08jobrunIdJ\x04\x08\x03\x10\x04\"\xd7\x01\n\rJobRunRequest\x12!\n\x0cproject_name\x18\x01 \x01(\tR\x0bprojectName\x12\x19\n\x08job_name\x18\x02 \x01(\tR\x07jobName\x12\x39\n\nstart_date\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tstartDate\x12\x35\n\x08\x65nd_date\x18\x04 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\x07\x65ndDate\x12\x16\n\x06\x66ilter\x18\x05 \x03(\tR\x06\x66ilter\"R\n\x0eJobRunResponse\x12@\n\x08job_runs\x18\x01 \x03(\x0b\x32%.raystack.optimus.core.v1beta1.JobRunR\x07jobRuns\"\xc8\x02\n\x0cInstanceSpec\x12\x14\n\x05state\x18\x01 \x01(\tR\x05state\x12\x43\n\x04\x64\x61ta\x18\x03 \x03(\x0b\x32/.raystack.optimus.core.v1beta1.InstanceSpecDataR\x04\x64\x61ta\x12;\n\x0b\x65xecuted_at\x18\x05 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\nexecutedAt\x12\x12\n\x04name\x18\x06 \x01(\tR\x04name\x12\x44\n\x04type\x18\x07 \x01(\x0e\x32\x30.raystack.optimus.core.v1beta1.InstanceSpec.TypeR\x04type\":\n\x04Type\x12\x14\n\x10TYPE_UNSPECIFIED\x10\x00\x12\r\n\tTYPE_TASK\x10\x01\x12\r\n\tTYPE_HOOK\x10\x02J\x04\x08\x02\x10\x03J\x04\x08\x04\x10\x05\"\xcd\x01\n\x10InstanceSpecData\x12\x12\n\x04name\x18\x01 \x01(\tR\x04name\x12\x14\n\x05value\x18\x02 \x01(\tR\x05value\x12H\n\x04type\x18\x05 \x01(\x0e\x32\x34.raystack.optimus.core.v1beta1.InstanceSpecData.TypeR\x04type\"9\n\x04Type\x12\x14\n\x10TYPE_UNSPECIFIED\x10\x00\x12\x0c\n\x08TYPE_ENV\x10\x01\x12\r\n\tTYPE_FILE\x10\x02J\x04\x08\x03\x10\x04J\x04\x08\x04\x10\x05\"\xc6\x03\n\x13JobRunInputResponse\x12P\n\x04\x65nvs\x18\x01 \x03(\x0b\x32<.raystack.optimus.core.v1beta1.JobRunInputResponse.EnvsEntryR\x04\x65nvs\x12S\n\x05\x66iles\x18\x02 \x03(\x0b\x32=.raystack.optimus.core.v1beta1.JobRunInputResponse.FilesEntryR\x05\x66iles\x12Y\n\x07secrets\x18\x03 \x03(\x0b\x32?.raystack.optimus.core.v1beta1.JobRunInputResponse.SecretsEntryR\x07secrets\x1a\x37\n\tEnvsEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\x14\n\x05value\x18\x02 \x01(\tR\x05value:\x02\x38\x01\x1a\x38\n\nFilesEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\x14\n\x05value\x18\x02 \x01(\tR\x05value:\x02\x38\x01\x1a:\n\x0cSecretsEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\x14\n\x05value\x18\x02 \x01(\tR\x05value:\x02\x38\x01\"\x8f\x01\n\nTaskWindow\x12-\n\x04size\x18\x01 \x01(\x0b\x32\x19.google.protobuf.DurationR\x04size\x12\x31\n\x06offset\x18\x02 \x01(\x0b\x32\x19.google.protobuf.DurationR\x06offset\x12\x1f\n\x0btruncate_to\x18\x03 \x01(\tR\ntruncateTo2\x8d\x06\n\rJobRunService\x12\xb9\x01\n\x0bJobRunInput\x12\x31.raystack.optimus.core.v1beta1.JobRunInputRequest\x1a\x32.raystack.optimus.core.v1beta1.JobRunInputResponse\"C\x82\xd3\xe4\x93\x02=\"8/v1beta1/project/{project_name}/job/{job_name}/run_input:\x01*\x12\xa1\x01\n\x06JobRun\x12,.raystack.optimus.core.v1beta1.JobRunRequest\x1a-.raystack.optimus.core.v1beta1.JobRunResponse\":\x82\xd3\xe4\x93\x02\x34\x12\x32/v1beta1/project/{project_name}/job/{job_name}/run\x12\xdf\x01\n\x10RegisterJobEvent\x12\x36.raystack.optimus.core.v1beta1.RegisterJobEventRequest\x1a\x37.raystack.optimus.core.v1beta1.RegisterJobEventResponse\"Z\x82\xd3\xe4\x93\x02T\"O/v1beta1/project/{project_name}/namespace/{namespace_name}/job/{job_name}/event:\x01*\x12\xb9\x01\n\x11UploadToScheduler\x12\x37.raystack.optimus.core.v1beta1.UploadToSchedulerRequest\x1a\x38.raystack.optimus.core.v1beta1.UploadToSchedulerResponse\"1\x82\xd3\xe4\x93\x02+\x1a&/v1beta1/project/{project_name}/upload:\x01*B\x90\x01\n\x1borg.raystack.proton.optimusB\rJobRunManagerP\x01Z\"github.com/raystack/proton/optimus\x92\x41;\x12\x05\x32\x03\x30.1\x1a\x0e\x31\x32\x37.0.0.1:9100\"\x04/api*\x01\x01r\x19\n\x17Optimus Job Run Serviceb\x06proto3')
33
+
34
+ _globals = globals()
35
+ _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
36
+ _builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'raystack.optimus.core.v1beta1.job_run_pb2', _globals)
37
+ if not _descriptor._USE_C_DESCRIPTORS:
38
+ _globals['DESCRIPTOR']._loaded_options = None
39
+ _globals['DESCRIPTOR']._serialized_options = b'\n\033org.raystack.proton.optimusB\rJobRunManagerP\001Z\"github.com/raystack/proton/optimus\222A;\022\0052\0030.1\032\016127.0.0.1:9100\"\004/api*\001\001r\031\n\027Optimus Job Run Service'
40
+ _globals['_JOBRUNINPUTRESPONSE_ENVSENTRY']._loaded_options = None
41
+ _globals['_JOBRUNINPUTRESPONSE_ENVSENTRY']._serialized_options = b'8\001'
42
+ _globals['_JOBRUNINPUTRESPONSE_FILESENTRY']._loaded_options = None
43
+ _globals['_JOBRUNINPUTRESPONSE_FILESENTRY']._serialized_options = b'8\001'
44
+ _globals['_JOBRUNINPUTRESPONSE_SECRETSENTRY']._loaded_options = None
45
+ _globals['_JOBRUNINPUTRESPONSE_SECRETSENTRY']._serialized_options = b'8\001'
46
+ _globals['_JOBRUNSERVICE'].methods_by_name['JobRunInput']._loaded_options = None
47
+ _globals['_JOBRUNSERVICE'].methods_by_name['JobRunInput']._serialized_options = b'\202\323\344\223\002=\"8/v1beta1/project/{project_name}/job/{job_name}/run_input:\001*'
48
+ _globals['_JOBRUNSERVICE'].methods_by_name['JobRun']._loaded_options = None
49
+ _globals['_JOBRUNSERVICE'].methods_by_name['JobRun']._serialized_options = b'\202\323\344\223\0024\0222/v1beta1/project/{project_name}/job/{job_name}/run'
50
+ _globals['_JOBRUNSERVICE'].methods_by_name['RegisterJobEvent']._loaded_options = None
51
+ _globals['_JOBRUNSERVICE'].methods_by_name['RegisterJobEvent']._serialized_options = b'\202\323\344\223\002T\"O/v1beta1/project/{project_name}/namespace/{namespace_name}/job/{job_name}/event:\001*'
52
+ _globals['_JOBRUNSERVICE'].methods_by_name['UploadToScheduler']._loaded_options = None
53
+ _globals['_JOBRUNSERVICE'].methods_by_name['UploadToScheduler']._serialized_options = b'\202\323\344\223\002+\032&/v1beta1/project/{project_name}/upload:\001*'
54
+ _globals['_UPLOADTOSCHEDULERREQUEST']._serialized_start=267
55
+ _globals['_UPLOADTOSCHEDULERREQUEST']._serialized_end=391
56
+ _globals['_UPLOADTOSCHEDULERRESPONSE']._serialized_start=393
57
+ _globals['_UPLOADTOSCHEDULERRESPONSE']._serialized_end=481
58
+ _globals['_REGISTERJOBEVENTREQUEST']._serialized_start=484
59
+ _globals['_REGISTERJOBEVENTREQUEST']._serialized_end=673
60
+ _globals['_REGISTERJOBEVENTRESPONSE']._serialized_start=675
61
+ _globals['_REGISTERJOBEVENTRESPONSE']._serialized_end=701
62
+ _globals['_JOBRUNINPUTREQUEST']._serialized_start=704
63
+ _globals['_JOBRUNINPUTREQUEST']._serialized_end=1008
64
+ _globals['_JOBRUNREQUEST']._serialized_start=1011
65
+ _globals['_JOBRUNREQUEST']._serialized_end=1226
66
+ _globals['_JOBRUNRESPONSE']._serialized_start=1228
67
+ _globals['_JOBRUNRESPONSE']._serialized_end=1310
68
+ _globals['_INSTANCESPEC']._serialized_start=1313
69
+ _globals['_INSTANCESPEC']._serialized_end=1641
70
+ _globals['_INSTANCESPEC_TYPE']._serialized_start=1571
71
+ _globals['_INSTANCESPEC_TYPE']._serialized_end=1629
72
+ _globals['_INSTANCESPECDATA']._serialized_start=1644
73
+ _globals['_INSTANCESPECDATA']._serialized_end=1849
74
+ _globals['_INSTANCESPECDATA_TYPE']._serialized_start=1780
75
+ _globals['_INSTANCESPECDATA_TYPE']._serialized_end=1837
76
+ _globals['_JOBRUNINPUTRESPONSE']._serialized_start=1852
77
+ _globals['_JOBRUNINPUTRESPONSE']._serialized_end=2306
78
+ _globals['_JOBRUNINPUTRESPONSE_ENVSENTRY']._serialized_start=2133
79
+ _globals['_JOBRUNINPUTRESPONSE_ENVSENTRY']._serialized_end=2188
80
+ _globals['_JOBRUNINPUTRESPONSE_FILESENTRY']._serialized_start=2190
81
+ _globals['_JOBRUNINPUTRESPONSE_FILESENTRY']._serialized_end=2246
82
+ _globals['_JOBRUNINPUTRESPONSE_SECRETSENTRY']._serialized_start=2248
83
+ _globals['_JOBRUNINPUTRESPONSE_SECRETSENTRY']._serialized_end=2306
84
+ _globals['_TASKWINDOW']._serialized_start=2309
85
+ _globals['_TASKWINDOW']._serialized_end=2452
86
+ _globals['_JOBRUNSERVICE']._serialized_start=2455
87
+ _globals['_JOBRUNSERVICE']._serialized_end=3236
88
+ # @@protoc_insertion_point(module_scope)
@@ -0,0 +1,164 @@
1
+ import datetime
2
+
3
+ from google.api import annotations_pb2 as _annotations_pb2
4
+ from google.protobuf import duration_pb2 as _duration_pb2
5
+ from google.protobuf import timestamp_pb2 as _timestamp_pb2
6
+ from protoc_gen_openapiv2.options import annotations_pb2 as _annotations_pb2_1
7
+ from raystack.optimus.core.v1beta1 import job_spec_pb2 as _job_spec_pb2
8
+ from google.protobuf.internal import containers as _containers
9
+ from google.protobuf.internal import enum_type_wrapper as _enum_type_wrapper
10
+ from google.protobuf import descriptor as _descriptor
11
+ from google.protobuf import message as _message
12
+ from collections.abc import Iterable as _Iterable, Mapping as _Mapping
13
+ from typing import ClassVar as _ClassVar, Optional as _Optional, Union as _Union
14
+
15
+ DESCRIPTOR: _descriptor.FileDescriptor
16
+
17
+ class UploadToSchedulerRequest(_message.Message):
18
+ __slots__ = ()
19
+ PROJECT_NAME_FIELD_NUMBER: _ClassVar[int]
20
+ NAMESPACE_NAME_FIELD_NUMBER: _ClassVar[int]
21
+ project_name: str
22
+ namespace_name: str
23
+ def __init__(self, project_name: _Optional[str] = ..., namespace_name: _Optional[str] = ...) -> None: ...
24
+
25
+ class UploadToSchedulerResponse(_message.Message):
26
+ __slots__ = ()
27
+ STATUS_FIELD_NUMBER: _ClassVar[int]
28
+ ERROR_MESSAGE_FIELD_NUMBER: _ClassVar[int]
29
+ status: bool
30
+ error_message: str
31
+ def __init__(self, status: _Optional[bool] = ..., error_message: _Optional[str] = ...) -> None: ...
32
+
33
+ class RegisterJobEventRequest(_message.Message):
34
+ __slots__ = ()
35
+ PROJECT_NAME_FIELD_NUMBER: _ClassVar[int]
36
+ JOB_NAME_FIELD_NUMBER: _ClassVar[int]
37
+ NAMESPACE_NAME_FIELD_NUMBER: _ClassVar[int]
38
+ EVENT_FIELD_NUMBER: _ClassVar[int]
39
+ project_name: str
40
+ job_name: str
41
+ namespace_name: str
42
+ event: _job_spec_pb2.JobEvent
43
+ def __init__(self, project_name: _Optional[str] = ..., job_name: _Optional[str] = ..., namespace_name: _Optional[str] = ..., event: _Optional[_Union[_job_spec_pb2.JobEvent, _Mapping]] = ...) -> None: ...
44
+
45
+ class RegisterJobEventResponse(_message.Message):
46
+ __slots__ = ()
47
+ def __init__(self) -> None: ...
48
+
49
+ class JobRunInputRequest(_message.Message):
50
+ __slots__ = ()
51
+ PROJECT_NAME_FIELD_NUMBER: _ClassVar[int]
52
+ JOB_NAME_FIELD_NUMBER: _ClassVar[int]
53
+ SCHEDULED_AT_FIELD_NUMBER: _ClassVar[int]
54
+ INSTANCE_NAME_FIELD_NUMBER: _ClassVar[int]
55
+ INSTANCE_TYPE_FIELD_NUMBER: _ClassVar[int]
56
+ JOBRUN_ID_FIELD_NUMBER: _ClassVar[int]
57
+ project_name: str
58
+ job_name: str
59
+ scheduled_at: _timestamp_pb2.Timestamp
60
+ instance_name: str
61
+ instance_type: InstanceSpec.Type
62
+ jobrun_id: str
63
+ def __init__(self, project_name: _Optional[str] = ..., job_name: _Optional[str] = ..., scheduled_at: _Optional[_Union[datetime.datetime, _timestamp_pb2.Timestamp, _Mapping]] = ..., instance_name: _Optional[str] = ..., instance_type: _Optional[_Union[InstanceSpec.Type, str]] = ..., jobrun_id: _Optional[str] = ...) -> None: ...
64
+
65
+ class JobRunRequest(_message.Message):
66
+ __slots__ = ()
67
+ PROJECT_NAME_FIELD_NUMBER: _ClassVar[int]
68
+ JOB_NAME_FIELD_NUMBER: _ClassVar[int]
69
+ START_DATE_FIELD_NUMBER: _ClassVar[int]
70
+ END_DATE_FIELD_NUMBER: _ClassVar[int]
71
+ FILTER_FIELD_NUMBER: _ClassVar[int]
72
+ project_name: str
73
+ job_name: str
74
+ start_date: _timestamp_pb2.Timestamp
75
+ end_date: _timestamp_pb2.Timestamp
76
+ filter: _containers.RepeatedScalarFieldContainer[str]
77
+ def __init__(self, project_name: _Optional[str] = ..., job_name: _Optional[str] = ..., start_date: _Optional[_Union[datetime.datetime, _timestamp_pb2.Timestamp, _Mapping]] = ..., end_date: _Optional[_Union[datetime.datetime, _timestamp_pb2.Timestamp, _Mapping]] = ..., filter: _Optional[_Iterable[str]] = ...) -> None: ...
78
+
79
+ class JobRunResponse(_message.Message):
80
+ __slots__ = ()
81
+ JOB_RUNS_FIELD_NUMBER: _ClassVar[int]
82
+ job_runs: _containers.RepeatedCompositeFieldContainer[_job_spec_pb2.JobRun]
83
+ def __init__(self, job_runs: _Optional[_Iterable[_Union[_job_spec_pb2.JobRun, _Mapping]]] = ...) -> None: ...
84
+
85
+ class InstanceSpec(_message.Message):
86
+ __slots__ = ()
87
+ class Type(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
88
+ __slots__ = ()
89
+ TYPE_UNSPECIFIED: _ClassVar[InstanceSpec.Type]
90
+ TYPE_TASK: _ClassVar[InstanceSpec.Type]
91
+ TYPE_HOOK: _ClassVar[InstanceSpec.Type]
92
+ TYPE_UNSPECIFIED: InstanceSpec.Type
93
+ TYPE_TASK: InstanceSpec.Type
94
+ TYPE_HOOK: InstanceSpec.Type
95
+ STATE_FIELD_NUMBER: _ClassVar[int]
96
+ DATA_FIELD_NUMBER: _ClassVar[int]
97
+ EXECUTED_AT_FIELD_NUMBER: _ClassVar[int]
98
+ NAME_FIELD_NUMBER: _ClassVar[int]
99
+ TYPE_FIELD_NUMBER: _ClassVar[int]
100
+ state: str
101
+ data: _containers.RepeatedCompositeFieldContainer[InstanceSpecData]
102
+ executed_at: _timestamp_pb2.Timestamp
103
+ name: str
104
+ type: InstanceSpec.Type
105
+ def __init__(self, state: _Optional[str] = ..., data: _Optional[_Iterable[_Union[InstanceSpecData, _Mapping]]] = ..., executed_at: _Optional[_Union[datetime.datetime, _timestamp_pb2.Timestamp, _Mapping]] = ..., name: _Optional[str] = ..., type: _Optional[_Union[InstanceSpec.Type, str]] = ...) -> None: ...
106
+
107
+ class InstanceSpecData(_message.Message):
108
+ __slots__ = ()
109
+ class Type(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
110
+ __slots__ = ()
111
+ TYPE_UNSPECIFIED: _ClassVar[InstanceSpecData.Type]
112
+ TYPE_ENV: _ClassVar[InstanceSpecData.Type]
113
+ TYPE_FILE: _ClassVar[InstanceSpecData.Type]
114
+ TYPE_UNSPECIFIED: InstanceSpecData.Type
115
+ TYPE_ENV: InstanceSpecData.Type
116
+ TYPE_FILE: InstanceSpecData.Type
117
+ NAME_FIELD_NUMBER: _ClassVar[int]
118
+ VALUE_FIELD_NUMBER: _ClassVar[int]
119
+ TYPE_FIELD_NUMBER: _ClassVar[int]
120
+ name: str
121
+ value: str
122
+ type: InstanceSpecData.Type
123
+ def __init__(self, name: _Optional[str] = ..., value: _Optional[str] = ..., type: _Optional[_Union[InstanceSpecData.Type, str]] = ...) -> None: ...
124
+
125
+ class JobRunInputResponse(_message.Message):
126
+ __slots__ = ()
127
+ class EnvsEntry(_message.Message):
128
+ __slots__ = ()
129
+ KEY_FIELD_NUMBER: _ClassVar[int]
130
+ VALUE_FIELD_NUMBER: _ClassVar[int]
131
+ key: str
132
+ value: str
133
+ def __init__(self, key: _Optional[str] = ..., value: _Optional[str] = ...) -> None: ...
134
+ class FilesEntry(_message.Message):
135
+ __slots__ = ()
136
+ KEY_FIELD_NUMBER: _ClassVar[int]
137
+ VALUE_FIELD_NUMBER: _ClassVar[int]
138
+ key: str
139
+ value: str
140
+ def __init__(self, key: _Optional[str] = ..., value: _Optional[str] = ...) -> None: ...
141
+ class SecretsEntry(_message.Message):
142
+ __slots__ = ()
143
+ KEY_FIELD_NUMBER: _ClassVar[int]
144
+ VALUE_FIELD_NUMBER: _ClassVar[int]
145
+ key: str
146
+ value: str
147
+ def __init__(self, key: _Optional[str] = ..., value: _Optional[str] = ...) -> None: ...
148
+ ENVS_FIELD_NUMBER: _ClassVar[int]
149
+ FILES_FIELD_NUMBER: _ClassVar[int]
150
+ SECRETS_FIELD_NUMBER: _ClassVar[int]
151
+ envs: _containers.ScalarMap[str, str]
152
+ files: _containers.ScalarMap[str, str]
153
+ secrets: _containers.ScalarMap[str, str]
154
+ def __init__(self, envs: _Optional[_Mapping[str, str]] = ..., files: _Optional[_Mapping[str, str]] = ..., secrets: _Optional[_Mapping[str, str]] = ...) -> None: ...
155
+
156
+ class TaskWindow(_message.Message):
157
+ __slots__ = ()
158
+ SIZE_FIELD_NUMBER: _ClassVar[int]
159
+ OFFSET_FIELD_NUMBER: _ClassVar[int]
160
+ TRUNCATE_TO_FIELD_NUMBER: _ClassVar[int]
161
+ size: _duration_pb2.Duration
162
+ offset: _duration_pb2.Duration
163
+ truncate_to: str
164
+ def __init__(self, size: _Optional[_Union[datetime.timedelta, _duration_pb2.Duration, _Mapping]] = ..., offset: _Optional[_Union[datetime.timedelta, _duration_pb2.Duration, _Mapping]] = ..., truncate_to: _Optional[str] = ...) -> None: ...