[wip] parse db name out of connection string

Author: timgraham

django_mongodb_backend/base.py

@@ -11,6 +11,7 @@
 from pymongo.collection import Collection
 from pymongo.driver_info import DriverInfo
 from pymongo.mongo_client import MongoClient
+from pymongo.uri_parser import parse_uri
 
 from . import __version__ as django_mongodb_backend_version
 from . import dbapi as Database
@@ -157,6 +158,14 @@ def __init__(self, settings_dict, alias=DEFAULT_DB_ALIAS):
         self.in_atomic_block_mongo = False
         # Current number of nested 'atomic' calls.
         self.nested_atomics = 0
+        if self.settings_dict["NAME"] == "":
+            host = self.settings_dict["HOST"]
+            if host.startswith(("mongodb://", "mongodb+srv://")):
+                uri = parse_uri(host)
+                if database := uri.get("database"):
+                    self.settings_dict["NAME"] = database
+            else:
+                raise ImproperlyConfigured('settings.DATABASES is missing the "NAME" value.')
 
     def get_collection(self, name, **kwargs):
         collection = Collection(self.database, name, **kwargs)
@@ -183,8 +192,6 @@ def init_connection_state(self):
 
     def get_connection_params(self):
         settings_dict = self.settings_dict
-        if not settings_dict["NAME"]:
-            raise ImproperlyConfigured('settings.DATABASES is missing the "NAME" value.')
         params = {
             "host": settings_dict["HOST"] or None,
             **settings_dict["OPTIONS"],

tests/backend_/test_base.py

@@ -12,7 +12,7 @@ def test_database_name_empty(self):
         settings["NAME"] = ""
         msg = 'settings.DATABASES is missing the "NAME" value.'
         with self.assertRaisesMessage(ImproperlyConfigured, msg):
-            DatabaseWrapper(settings).get_connection_params()
+            DatabaseWrapper(settings)
 
     def test_host(self):
         settings = connection.settings_dict.copy()