mergetbapi 1.3.25__tar.gz → 1.3.26__tar.gz

{mergetbapi-1.3.25 → mergetbapi-1.3.26}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mergetbapi
-Version: 1.3.25
+Version: 1.3.26
 Summary: Client interface to the MergeTB API
 Home-page: https://gitlab.com/mergetb/api
 Author: Brian Kocoloski

{mergetbapi-1.3.25 → mergetbapi-1.3.26}/mergetbapi/portal/v1/__init__.py

@@ -164,6 +164,11 @@ class RealizeResponseCode(betterproto.Enum):
     InternalError = 3
 
 
+class SaveDiskMaterializationMode(betterproto.Enum):
+    DiskOnly = 0
+    AllState = 1
+
+
 class RebootMaterializationMode(betterproto.Enum):
     Reboot = 0
     Cycle = 1
@@ -2467,6 +2472,26 @@ class Actuator(betterproto.Message):
     model: "__xir__.Actuator" = betterproto.message_field(2)
 
 
+@dataclass(eq=False, repr=False)
+class SaveDiskMaterializationRequest(betterproto.Message):
+    mode: "SaveDiskMaterializationMode" = betterproto.enum_field(1)
+    project: str = betterproto.string_field(2)
+    experiment: str = betterproto.string_field(3)
+    realization: str = betterproto.string_field(4)
+    hostnames: List[str] = betterproto.string_field(5)
+    """
+    hostnames are the facility's experimental node's name as defined
+     in the user-defined experiment model.
+    """
+
+    ver: int = betterproto.int64_field(6)
+
+
+@dataclass(eq=False, repr=False)
+class SaveDiskMaterializationResponse(betterproto.Message):
+    image_names: List[str] = betterproto.string_field(1)
+
+
 @dataclass(eq=False, repr=False)
 class RebootMaterializationRequest(betterproto.Message):
     mode: "RebootMaterializationMode" = betterproto.enum_field(1)
@@ -4104,6 +4129,23 @@ class MaterializeStub(betterproto.ServiceStub):
             metadata=metadata,
         )
 
+    async def save_disk_materialization(
+        self,
+        save_disk_materialization_request: "SaveDiskMaterializationRequest",
+        *,
+        timeout: Optional[float] = None,
+        deadline: Optional["Deadline"] = None,
+        metadata: Optional["MetadataLike"] = None
+    ) -> "SaveDiskMaterializationResponse":
+        return await self._unary_unary(
+            "/portal.v1.Materialize/SaveDiskMaterialization",
+            save_disk_materialization_request,
+            SaveDiskMaterializationResponse,
+            timeout=timeout,
+            deadline=deadline,
+            metadata=metadata,
+        )
+
     async def new_ingress(
         self,
         new_ingress_request: "NewIngressRequest",
@@ -6321,6 +6363,11 @@ class MaterializeBase(ServiceBase):
     ) -> "RebootMaterializationResponse":
         raise grpclib.GRPCError(grpclib.const.Status.UNIMPLEMENTED)
 
+    async def save_disk_materialization(
+        self, save_disk_materialization_request: "SaveDiskMaterializationRequest"
+    ) -> "SaveDiskMaterializationResponse":
+        raise grpclib.GRPCError(grpclib.const.Status.UNIMPLEMENTED)
+
     async def new_ingress(
         self, new_ingress_request: "NewIngressRequest"
     ) -> "NewIngressResponse":
@@ -6394,6 +6441,14 @@ class MaterializeBase(ServiceBase):
         response = await self.reboot_materialization(request)
         await stream.send_message(response)
 
+    async def __rpc_save_disk_materialization(
+        self,
+        stream: "grpclib.server.Stream[SaveDiskMaterializationRequest, SaveDiskMaterializationResponse]",
+    ) -> None:
+        request = await stream.recv_message()
+        response = await self.save_disk_materialization(request)
+        await stream.send_message(response)
+
     async def __rpc_new_ingress(
         self, stream: "grpclib.server.Stream[NewIngressRequest, NewIngressResponse]"
     ) -> None:
@@ -6459,6 +6514,12 @@ class MaterializeBase(ServiceBase):
                 RebootMaterializationRequest,
                 RebootMaterializationResponse,
             ),
+            "/portal.v1.Materialize/SaveDiskMaterialization": grpclib.const.Handler(
+                self.__rpc_save_disk_materialization,
+                grpclib.const.Cardinality.UNARY_UNARY,
+                SaveDiskMaterializationRequest,
+                SaveDiskMaterializationResponse,
+            ),
             "/portal.v1.Materialize/NewIngress": grpclib.const.Handler(
                 self.__rpc_new_ingress,
                 grpclib.const.Cardinality.UNARY_UNARY,

{mergetbapi-1.3.25 → mergetbapi-1.3.26}/mergetbapi.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mergetbapi
-Version: 1.3.25
+Version: 1.3.26
 Summary: Client interface to the MergeTB API
 Home-page: https://gitlab.com/mergetb/api
 Author: Brian Kocoloski