Hopiu/dj-database-url
1
0
Fork 0
mirror of https://github.com/jazzband/dj-database-url.git synced 2026-03-16 22:20:24 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

Add disable_server_side_cursors parameter (#231)

Browse source
This commit is contained in:
Fabian Binz 2023-11-08 23:21:29 +01:00 committed by GitHub
parent ee2cb860f5
commit 44bfba5fa2
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 19 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
dj_database_url/__init__.py
View file

@ -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,
}
)

9
tests/test_dj_database_url.py
View file

@ -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()