raystack-proton 2025.10.8.102837__py3-none-any.whl

raystack/optimus/core/v1beta1/job_run_connect.py

@@ -0,0 +1,314 @@
+# -*- coding: utf-8 -*-
+# Generated by https://github.com/connectrpc/connect-python.  DO NOT EDIT!
+# source: raystack/optimus/core/v1beta1/job_run.proto
+
+from collections.abc import AsyncIterator, Iterable, Iterator, Mapping
+from typing import Protocol
+
+from connectrpc.client import ConnectClient, ConnectClientSync
+from connectrpc.code import Code
+from connectrpc.errors import ConnectError
+from connectrpc.interceptor import Interceptor, InterceptorSync
+from connectrpc.method import IdempotencyLevel, MethodInfo
+from connectrpc.request import Headers, RequestContext
+from connectrpc.server import ConnectASGIApplication, ConnectWSGIApplication, Endpoint, EndpointSync
+import raystack.optimus.core.v1beta1.job_run_pb2 as raystack_dot_optimus_dot_core_dot_v1beta1_dot_job__run__pb2
+
+
+class JobRunService(Protocol):
+    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:
+        raise ConnectError(Code.UNIMPLEMENTED, "Not implemented")
+
+    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:
+        raise ConnectError(Code.UNIMPLEMENTED, "Not implemented")
+
+    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:
+        raise ConnectError(Code.UNIMPLEMENTED, "Not implemented")
+
+    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:
+        raise ConnectError(Code.UNIMPLEMENTED, "Not implemented")
+
+
+class JobRunServiceASGIApplication(ConnectASGIApplication):
+    def __init__(self, service: JobRunService, *, interceptors: Iterable[Interceptor]=(), read_max_bytes: int | None = None) -> None:
+        super().__init__(
+            endpoints={
+                "/raystack.optimus.core.v1beta1.JobRunService/JobRunInput": Endpoint.unary(
+                    method=MethodInfo(
+                        name="JobRunInput",
+                        service_name="raystack.optimus.core.v1beta1.JobRunService",
+                        input=raystack_dot_optimus_dot_core_dot_v1beta1_dot_job__run__pb2.JobRunInputRequest,
+                        output=raystack_dot_optimus_dot_core_dot_v1beta1_dot_job__run__pb2.JobRunInputResponse,
+                        idempotency_level=IdempotencyLevel.UNKNOWN,
+                    ),
+                    function=service.job_run_input,
+                ),
+                "/raystack.optimus.core.v1beta1.JobRunService/JobRun": Endpoint.unary(
+                    method=MethodInfo(
+                        name="JobRun",
+                        service_name="raystack.optimus.core.v1beta1.JobRunService",
+                        input=raystack_dot_optimus_dot_core_dot_v1beta1_dot_job__run__pb2.JobRunRequest,
+                        output=raystack_dot_optimus_dot_core_dot_v1beta1_dot_job__run__pb2.JobRunResponse,
+                        idempotency_level=IdempotencyLevel.UNKNOWN,
+                    ),
+                    function=service.job_run,
+                ),
+                "/raystack.optimus.core.v1beta1.JobRunService/RegisterJobEvent": Endpoint.unary(
+                    method=MethodInfo(
+                        name="RegisterJobEvent",
+                        service_name="raystack.optimus.core.v1beta1.JobRunService",
+                        input=raystack_dot_optimus_dot_core_dot_v1beta1_dot_job__run__pb2.RegisterJobEventRequest,
+                        output=raystack_dot_optimus_dot_core_dot_v1beta1_dot_job__run__pb2.RegisterJobEventResponse,
+                        idempotency_level=IdempotencyLevel.UNKNOWN,
+                    ),
+                    function=service.register_job_event,
+                ),
+                "/raystack.optimus.core.v1beta1.JobRunService/UploadToScheduler": Endpoint.unary(
+                    method=MethodInfo(
+                        name="UploadToScheduler",
+                        service_name="raystack.optimus.core.v1beta1.JobRunService",
+                        input=raystack_dot_optimus_dot_core_dot_v1beta1_dot_job__run__pb2.UploadToSchedulerRequest,
+                        output=raystack_dot_optimus_dot_core_dot_v1beta1_dot_job__run__pb2.UploadToSchedulerResponse,
+                        idempotency_level=IdempotencyLevel.UNKNOWN,
+                    ),
+                    function=service.upload_to_scheduler,
+                ),
+            },
+            interceptors=interceptors,
+            read_max_bytes=read_max_bytes,
+        )
+
+    @property
+    def path(self) -> str:
+        """Returns the URL path to mount the application to when serving multiple applications."""
+        return "/raystack.optimus.core.v1beta1.JobRunService"
+
+
+class JobRunServiceClient(ConnectClient):
+    async def job_run_input(
+        self,
+        request: raystack_dot_optimus_dot_core_dot_v1beta1_dot_job__run__pb2.JobRunInputRequest,
+        *,
+        headers: Headers | Mapping[str, str] | None = None,
+        timeout_ms: int | None = None,
+    ) -> raystack_dot_optimus_dot_core_dot_v1beta1_dot_job__run__pb2.JobRunInputResponse:
+        return await self.execute_unary(
+            request=request,
+            method=MethodInfo(
+                name="JobRunInput",
+                service_name="raystack.optimus.core.v1beta1.JobRunService",
+                input=raystack_dot_optimus_dot_core_dot_v1beta1_dot_job__run__pb2.JobRunInputRequest,
+                output=raystack_dot_optimus_dot_core_dot_v1beta1_dot_job__run__pb2.JobRunInputResponse,
+                idempotency_level=IdempotencyLevel.UNKNOWN,
+            ),
+            headers=headers,
+            timeout_ms=timeout_ms,
+        )
+
+    async def job_run(
+        self,
+        request: raystack_dot_optimus_dot_core_dot_v1beta1_dot_job__run__pb2.JobRunRequest,
+        *,
+        headers: Headers | Mapping[str, str] | None = None,
+        timeout_ms: int | None = None,
+    ) -> raystack_dot_optimus_dot_core_dot_v1beta1_dot_job__run__pb2.JobRunResponse:
+        return await self.execute_unary(
+            request=request,
+            method=MethodInfo(
+                name="JobRun",
+                service_name="raystack.optimus.core.v1beta1.JobRunService",
+                input=raystack_dot_optimus_dot_core_dot_v1beta1_dot_job__run__pb2.JobRunRequest,
+                output=raystack_dot_optimus_dot_core_dot_v1beta1_dot_job__run__pb2.JobRunResponse,
+                idempotency_level=IdempotencyLevel.UNKNOWN,
+            ),
+            headers=headers,
+            timeout_ms=timeout_ms,
+        )
+
+    async def register_job_event(
+        self,
+        request: raystack_dot_optimus_dot_core_dot_v1beta1_dot_job__run__pb2.RegisterJobEventRequest,
+        *,
+        headers: Headers | Mapping[str, str] | None = None,
+        timeout_ms: int | None = None,
+    ) -> raystack_dot_optimus_dot_core_dot_v1beta1_dot_job__run__pb2.RegisterJobEventResponse:
+        return await self.execute_unary(
+            request=request,
+            method=MethodInfo(
+                name="RegisterJobEvent",
+                service_name="raystack.optimus.core.v1beta1.JobRunService",
+                input=raystack_dot_optimus_dot_core_dot_v1beta1_dot_job__run__pb2.RegisterJobEventRequest,
+                output=raystack_dot_optimus_dot_core_dot_v1beta1_dot_job__run__pb2.RegisterJobEventResponse,
+                idempotency_level=IdempotencyLevel.UNKNOWN,
+            ),
+            headers=headers,
+            timeout_ms=timeout_ms,
+        )
+
+    async def upload_to_scheduler(
+        self,
+        request: raystack_dot_optimus_dot_core_dot_v1beta1_dot_job__run__pb2.UploadToSchedulerRequest,
+        *,
+        headers: Headers | Mapping[str, str] | None = None,
+        timeout_ms: int | None = None,
+    ) -> raystack_dot_optimus_dot_core_dot_v1beta1_dot_job__run__pb2.UploadToSchedulerResponse:
+        return await self.execute_unary(
+            request=request,
+            method=MethodInfo(
+                name="UploadToScheduler",
+                service_name="raystack.optimus.core.v1beta1.JobRunService",
+                input=raystack_dot_optimus_dot_core_dot_v1beta1_dot_job__run__pb2.UploadToSchedulerRequest,
+                output=raystack_dot_optimus_dot_core_dot_v1beta1_dot_job__run__pb2.UploadToSchedulerResponse,
+                idempotency_level=IdempotencyLevel.UNKNOWN,
+            ),
+            headers=headers,
+            timeout_ms=timeout_ms,
+        )
+
+
+class JobRunServiceSync(Protocol):
+    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:
+        raise ConnectError(Code.UNIMPLEMENTED, "Not implemented")
+    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:
+        raise ConnectError(Code.UNIMPLEMENTED, "Not implemented")
+    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:
+        raise ConnectError(Code.UNIMPLEMENTED, "Not implemented")
+    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:
+        raise ConnectError(Code.UNIMPLEMENTED, "Not implemented")
+
+
+class JobRunServiceWSGIApplication(ConnectWSGIApplication):
+    def __init__(self, service: JobRunServiceSync, interceptors: Iterable[InterceptorSync]=(), read_max_bytes: int | None = None) -> None:
+        super().__init__(
+            endpoints={
+                "/raystack.optimus.core.v1beta1.JobRunService/JobRunInput": EndpointSync.unary(
+                    method=MethodInfo(
+                        name="JobRunInput",
+                        service_name="raystack.optimus.core.v1beta1.JobRunService",
+                        input=raystack_dot_optimus_dot_core_dot_v1beta1_dot_job__run__pb2.JobRunInputRequest,
+                        output=raystack_dot_optimus_dot_core_dot_v1beta1_dot_job__run__pb2.JobRunInputResponse,
+                        idempotency_level=IdempotencyLevel.UNKNOWN,
+                    ),
+                    function=service.job_run_input,
+                ),
+                "/raystack.optimus.core.v1beta1.JobRunService/JobRun": EndpointSync.unary(
+                    method=MethodInfo(
+                        name="JobRun",
+                        service_name="raystack.optimus.core.v1beta1.JobRunService",
+                        input=raystack_dot_optimus_dot_core_dot_v1beta1_dot_job__run__pb2.JobRunRequest,
+                        output=raystack_dot_optimus_dot_core_dot_v1beta1_dot_job__run__pb2.JobRunResponse,
+                        idempotency_level=IdempotencyLevel.UNKNOWN,
+                    ),
+                    function=service.job_run,
+                ),
+                "/raystack.optimus.core.v1beta1.JobRunService/RegisterJobEvent": EndpointSync.unary(
+                    method=MethodInfo(
+                        name="RegisterJobEvent",
+                        service_name="raystack.optimus.core.v1beta1.JobRunService",
+                        input=raystack_dot_optimus_dot_core_dot_v1beta1_dot_job__run__pb2.RegisterJobEventRequest,
+                        output=raystack_dot_optimus_dot_core_dot_v1beta1_dot_job__run__pb2.RegisterJobEventResponse,
+                        idempotency_level=IdempotencyLevel.UNKNOWN,
+                    ),
+                    function=service.register_job_event,
+                ),
+                "/raystack.optimus.core.v1beta1.JobRunService/UploadToScheduler": EndpointSync.unary(
+                    method=MethodInfo(
+                        name="UploadToScheduler",
+                        service_name="raystack.optimus.core.v1beta1.JobRunService",
+                        input=raystack_dot_optimus_dot_core_dot_v1beta1_dot_job__run__pb2.UploadToSchedulerRequest,
+                        output=raystack_dot_optimus_dot_core_dot_v1beta1_dot_job__run__pb2.UploadToSchedulerResponse,
+                        idempotency_level=IdempotencyLevel.UNKNOWN,
+                    ),
+                    function=service.upload_to_scheduler,
+                ),
+            },
+            interceptors=interceptors,
+            read_max_bytes=read_max_bytes,
+        )
+
+    @property
+    def path(self) -> str:
+        """Returns the URL path to mount the application to when serving multiple applications."""
+        return "/raystack.optimus.core.v1beta1.JobRunService"
+
+
+class JobRunServiceClientSync(ConnectClientSync):
+    def job_run_input(
+        self,
+        request: raystack_dot_optimus_dot_core_dot_v1beta1_dot_job__run__pb2.JobRunInputRequest,
+        *,
+        headers: Headers | Mapping[str, str] | None = None,
+        timeout_ms: int | None = None,
+    ) -> raystack_dot_optimus_dot_core_dot_v1beta1_dot_job__run__pb2.JobRunInputResponse:
+        return self.execute_unary(
+            request=request,
+            method=MethodInfo(
+                name="JobRunInput",
+                service_name="raystack.optimus.core.v1beta1.JobRunService",
+                input=raystack_dot_optimus_dot_core_dot_v1beta1_dot_job__run__pb2.JobRunInputRequest,
+                output=raystack_dot_optimus_dot_core_dot_v1beta1_dot_job__run__pb2.JobRunInputResponse,
+                idempotency_level=IdempotencyLevel.UNKNOWN,
+            ),
+            headers=headers,
+            timeout_ms=timeout_ms,
+        )
+
+    def job_run(
+        self,
+        request: raystack_dot_optimus_dot_core_dot_v1beta1_dot_job__run__pb2.JobRunRequest,
+        *,
+        headers: Headers | Mapping[str, str] | None = None,
+        timeout_ms: int | None = None,
+    ) -> raystack_dot_optimus_dot_core_dot_v1beta1_dot_job__run__pb2.JobRunResponse:
+        return self.execute_unary(
+            request=request,
+            method=MethodInfo(
+                name="JobRun",
+                service_name="raystack.optimus.core.v1beta1.JobRunService",
+                input=raystack_dot_optimus_dot_core_dot_v1beta1_dot_job__run__pb2.JobRunRequest,
+                output=raystack_dot_optimus_dot_core_dot_v1beta1_dot_job__run__pb2.JobRunResponse,
+                idempotency_level=IdempotencyLevel.UNKNOWN,
+            ),
+            headers=headers,
+            timeout_ms=timeout_ms,
+        )
+
+    def register_job_event(
+        self,
+        request: raystack_dot_optimus_dot_core_dot_v1beta1_dot_job__run__pb2.RegisterJobEventRequest,
+        *,
+        headers: Headers | Mapping[str, str] | None = None,
+        timeout_ms: int | None = None,
+    ) -> raystack_dot_optimus_dot_core_dot_v1beta1_dot_job__run__pb2.RegisterJobEventResponse:
+        return self.execute_unary(
+            request=request,
+            method=MethodInfo(
+                name="RegisterJobEvent",
+                service_name="raystack.optimus.core.v1beta1.JobRunService",
+                input=raystack_dot_optimus_dot_core_dot_v1beta1_dot_job__run__pb2.RegisterJobEventRequest,
+                output=raystack_dot_optimus_dot_core_dot_v1beta1_dot_job__run__pb2.RegisterJobEventResponse,
+                idempotency_level=IdempotencyLevel.UNKNOWN,
+            ),
+            headers=headers,
+            timeout_ms=timeout_ms,
+        )
+
+    def upload_to_scheduler(
+        self,
+        request: raystack_dot_optimus_dot_core_dot_v1beta1_dot_job__run__pb2.UploadToSchedulerRequest,
+        *,
+        headers: Headers | Mapping[str, str] | None = None,
+        timeout_ms: int | None = None,
+    ) -> raystack_dot_optimus_dot_core_dot_v1beta1_dot_job__run__pb2.UploadToSchedulerResponse:
+        return self.execute_unary(
+            request=request,
+            method=MethodInfo(
+                name="UploadToScheduler",
+                service_name="raystack.optimus.core.v1beta1.JobRunService",
+                input=raystack_dot_optimus_dot_core_dot_v1beta1_dot_job__run__pb2.UploadToSchedulerRequest,
+                output=raystack_dot_optimus_dot_core_dot_v1beta1_dot_job__run__pb2.UploadToSchedulerResponse,
+                idempotency_level=IdempotencyLevel.UNKNOWN,
+            ),
+            headers=headers,
+            timeout_ms=timeout_ms,
+        )

raystack/optimus/core/v1beta1/job_run_pb2.py

@@ -0,0 +1,88 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# NO CHECKED-IN PROTOBUF GENCODE
+# source: raystack/optimus/core/v1beta1/job_run.proto
+# Protobuf Python Version: 6.32.1
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import runtime_version as _runtime_version
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+_runtime_version.ValidateProtobufRuntimeVersion(
+    _runtime_version.Domain.PUBLIC,
+    6,
+    32,
+    1,
+    '',
+    'raystack/optimus/core/v1beta1/job_run.proto'
+)
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+from google.api import annotations_pb2 as google_dot_api_dot_annotations__pb2
+from google.protobuf import duration_pb2 as google_dot_protobuf_dot_duration__pb2
+from google.protobuf import timestamp_pb2 as google_dot_protobuf_dot_timestamp__pb2
+from protoc_gen_openapiv2.options import annotations_pb2 as protoc__gen__openapiv2_dot_options_dot_annotations__pb2
+from raystack.optimus.core.v1beta1 import job_spec_pb2 as raystack_dot_optimus_dot_core_dot_v1beta1_dot_job__spec__pb2
+
+
+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')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'raystack.optimus.core.v1beta1.job_run_pb2', _globals)
+if not _descriptor._USE_C_DESCRIPTORS:
+  _globals['DESCRIPTOR']._loaded_options = None
+  _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'
+  _globals['_JOBRUNINPUTRESPONSE_ENVSENTRY']._loaded_options = None
+  _globals['_JOBRUNINPUTRESPONSE_ENVSENTRY']._serialized_options = b'8\001'
+  _globals['_JOBRUNINPUTRESPONSE_FILESENTRY']._loaded_options = None
+  _globals['_JOBRUNINPUTRESPONSE_FILESENTRY']._serialized_options = b'8\001'
+  _globals['_JOBRUNINPUTRESPONSE_SECRETSENTRY']._loaded_options = None
+  _globals['_JOBRUNINPUTRESPONSE_SECRETSENTRY']._serialized_options = b'8\001'
+  _globals['_JOBRUNSERVICE'].methods_by_name['JobRunInput']._loaded_options = None
+  _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*'
+  _globals['_JOBRUNSERVICE'].methods_by_name['JobRun']._loaded_options = None
+  _globals['_JOBRUNSERVICE'].methods_by_name['JobRun']._serialized_options = b'\202\323\344\223\0024\0222/v1beta1/project/{project_name}/job/{job_name}/run'
+  _globals['_JOBRUNSERVICE'].methods_by_name['RegisterJobEvent']._loaded_options = None
+  _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*'
+  _globals['_JOBRUNSERVICE'].methods_by_name['UploadToScheduler']._loaded_options = None
+  _globals['_JOBRUNSERVICE'].methods_by_name['UploadToScheduler']._serialized_options = b'\202\323\344\223\002+\032&/v1beta1/project/{project_name}/upload:\001*'
+  _globals['_UPLOADTOSCHEDULERREQUEST']._serialized_start=267
+  _globals['_UPLOADTOSCHEDULERREQUEST']._serialized_end=391
+  _globals['_UPLOADTOSCHEDULERRESPONSE']._serialized_start=393
+  _globals['_UPLOADTOSCHEDULERRESPONSE']._serialized_end=481
+  _globals['_REGISTERJOBEVENTREQUEST']._serialized_start=484
+  _globals['_REGISTERJOBEVENTREQUEST']._serialized_end=673
+  _globals['_REGISTERJOBEVENTRESPONSE']._serialized_start=675
+  _globals['_REGISTERJOBEVENTRESPONSE']._serialized_end=701
+  _globals['_JOBRUNINPUTREQUEST']._serialized_start=704
+  _globals['_JOBRUNINPUTREQUEST']._serialized_end=1008
+  _globals['_JOBRUNREQUEST']._serialized_start=1011
+  _globals['_JOBRUNREQUEST']._serialized_end=1226
+  _globals['_JOBRUNRESPONSE']._serialized_start=1228
+  _globals['_JOBRUNRESPONSE']._serialized_end=1310
+  _globals['_INSTANCESPEC']._serialized_start=1313
+  _globals['_INSTANCESPEC']._serialized_end=1641
+  _globals['_INSTANCESPEC_TYPE']._serialized_start=1571
+  _globals['_INSTANCESPEC_TYPE']._serialized_end=1629
+  _globals['_INSTANCESPECDATA']._serialized_start=1644
+  _globals['_INSTANCESPECDATA']._serialized_end=1849
+  _globals['_INSTANCESPECDATA_TYPE']._serialized_start=1780
+  _globals['_INSTANCESPECDATA_TYPE']._serialized_end=1837
+  _globals['_JOBRUNINPUTRESPONSE']._serialized_start=1852
+  _globals['_JOBRUNINPUTRESPONSE']._serialized_end=2306
+  _globals['_JOBRUNINPUTRESPONSE_ENVSENTRY']._serialized_start=2133
+  _globals['_JOBRUNINPUTRESPONSE_ENVSENTRY']._serialized_end=2188
+  _globals['_JOBRUNINPUTRESPONSE_FILESENTRY']._serialized_start=2190
+  _globals['_JOBRUNINPUTRESPONSE_FILESENTRY']._serialized_end=2246
+  _globals['_JOBRUNINPUTRESPONSE_SECRETSENTRY']._serialized_start=2248
+  _globals['_JOBRUNINPUTRESPONSE_SECRETSENTRY']._serialized_end=2306
+  _globals['_TASKWINDOW']._serialized_start=2309
+  _globals['_TASKWINDOW']._serialized_end=2452
+  _globals['_JOBRUNSERVICE']._serialized_start=2455
+  _globals['_JOBRUNSERVICE']._serialized_end=3236
+# @@protoc_insertion_point(module_scope)

raystack/optimus/core/v1beta1/job_run_pb2.pyi

@@ -0,0 +1,164 @@
+import datetime
+
+from google.api import annotations_pb2 as _annotations_pb2
+from google.protobuf import duration_pb2 as _duration_pb2
+from google.protobuf import timestamp_pb2 as _timestamp_pb2
+from protoc_gen_openapiv2.options import annotations_pb2 as _annotations_pb2_1
+from raystack.optimus.core.v1beta1 import job_spec_pb2 as _job_spec_pb2
+from google.protobuf.internal import containers as _containers
+from google.protobuf.internal import enum_type_wrapper as _enum_type_wrapper
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import message as _message
+from collections.abc import Iterable as _Iterable, Mapping as _Mapping
+from typing import ClassVar as _ClassVar, Optional as _Optional, Union as _Union
+
+DESCRIPTOR: _descriptor.FileDescriptor
+
+class UploadToSchedulerRequest(_message.Message):
+    __slots__ = ("project_name", "namespace_name")
+    PROJECT_NAME_FIELD_NUMBER: _ClassVar[int]
+    NAMESPACE_NAME_FIELD_NUMBER: _ClassVar[int]
+    project_name: str
+    namespace_name: str
+    def __init__(self, project_name: _Optional[str] = ..., namespace_name: _Optional[str] = ...) -> None: ...
+
+class UploadToSchedulerResponse(_message.Message):
+    __slots__ = ("status", "error_message")
+    STATUS_FIELD_NUMBER: _ClassVar[int]
+    ERROR_MESSAGE_FIELD_NUMBER: _ClassVar[int]
+    status: bool
+    error_message: str
+    def __init__(self, status: _Optional[bool] = ..., error_message: _Optional[str] = ...) -> None: ...
+
+class RegisterJobEventRequest(_message.Message):
+    __slots__ = ("project_name", "job_name", "namespace_name", "event")
+    PROJECT_NAME_FIELD_NUMBER: _ClassVar[int]
+    JOB_NAME_FIELD_NUMBER: _ClassVar[int]
+    NAMESPACE_NAME_FIELD_NUMBER: _ClassVar[int]
+    EVENT_FIELD_NUMBER: _ClassVar[int]
+    project_name: str
+    job_name: str
+    namespace_name: str
+    event: _job_spec_pb2.JobEvent
+    def __init__(self, project_name: _Optional[str] = ..., job_name: _Optional[str] = ..., namespace_name: _Optional[str] = ..., event: _Optional[_Union[_job_spec_pb2.JobEvent, _Mapping]] = ...) -> None: ...
+
+class RegisterJobEventResponse(_message.Message):
+    __slots__ = ()
+    def __init__(self) -> None: ...
+
+class JobRunInputRequest(_message.Message):
+    __slots__ = ("project_name", "job_name", "scheduled_at", "instance_name", "instance_type", "jobrun_id")
+    PROJECT_NAME_FIELD_NUMBER: _ClassVar[int]
+    JOB_NAME_FIELD_NUMBER: _ClassVar[int]
+    SCHEDULED_AT_FIELD_NUMBER: _ClassVar[int]
+    INSTANCE_NAME_FIELD_NUMBER: _ClassVar[int]
+    INSTANCE_TYPE_FIELD_NUMBER: _ClassVar[int]
+    JOBRUN_ID_FIELD_NUMBER: _ClassVar[int]
+    project_name: str
+    job_name: str
+    scheduled_at: _timestamp_pb2.Timestamp
+    instance_name: str
+    instance_type: InstanceSpec.Type
+    jobrun_id: str
+    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: ...
+
+class JobRunRequest(_message.Message):
+    __slots__ = ("project_name", "job_name", "start_date", "end_date", "filter")
+    PROJECT_NAME_FIELD_NUMBER: _ClassVar[int]
+    JOB_NAME_FIELD_NUMBER: _ClassVar[int]
+    START_DATE_FIELD_NUMBER: _ClassVar[int]
+    END_DATE_FIELD_NUMBER: _ClassVar[int]
+    FILTER_FIELD_NUMBER: _ClassVar[int]
+    project_name: str
+    job_name: str
+    start_date: _timestamp_pb2.Timestamp
+    end_date: _timestamp_pb2.Timestamp
+    filter: _containers.RepeatedScalarFieldContainer[str]
+    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: ...
+
+class JobRunResponse(_message.Message):
+    __slots__ = ("job_runs",)
+    JOB_RUNS_FIELD_NUMBER: _ClassVar[int]
+    job_runs: _containers.RepeatedCompositeFieldContainer[_job_spec_pb2.JobRun]
+    def __init__(self, job_runs: _Optional[_Iterable[_Union[_job_spec_pb2.JobRun, _Mapping]]] = ...) -> None: ...
+
+class InstanceSpec(_message.Message):
+    __slots__ = ("state", "data", "executed_at", "name", "type")
+    class Type(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
+        __slots__ = ()
+        TYPE_UNSPECIFIED: _ClassVar[InstanceSpec.Type]
+        TYPE_TASK: _ClassVar[InstanceSpec.Type]
+        TYPE_HOOK: _ClassVar[InstanceSpec.Type]
+    TYPE_UNSPECIFIED: InstanceSpec.Type
+    TYPE_TASK: InstanceSpec.Type
+    TYPE_HOOK: InstanceSpec.Type
+    STATE_FIELD_NUMBER: _ClassVar[int]
+    DATA_FIELD_NUMBER: _ClassVar[int]
+    EXECUTED_AT_FIELD_NUMBER: _ClassVar[int]
+    NAME_FIELD_NUMBER: _ClassVar[int]
+    TYPE_FIELD_NUMBER: _ClassVar[int]
+    state: str
+    data: _containers.RepeatedCompositeFieldContainer[InstanceSpecData]
+    executed_at: _timestamp_pb2.Timestamp
+    name: str
+    type: InstanceSpec.Type
+    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: ...
+
+class InstanceSpecData(_message.Message):
+    __slots__ = ("name", "value", "type")
+    class Type(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
+        __slots__ = ()
+        TYPE_UNSPECIFIED: _ClassVar[InstanceSpecData.Type]
+        TYPE_ENV: _ClassVar[InstanceSpecData.Type]
+        TYPE_FILE: _ClassVar[InstanceSpecData.Type]
+    TYPE_UNSPECIFIED: InstanceSpecData.Type
+    TYPE_ENV: InstanceSpecData.Type
+    TYPE_FILE: InstanceSpecData.Type
+    NAME_FIELD_NUMBER: _ClassVar[int]
+    VALUE_FIELD_NUMBER: _ClassVar[int]
+    TYPE_FIELD_NUMBER: _ClassVar[int]
+    name: str
+    value: str
+    type: InstanceSpecData.Type
+    def __init__(self, name: _Optional[str] = ..., value: _Optional[str] = ..., type: _Optional[_Union[InstanceSpecData.Type, str]] = ...) -> None: ...
+
+class JobRunInputResponse(_message.Message):
+    __slots__ = ("envs", "files", "secrets")
+    class EnvsEntry(_message.Message):
+        __slots__ = ("key", "value")
+        KEY_FIELD_NUMBER: _ClassVar[int]
+        VALUE_FIELD_NUMBER: _ClassVar[int]
+        key: str
+        value: str
+        def __init__(self, key: _Optional[str] = ..., value: _Optional[str] = ...) -> None: ...
+    class FilesEntry(_message.Message):
+        __slots__ = ("key", "value")
+        KEY_FIELD_NUMBER: _ClassVar[int]
+        VALUE_FIELD_NUMBER: _ClassVar[int]
+        key: str
+        value: str
+        def __init__(self, key: _Optional[str] = ..., value: _Optional[str] = ...) -> None: ...
+    class SecretsEntry(_message.Message):
+        __slots__ = ("key", "value")
+        KEY_FIELD_NUMBER: _ClassVar[int]
+        VALUE_FIELD_NUMBER: _ClassVar[int]
+        key: str
+        value: str
+        def __init__(self, key: _Optional[str] = ..., value: _Optional[str] = ...) -> None: ...
+    ENVS_FIELD_NUMBER: _ClassVar[int]
+    FILES_FIELD_NUMBER: _ClassVar[int]
+    SECRETS_FIELD_NUMBER: _ClassVar[int]
+    envs: _containers.ScalarMap[str, str]
+    files: _containers.ScalarMap[str, str]
+    secrets: _containers.ScalarMap[str, str]
+    def __init__(self, envs: _Optional[_Mapping[str, str]] = ..., files: _Optional[_Mapping[str, str]] = ..., secrets: _Optional[_Mapping[str, str]] = ...) -> None: ...
+
+class TaskWindow(_message.Message):
+    __slots__ = ("size", "offset", "truncate_to")
+    SIZE_FIELD_NUMBER: _ClassVar[int]
+    OFFSET_FIELD_NUMBER: _ClassVar[int]
+    TRUNCATE_TO_FIELD_NUMBER: _ClassVar[int]
+    size: _duration_pb2.Duration
+    offset: _duration_pb2.Duration
+    truncate_to: str
+    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: ...