def getService(name):
try:
response = Cuebot.getStub('service').GetService(
service_pb2.ServiceGetServiceRequest(name=name),
timeout=Cuebot.Timeout)
except grpc.RpcError as e:
if e.code() == grpc.StatusCode.NOT_FOUND:
return None
raise e
return Service(response.service)
def getService(name):
"""Returns a service by name.
:type name: str
:param name: service name to find
"""
try:
response = Cuebot.getStub('service').GetService(
service_pb2.ServiceGetServiceRequest(name=name),
timeout=Cuebot.Timeout)
except grpc.RpcError as e:
# pylint: disable=no-member
if e.code() == grpc.StatusCode.NOT_FOUND:
return None
# pylint: enable=no-member
raise e
return Service(response.service)
class Service(object):
stub = Cuebot.getStub('service')
def __init__(self, service=None):
self.data = service or service_pb2.Service()
def create(self):
response = self.stub.CreateService(
service_pb2.ServiceCreateServiceRequest(data=self.data),
timeout=Cuebot.Timeout)
return Service(response.service)
def delete(self):
return self.stub.Delete(
service_pb2.ServiceDeleteRequest(service=self.data),
timeout=Cuebot.Timeout)
@classmethod
def getDefaultServices(cls):
response = cls.stub.GetDefaultServices(
service_pb2.ServiceGetDefaultServicesRequest(),
timeout=Cuebot.Timeout)
return [Service(data) for data in response.services.services]
@classmethod
def getService(cls, name):
try:
response = cls.stub.GetService(
service_pb2.ServiceGetServiceRequest(name=name),
timeout=Cuebot.Timeout)
except grpc.RpcError, e:
if e.code() == grpc.StatusCode.NOT_FOUND:
return None
raise e
return Service(response.service)
def __init__(self, show):
self.data = show
self.stub = Cuebot.getStub('show')
def __init__(self, allocation=None):
self.data = allocation
self.stub = Cuebot.getStub('allocation')
def __init__(self, comment=None):
self.data = comment
self.stub = Cuebot.getStub('comment')
def getDefaultServices():
"""Returns the default services."""
response = Cuebot.getStub('service').GetDefaultServices(
service_pb2.ServiceGetDefaultServicesRequest(),
timeout=Cuebot.Timeout)
return [Service(data) for data in response.services.services]
def __init__(self, service=None):
self.data = service or service_pb2.Service()
self.stub = Cuebot.getStub('service')
def __init__(self, depend=None):
self.data = depend
self.stub = Cuebot.getStub('depend')
def __init__(self, layer):
self.data = layer
self.stub = Cuebot.getStub('layer')
def __init__(self, proc):
self.data = proc
self.stub = Cuebot.getStub('proc')
def __init__(self, task=None):
self.data = task
self.stub = Cuebot.getStub('task')
def __init__(self, subscription=None):
self.data = subscription
self.stub = Cuebot.getStub('subscription')
def __init__(self, deed=None):
self.data = deed
self.stub = Cuebot.getStub('comment')
