20 lines
857 B
Python
20 lines
857 B
Python
# implementation Conainer methods
|
|
import grpc
|
|
import protos.models.container.service_pb2_grpc as service_pb2_grpc_container
|
|
import protos.models.container.service_pb2 as service_pb2_container
|
|
|
|
|
|
|
|
class ContainerClient:
|
|
def __init__(self, channel):
|
|
self.container_stub = service_pb2_grpc_container.ContainerServiceStub(channel)
|
|
|
|
def create_container(self, container_name) -> bytes:
|
|
request = service_pb2_container.PutRequest(name=container_name)
|
|
response: service_pb2_container.PutResponse = self.container_stub.Put(request)
|
|
return response.body.container_id
|
|
|
|
def get_container(self, container_id) -> service_pb2_container.GetResponse:
|
|
request = service_pb2_container.GetRequest(id=container_id)
|
|
response: service_pb2_container.GetResponse = self.container_stub.Get(request)
|
|
return response
|