removed warning in ensure index

hivesolutions · Nov 26, 2015 · 8ecbce1 · 8ecbce1
1 parent e5b0c18
commit 8ecbce1
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 4 deletions.
diff --git a/src/appier/data.py b/src/appier/data.py
@@ -79,4 +79,4 @@ def count(self, *args, **kwargs):
         return self._base.count(*args, **kwargs)
 
     def ensure_index(self, *args, **kwargs):
-        return self._base.ensure_index(*args, **kwargs)
+        return mongo._store_ensure_index(self._base, *args, **kwargs)
diff --git a/src/appier/mongo.py b/src/appier/mongo.py
@@ -123,6 +123,10 @@ def is_mongo(obj):
 def is_new():
     return int(pymongo.version[0]) >= 3 if pymongo else False
 
+def _store_find_and_modify(store, *args, **kwargs):
+    if is_new(): store.find_one_and_update(*args, **kwargs)
+    else: store.find_and_modify(*args, **kwargs)
+
 def _store_insert(store, *args, **kwargs):
     if is_new(): store.insert_one(*args, **kwargs)
     else: store.insert(*args, **kwargs)
@@ -131,6 +135,6 @@ def _store_update(store, *args, **kwargs):
     if is_new(): store.update_one(*args, **kwargs)
     else: store.update(*args, **kwargs)
 
-def _store_find_and_modify(store, *args, **kwargs):
-    if is_new(): store.find_one_and_update(*args, **kwargs)
-    else: store.find_and_modify(*args, **kwargs)
+def _store_ensure_index(store, *args, **kwargs):
+    if is_new(): store.create_index(*args, **kwargs)
+    else: store.ensure_index(*args, **kwargs)