Use database_from_env

uktrade · Jun 14, 2024 · 6a6b7d6 · 6a6b7d6
1 parent 4051ac2
commit 6a6b7d6
Showing 1 changed file with 3 additions and 5 deletions.
diff --git a/demodjango/settings.py b/demodjango/settings.py
@@ -12,17 +12,15 @@
 
 import os
 import sys
-import tempfile
 from pathlib import Path
+
 import dj_database_url
 import environ
 import sentry_sdk
+from dbt_copilot_python.database import database_url_from_env, database_from_env
 from dbt_copilot_python.network import setup_allowed_hosts
-from dbt_copilot_python.utility import is_copilot
 from django_log_formatter_asim import ASIMFormatter
 from dotenv import load_dotenv
-
-from dbt_copilot_python.database import database_url_from_env
 from sentry_sdk.integrations.django import DjangoIntegration
 
 load_dotenv()
@@ -156,7 +154,7 @@
 if RDS_POSTGRES_CREDENTIALS:
     DATABASES = {
     "default": dj_database_url.config(
-        default=database_url_from_env("RDS_POSTGRES_CREDENTIALS")
+        default=database_from_env("RDS_POSTGRES_CREDENTIALS")
     )
 }