Merge e731f70 into a3d2f18

inveniosoftware · Sep 25, 2015 · dec5a5c · dec5a5c
2 parents a3d2f18 + e731f70
commit dec5a5c
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/invenio_celery/__init__.py b/invenio_celery/__init__.py
@@ -27,6 +27,7 @@
 
 from celery import Celery, signals
 from celery.loaders.base import BaseLoader
+from flask import has_app_context
 
 from . import registry
 from .version import __version__
@@ -131,6 +132,8 @@ def read_configuration(self):
     def close_database(self, **dummy_kwargs):
         if self.db:
             self.db.session.remove()
+            if has_app_context():
+                self.db.engine.dispose()
 
     def import_default_modules(self):
         """ Called before on_worker_init """