[#3] Added generate proto script

Signed-off-by: Ilyas Niyazov <i.niyazov@yadro.com>
This commit is contained in:
Ilyas Niyazov 2025-02-26 16:24:04 +03:00
parent 19282f13cc
commit 9a1b5d778b
5 changed files with 92 additions and 0 deletions

View file

@ -0,0 +1,20 @@
# 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