# 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