Skip to content

Commit 72eeb7e

Browse files
committed
♻️ refactor(client): split off prep request
1 parent 7ea832f commit 72eeb7e

File tree

1 file changed

+8
-4
lines changed

1 file changed

+8
-4
lines changed

dsws_client/client.py

Lines changed: 8 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -330,10 +330,7 @@ def _execute_request(
330330
) -> ResponseCls:
331331
"""Execute a request."""
332332
logger.debug("executing request")
333-
if self._app_id is not None:
334-
request.properties.append(DSStringKVPair("__AppId", self._app_id))
335-
if self._data_source is not None:
336-
request.properties.append(DSStringKVPair("Source", self._data_source))
333+
self._prep_request(request)
337334
request_data = msgspec.json.encode(request)
338335
if self._debug:
339336
sys.stdout.write(f"sending request: {request_data!s}")
@@ -352,3 +349,10 @@ def _execute_request(
352349
if self._debug:
353350
sys.stdout.write(f"received response: {response_decoded!s}")
354351
return response_decoded
352+
353+
def _prep_request(self, request: DSRequest) -> None:
354+
"""Prepare a request."""
355+
if self._app_id is not None:
356+
request.properties.append(DSStringKVPair("__AppId", self._app_id))
357+
if self._data_source is not None:
358+
request.properties.append(DSStringKVPair("Source", self._data_source))

0 commit comments

Comments
 (0)