yolo

NeuroVault · Sep 3, 2015 · ad1d331 · ad1d331
1 parent 6443441
commit ad1d331
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/tasks.py b/tasks.py
@@ -20,14 +20,16 @@
 
 @task
 def reindex(src, dest):
-    from scrapi.processing.elasticsearch import es
+    from scrapi.processing.elasticsearch import DatabaseManager
+    es = DatabaseManager().es
     helpers.reindex(es, src, dest)
     es.indices.delete(src)
 
 
 @task
 def alias(alias, index):
-    from scrapi.processing.elasticsearch import es
+    from scrapi.processing.elasticsearch import DatabaseManager
+    es = DatabaseManager().es
     es.indices.delete_alias(index=alias, name='_all', ignore=404)
     es.indices.put_alias(alias, index)
 
@@ -205,7 +207,8 @@ def lint(name):
 
 @task
 def provider_map(delete=False):
-    from scrapi.processing.elasticsearch import es
+    from scrapi.processing.elasticsearch import DatabaseManager
+    es = DatabaseManager().es
     if delete:
         es.indices.delete(index='share_providers', ignore=[404])