diff --git a/django_mongodb_backend/base.py b/django_mongodb_backend/base.py index a052a927b..d6b1d67cb 100644 --- a/django_mongodb_backend/base.py +++ b/django_mongodb_backend/base.py @@ -3,6 +3,7 @@ from django.core.exceptions import ImproperlyConfigured from django.db.backends.base.base import BaseDatabaseWrapper +from django.utils.asyncio import async_unsafe from pymongo.collection import Collection from pymongo.driver_info import DriverInfo from pymongo.mongo_client import MongoClient @@ -172,6 +173,7 @@ def get_connection_params(self): **settings_dict["OPTIONS"], } + @async_unsafe def get_new_connection(self, conn_params): return MongoClient(**conn_params, driver=self._driver_info()) @@ -189,11 +191,13 @@ def _rollback(self): def set_autocommit(self, autocommit, force_begin_transaction_with_broken_autocommit=False): pass + @async_unsafe def close(self): super().close() with contextlib.suppress(AttributeError): del self.database + @async_unsafe def cursor(self): return Cursor()