Revert SharingPool usage

kolibri/conftest.py

@@ -14,6 +14,9 @@ def global_fixture():
     if not os.path.exists(os.path.join(TEMP_KOLIBRI_HOME, "content")):
         os.mkdir(os.path.join(TEMP_KOLIBRI_HOME, "content"))
     yield  # wait until the test ended
+    from kolibri.core.tasks.main import connection
+
+    connection.dispose()
     if os.path.exists(TEMP_KOLIBRI_HOME):
         try:
             shutil.rmtree(TEMP_KOLIBRI_HOME)

kolibri/core/content/management/commands/generate_schema.py

@@ -24,8 +24,8 @@
 )
 from kolibri.core.content.utils.sqlalchemybridge import get_default_db_string
 from kolibri.core.content.utils.sqlalchemybridge import prepare_base
+from kolibri.core.content.utils.sqlalchemybridge import SharingPool
 from kolibri.core.content.utils.sqlalchemybridge import SQLALCHEMY_CLASSES_PATH_TEMPLATE
-from kolibri.core.utils.sqlalchemy_utils import SharingPool
 
 DATA_PATH_TEMPLATE = os.path.join(
     os.path.dirname(__file__), "../../fixtures/{name}_content_data.json"

kolibri/core/content/test/sqlalchemytesting.py

@@ -1,7 +1,7 @@
 from sqlalchemy import create_engine
 
 from kolibri.core.content.utils.sqlalchemybridge import get_default_db_string
-from kolibri.core.utils.sqlalchemy_utils import SharingPool
+from kolibri.core.content.utils.sqlalchemybridge import SharingPool
 
 
 def django_connection_engine():

kolibri/core/content/utils/sqlalchemybridge.py

@@ -44,6 +44,46 @@ class ClassNotFoundError(Exception):
     pass
 
 
+# get_conn and SharingPool code modified from:
+# http://nathansnoggin.blogspot.com/2013/11/integrating-sqlalchemy-into-django.html
+
+
+def get_conn(self):
+    """
+    custom connection factory, so we can share with django
+    """
+    from django.db import connections
+
+    conn = connections["default"]
+    return conn.connection
+
+
+class SharingPool(NullPool):
+    """
+    custom connection pool that doesn't close connections, and uses our
+    custom connection factory
+    """
+
+    def __init__(self, get_connection, **kwargs):
+        kwargs["reset_on_return"] = False
+        super(SharingPool, self).__init__(get_conn, **kwargs)
+
+    def status(self):
+        return "Sharing Pool"
+
+    def _do_return_conn(self, conn):
+        pass
+
+    def _do_get(self):
+        return self._create_connection()
+
+    def _close_connection(self, connection):
+        pass
+
+    def dispose(self):
+        pass
+
+
 def sqlite_connection_string(db_path):
     # Call normpath to ensure that Windows paths are properly formatted
     return "sqlite:///{db_path}".format(db_path=os.path.normpath(db_path))

kolibri/core/tasks/main.py

@@ -13,7 +13,6 @@
 from kolibri.core.tasks.queue import Queue
 from kolibri.core.tasks.scheduler import Scheduler
 from kolibri.core.tasks.worker import Worker
-from kolibri.core.utils.sqlalchemy_utils import SharingPool
 from kolibri.utils import conf
 
 
@@ -28,8 +27,6 @@ def __create_engine():
                 path=os.path.join(conf.KOLIBRI_HOME, "job_storage.sqlite3")
             ),
             connect_args={"check_same_thread": False},
-            # Use NullPool for SQLite as we use a completely separate database
-            # file, so no need to share anything with Django.
             poolclass=NullPool,
         )
 
@@ -49,10 +46,6 @@ def __create_engine():
             ),
             pool_pre_ping=True,
             client_encoding="utf8",
-            # Use our SharingPool for Postgres, so as to ensure that
-            # we share underlying database connections with Django
-            # this results in cleaner shutdown and clean up of connections
-            poolclass=SharingPool,
         )