use namespace for objects

Mirantis · Sep 28, 2017 · eecd336 · eecd336
1 parent 0b07655
commit eecd336
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/kqueen/models.py b/kqueen/models.py
@@ -12,6 +12,7 @@
 class EtcdOrm:
     def __init__(self, **kwargs):
         self.client = etcd.Client()
+        self.namespace = 'default'
         self.prefix = kwargs.get('prefix', '/kqueen/obj/')
 
 
@@ -35,7 +36,11 @@ def get_model_name(cls):
 
     @classmethod
     def get_db_prefix(cls):
-        return '{}{}/'.format(db.prefix, cls.get_model_name())
+        return '{prefix}{namespace}/{model}/'.format(
+            prefix=db.prefix,
+            namespace=db.namespace,
+            model=cls.get_model_name(),
+        )
 
     @classmethod
     def create(cls, **kwargs):

diff --git a/kqueen/test_models.py b/kqueen/test_models.py
@@ -18,7 +18,7 @@ def test_get_model_name(self, model_class, req):
 
     def test_get_db_prefix(self):
         model_class = Cluster
-        req = '/kqueen/obj/cluster/'
+        req = '/kqueen/obj/default/cluster/'
 
         assert model_class.get_db_prefix() == req