Skip to content

Commit 3e5eea1

Browse files
fix(client): preserve hardcoded query params when merging with user params
1 parent 7f4c7c5 commit 3e5eea1

File tree

2 files changed

+52
-0
lines changed

2 files changed

+52
-0
lines changed

src/cas_parser/_base_client.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -558,6 +558,10 @@ def _build_request(
558558
files = cast(HttpxRequestFiles, ForceMultipartDict())
559559

560560
prepared_url = self._prepare_url(options.url)
561+
# preserve hard-coded query params from the url
562+
if params and prepared_url.query:
563+
params = {**dict(prepared_url.params.items()), **params}
564+
prepared_url = prepared_url.copy_with(raw_path=prepared_url.raw_path.split(b"?", 1)[0])
561565
if "_" in prepared_url.host:
562566
# work around https://github.com/encode/httpx/discussions/2880
563567
kwargs["extensions"] = {"sni_hostname": prepared_url.host.replace("_", "-")}

tests/test_client.py

Lines changed: 48 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -429,6 +429,30 @@ def test_default_query_option(self) -> None:
429429

430430
client.close()
431431

432+
def test_hardcoded_query_params_in_url(self, client: CasParser) -> None:
433+
request = client._build_request(FinalRequestOptions(method="get", url="/foo?beta=true"))
434+
url = httpx.URL(request.url)
435+
assert dict(url.params) == {"beta": "true"}
436+
437+
request = client._build_request(
438+
FinalRequestOptions(
439+
method="get",
440+
url="/foo?beta=true",
441+
params={"limit": "10", "page": "abc"},
442+
)
443+
)
444+
url = httpx.URL(request.url)
445+
assert dict(url.params) == {"beta": "true", "limit": "10", "page": "abc"}
446+
447+
request = client._build_request(
448+
FinalRequestOptions(
449+
method="get",
450+
url="/files/a%2Fb?beta=true",
451+
params={"limit": "10"},
452+
)
453+
)
454+
assert request.url.raw_path == b"/files/a%2Fb?beta=true&limit=10"
455+
432456
def test_request_extra_json(self, client: CasParser) -> None:
433457
request = client._build_request(
434458
FinalRequestOptions(
@@ -1320,6 +1344,30 @@ async def test_default_query_option(self) -> None:
13201344

13211345
await client.close()
13221346

1347+
async def test_hardcoded_query_params_in_url(self, async_client: AsyncCasParser) -> None:
1348+
request = async_client._build_request(FinalRequestOptions(method="get", url="/foo?beta=true"))
1349+
url = httpx.URL(request.url)
1350+
assert dict(url.params) == {"beta": "true"}
1351+
1352+
request = async_client._build_request(
1353+
FinalRequestOptions(
1354+
method="get",
1355+
url="/foo?beta=true",
1356+
params={"limit": "10", "page": "abc"},
1357+
)
1358+
)
1359+
url = httpx.URL(request.url)
1360+
assert dict(url.params) == {"beta": "true", "limit": "10", "page": "abc"}
1361+
1362+
request = async_client._build_request(
1363+
FinalRequestOptions(
1364+
method="get",
1365+
url="/files/a%2Fb?beta=true",
1366+
params={"limit": "10"},
1367+
)
1368+
)
1369+
assert request.url.raw_path == b"/files/a%2Fb?beta=true&limit=10"
1370+
13231371
def test_request_extra_json(self, client: CasParser) -> None:
13241372
request = client._build_request(
13251373
FinalRequestOptions(

0 commit comments

Comments
 (0)