From 41a399c0fe4a9f705a0f5616fa1055116a005b8b Mon Sep 17 00:00:00 2001 From: Joshua Harlow Date: Tue, 20 Oct 2015 17:46:42 -0700 Subject: [PATCH] Use conductor entity class constant instead of raw string Now that the conductor base class provides a constant that all entity objects produced from it should provide we can now use that to ensure we are only registering a limited set of entity kinds. Change-Id: I0e71a8da64a228fdc68c9de941aaccf54d493d96 --- taskflow/jobs/backends/impl_zookeeper.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/taskflow/jobs/backends/impl_zookeeper.py b/taskflow/jobs/backends/impl_zookeeper.py index dc38a0e4..a6237889 100644 --- a/taskflow/jobs/backends/impl_zookeeper.py +++ b/taskflow/jobs/backends/impl_zookeeper.py @@ -30,6 +30,7 @@ from oslo_utils import timeutils from oslo_utils import uuidutils import six +from taskflow.conductors import base as c_base from taskflow import exceptions as excp from taskflow.jobs import base from taskflow import logging @@ -566,7 +567,7 @@ class ZookeeperJobBoard(base.NotifyingJobBoard): def register_entity(self, entity): entity_type = entity.kind - if entity_type == 'conductor': + if entity_type == c_base.Conductor.ENTITY_KIND: entity_path = k_paths.join(self.entity_path, entity_type) self._client.ensure_path(entity_path)