From 44bfba5fa210df0b0ed71c9480ddbdd06a2e9d63 Mon Sep 17 00:00:00 2001
From: Fabian Binz <fabian.binz@gmail.com>
Date: Wed, 8 Nov 2023 23:21:29 +0100
Subject: [PATCH] Add disable_server_side_cursors parameter (#231)

---
 dj_database_url/__init__.py   | 11 ++++++++++-
 tests/test_dj_database_url.py |  9 +++++++++
 2 files changed, 19 insertions(+), 1 deletion(-)

diff --git a/dj_database_url/__init__.py b/dj_database_url/__init__.py
index 7fbd116..9cf9787 100644
--- a/dj_database_url/__init__.py
+++ b/dj_database_url/__init__.py
@@ -58,6 +58,7 @@ def config(
     engine: Optional[str] = None,
     conn_max_age: Optional[int] = 0,
     conn_health_checks: bool = False,
+    disable_server_side_cursors: bool = False,
     ssl_require: bool = False,
     test_options: Optional[Dict] = None,
 ) -> DBConfig:
@@ -71,7 +72,13 @@ def config(
 
     if s:
         return parse(
-            s, engine, conn_max_age, conn_health_checks, ssl_require, test_options
+            s,
+            engine,
+            conn_max_age,
+            conn_health_checks,
+            disable_server_side_cursors,
+            ssl_require,
+            test_options,
         )
 
     return {}
@@ -82,6 +89,7 @@ def parse(
     engine: Optional[str] = None,
     conn_max_age: Optional[int] = 0,
     conn_health_checks: bool = False,
+    disable_server_side_cursors: bool = False,
     ssl_require: bool = False,
     test_options: Optional[dict] = None,
 ) -> DBConfig:
@@ -146,6 +154,7 @@ def parse(
             "PORT": port or "",
             "CONN_MAX_AGE": conn_max_age,
             "CONN_HEALTH_CHECKS": conn_health_checks,
+            "DISABLE_SERVER_SIDE_CURSORS": disable_server_side_cursors,
             "ENGINE": engine,
         }
     )
diff --git a/tests/test_dj_database_url.py b/tests/test_dj_database_url.py
index 5728a4b..f8a38d0 100644
--- a/tests/test_dj_database_url.py
+++ b/tests/test_dj_database_url.py
@@ -592,6 +592,15 @@ class DatabaseTestSuite(unittest.TestCase):
         )
         assert url["OPTIONS"] == {'connect_timout': 3}
 
+    @mock.patch.dict(
+        os.environ,
+        {"DATABASE_URL": "postgres://user:password@instance.amazonaws.com:5431/d8r8?"},
+    )
+    def test_server_side_cursors__config(self):
+        url = dj_database_url.config(disable_server_side_cursors=True)
+
+        assert url["DISABLE_SERVER_SIDE_CURSORS"] is True
+
 
 if __name__ == "__main__":
     unittest.main()