import grpc from common.database.api.Database import Database from common.exceptions.ServiceException import ServiceException def check_service_exists(database : Database, context_id : str, service_id : str): db_context = database.context(context_id).create() if db_context.services.contains(service_id): return msg = 'Context({})/Service({}) does not exist in the database.' msg = msg.format(context_id, service_id) raise ServiceException(grpc.StatusCode.NOT_FOUND, msg) def check_service_not_exists(database : Database, context_id : str, service_id : str): db_context = database.context(context_id).create() if not db_context.services.contains(service_id): return msg = 'Context({})/Service({}) already exists in the database.' msg = msg.format(context_id, service_id) raise ServiceException(grpc.StatusCode.ALREADY_EXISTS, msg)