From 3ef273f086c7d01c82dbbff43673a723844e7116 Mon Sep 17 00:00:00 2001 From: Tom Parker-Shemilt Date: Fri, 30 Dec 2022 09:48:07 +0000 Subject: [PATCH] Avoid duplicating the uses_netloc and schemes lines (#201) --- dj_database_url.py | 24 ++++-------------------- 1 file changed, 4 insertions(+), 20 deletions(-) diff --git a/dj_database_url.py b/dj_database_url.py index ce3a7df..041ad11 100644 --- a/dj_database_url.py +++ b/dj_database_url.py @@ -4,26 +4,6 @@ from typing import Any, Dict, Optional, Union from typing_extensions import TypedDict -# Register database schemes in URLs. -urlparse.uses_netloc.append("postgres") -urlparse.uses_netloc.append("postgresql") -urlparse.uses_netloc.append("pgsql") -urlparse.uses_netloc.append("postgis") -urlparse.uses_netloc.append("mysql") -urlparse.uses_netloc.append("mysql2") -urlparse.uses_netloc.append("mysqlgis") -urlparse.uses_netloc.append("mysql-connector") -urlparse.uses_netloc.append("mssql") -urlparse.uses_netloc.append("mssqlms") -urlparse.uses_netloc.append("spatialite") -urlparse.uses_netloc.append("sqlite") -urlparse.uses_netloc.append("oracle") -urlparse.uses_netloc.append("oraclegis") -urlparse.uses_netloc.append("redshift") -urlparse.uses_netloc.append("cockroach") -urlparse.uses_netloc.append("timescale") -urlparse.uses_netloc.append("timescalegis") - DEFAULT_ENV = "DATABASE_URL" SCHEMES = { @@ -47,6 +27,10 @@ SCHEMES = { "timescalegis": "timescale.db.backends.postgis", } +# Register database schemes in URLs. +for key in SCHEMES.keys(): + urlparse.uses_netloc.append(key) + # From https://docs.djangoproject.com/en/4.0/ref/settings/#databases class DBConfig(TypedDict, total=False):