Move DATABASES config to utils

Author: aclark4life

config/allauth/allauth_settings.py

@@ -1,13 +1,12 @@
-import django_mongodb_backend
-import os
-
 from bson import ObjectId
 from pathlib import Path
 
 from django.contrib.auth.hashers import PBKDF2PasswordHasher
 
-DATABASE_URL = os.environ.get("MONGODB_URI", "mongodb://localhost:27017/djangotests")
+from django_mongodb_cli.utils import get_databases
+
 
+DATABASES = get_databases("allauth")
 
 SECRET_KEY = "psst"
 SITE_ID = ObjectId()
@@ -18,10 +17,6 @@
 USE_I18N = False
 USE_TZ = True
 
-DATABASES = {
-    "default": django_mongodb_backend.parse_uri(DATABASE_URL),
-}
-
 ROOT_URLCONF = "tests.regular.urls"
 LOGIN_URL = "/accounts/login/"
 

config/debug_toolbar/debug_toolbar_settings.py

@@ -1,6 +1,10 @@
 """Django settings for tests."""
 
 import os
+from django_mongodb_cli.utils import get_databases
+
+
+DATABASES = get_databases("debug_toolbar")
 
 BASE_DIR = os.path.dirname(os.path.dirname(__file__))
 
@@ -91,37 +95,7 @@
     "second": {"BACKEND": "django.core.cache.backends.locmem.LocMemCache"},
 }
 
-DATABASES = {
-    "default": {
-        "ENGINE": "django.{}db.backends.{}".format(
-            "contrib.gis." if USE_GIS else "", os.getenv("DB_BACKEND", "sqlite3")
-        ),
-        "NAME": os.getenv("DB_NAME", ":memory:"),
-        "USER": os.getenv("DB_USER"),
-        "PASSWORD": os.getenv("DB_PASSWORD"),
-        "HOST": os.getenv("DB_HOST", ""),
-        "PORT": os.getenv("DB_PORT", ""),
-        "TEST": {
-            "USER": "default_test",
-        },
-    },
-    "replica": {
-        "ENGINE": "django.{}db.backends.{}".format(
-            "contrib.gis." if USE_GIS else "", os.getenv("DB_BACKEND", "sqlite3")
-        ),
-        "NAME": os.getenv("DB_NAME", ":memory:"),
-        "USER": os.getenv("DB_USER"),
-        "PASSWORD": os.getenv("DB_PASSWORD"),
-        "HOST": os.getenv("DB_HOST", ""),
-        "PORT": os.getenv("DB_PORT", ""),
-        "TEST": {
-            "USER": "default_test",
-            "MIRROR": "default",
-        },
-    },
-}
-
-DEFAULT_AUTO_FIELD = "django.db.models.AutoField"
+DEFAULT_AUTO_FIELD = "django_mongodb_backend.fields.ObjectIdAutoField"
 
 # Debug Toolbar configuration
 

config/django/django_settings.py

@@ -1,11 +1,8 @@
-import django_mongodb_backend
-import os
+from django_mongodb_cli.utils import get_databases
 
 
-MONGODB_URI = os.environ.get("MONGODB_URI", "mongodb://localhost:27017/djangotests")
-DATABASES = {
-    "default": django_mongodb_backend.parse_uri(MONGODB_URI),
-}
+DATABASES = get_databases("django")
+
 DEFAULT_AUTO_FIELD = "django_mongodb_backend.fields.ObjectIdAutoField"
 PASSWORD_HASHERS = ("django.contrib.auth.hashers.MD5PasswordHasher",)
 SECRET_KEY = "django_tests_secret_key"

config/filter/filter_settings.py

@@ -1,12 +1,12 @@
 # ensure package/conf is importable
 from django_filters.conf import DEFAULTS
 
-import django_mongodb_backend
-import os
 
-DATABASES = {}
-DATABASE_URL = os.environ.get("MONGODB_URI", "mongodb://localhost:27017/djangotests")
-DATABASES["default"] = django_mongodb_backend.parse_uri(DATABASE_URL)
+from django_mongodb_cli.utils import get_databases
+
+
+DATABASES = get_databases("django_filter")
+
 
 INSTALLED_APPS = (
     "tests.mongo_apps.MongoContentTypesConfig",

config/rest_framework/rest_framework_settings.py

@@ -1,11 +1,10 @@
 import os
 
 import django
-import django_mongodb_backend
 from bson import ObjectId
 from django.core import management
 
-DATABASE_URL = os.environ.get("MONGODB_URI", "mongodb://localhost:27017/djangotests")
+from django_mongodb_cli.utils import get_databases
 
 
 def pytest_addoption(parser):
@@ -23,10 +22,8 @@ def pytest_configure(config):
 
     settings.configure(
         DEBUG_PROPAGATE_EXCEPTIONS=True,
-        DATABASES={
-            "default": django_mongodb_backend.parse_uri(DATABASE_URL),
-        },
-        SITE_ID=ObjectId(),
+        DATABASES=get_databases("rest_framework"),
+        SITE_ID=ObjectId("000000000000000000000001"),
         SECRET_KEY="not very secret in tests",
         USE_I18N=True,
         STATIC_URL="/static/",

config/wagtail/wagtail_settings.py

@@ -6,6 +6,7 @@
 from django.utils.translation import gettext_lazy as _
 
 from wagtail.test.numberformat import patch_number_formats
+from django_mongodb_cli.utils import get_databases
 
 WAGTAIL_CHECK_TEMPLATE_NUMBER_FORMAT = (
     os.environ.get("WAGTAIL_CHECK_TEMPLATE_NUMBER_FORMAT", "0") == "1"
@@ -24,36 +25,7 @@
 
 TIME_ZONE = "Asia/Tokyo"
 
-DATABASES = {
-    "default": {
-        "ENGINE": os.environ.get("DATABASE_ENGINE", "django.db.backends.sqlite3"),
-        "NAME": os.environ.get("DATABASE_NAME", ":memory:"),
-        "USER": os.environ.get("DATABASE_USER", ""),
-        "PASSWORD": os.environ.get("DATABASE_PASSWORD", ""),
-        "HOST": os.environ.get("DATABASE_HOST", ""),
-        "PORT": os.environ.get("DATABASE_PORT", ""),
-        "TEST": {"NAME": os.environ.get("DATABASE_NAME", "")},
-    }
-}
-
-# Set regular database name when a non-SQLite db is used
-if DATABASES["default"]["ENGINE"] != "django.db.backends.sqlite3":
-    DATABASES["default"]["NAME"] = os.environ.get("DATABASE_NAME", "wagtail")
-
-# Add extra options when mssql is used (on for example appveyor)
-if DATABASES["default"]["ENGINE"] == "sql_server.pyodbc":
-    DATABASES["default"]["OPTIONS"] = {
-        "driver": os.environ.get("DATABASE_DRIVER", "SQL Server Native Client 11.0"),
-        "MARS_Connection": "True",
-        "host_is_server": True,  # Applies to FreeTDS driver only
-    }
-
-
-# explicitly set charset / collation to utf8 on mysql
-if DATABASES["default"]["ENGINE"] == "django.db.backends.mysql":
-    DATABASES["default"]["TEST"]["CHARSET"] = "utf8"
-    DATABASES["default"]["TEST"]["COLLATION"] = "utf8_general_ci"
-
+DATABASES = get_databases("wagtail")
 
 SECRET_KEY = "not needed"
 

django_mongodb_cli/utils.py

@@ -1,4 +1,5 @@
 import click
+import django_mongodb_backend
 import git
 import os
 import shutil
@@ -108,6 +109,16 @@ def get_management_command(command=None):
     return base_command
 
 
+def get_databases(app):
+    """Get the databases configuration for the specified app."""
+
+    DATABASE_URL = os.environ.get(
+        "MONGODB_URI", f"mongodb://localhost:27017/{app}_tests"
+    )
+    DATABASES = {"default": django_mongodb_backend.parse_uri(DATABASE_URL)}
+    return DATABASES
+
+
 def get_repos(pyproject_path):
     with open(pyproject_path, "r") as f:
         pyproject_data = toml.load(f)