Skip to content

Commit 82f8c34

Browse files
committed
Use helper for making requests
1 parent dc1f6d1 commit 82f8c34

File tree

1 file changed

+35
-48
lines changed

1 file changed

+35
-48
lines changed

src/vws/vws.py

Lines changed: 35 additions & 48 deletions
Original file line numberDiff line numberDiff line change
@@ -130,6 +130,32 @@ def __init__(
130130
self._server_secret_key = server_secret_key.encode()
131131
self._base_vws_url = base_vws_url
132132

133+
def _make_request(
134+
self,
135+
method: str,
136+
content: bytes,
137+
request_path: str,
138+
expected_result_code: str,
139+
) -> Response:
140+
"""
141+
XXX
142+
"""
143+
response = _target_api_request(
144+
server_access_key=self._server_access_key,
145+
server_secret_key=self._server_secret_key,
146+
method=method,
147+
content=content,
148+
request_path=request_path,
149+
base_vws_url=self._base_vws_url,
150+
)
151+
152+
_raise_for_result_code(
153+
response=response,
154+
expected_result_code=expected_result_code,
155+
)
156+
157+
return response
158+
133159
def add_target(
134160
self,
135161
name: str,
@@ -173,17 +199,10 @@ def add_target(
173199

174200
content = bytes(json.dumps(data), encoding='utf-8')
175201

176-
response = _target_api_request(
177-
server_access_key=self._server_access_key,
178-
server_secret_key=self._server_secret_key,
202+
response = self._make_request(
179203
method='POST',
180204
content=content,
181205
request_path='/targets',
182-
base_vws_url=self._base_vws_url,
183-
)
184-
185-
_raise_for_result_code(
186-
response=response,
187206
expected_result_code='TargetCreated',
188207
)
189208

@@ -202,19 +221,13 @@ def get_target_record(self, target_id: str) -> Dict[str, Union[str, int]]:
202221
Returns:
203222
Response details of a target from Vuforia.
204223
"""
205-
response = _target_api_request(
206-
server_access_key=self._server_access_key,
207-
server_secret_key=self._server_secret_key,
224+
response = self._make_request(
208225
method='GET',
209226
content=b'',
210227
request_path=f'/targets/{target_id}',
211-
base_vws_url=self._base_vws_url,
212-
)
213-
214-
_raise_for_result_code(
215-
response=response,
216228
expected_result_code='Success',
217229
)
230+
218231
return dict(response.json()['target_record'])
219232

220233
@timeout_decorator.timeout(seconds=60 * 5)
@@ -250,19 +263,13 @@ def list_targets(self) -> List[str]:
250263
Returns:
251264
The IDs of all targets in the database.
252265
"""
253-
response = _target_api_request(
254-
server_access_key=self._server_access_key,
255-
server_secret_key=self._server_secret_key,
266+
response = self._make_request(
256267
method='GET',
257268
content=b'',
258269
request_path='/targets',
259-
base_vws_url=self._base_vws_url,
260-
)
261-
262-
_raise_for_result_code(
263-
response=response,
264270
expected_result_code='Success',
265271
)
272+
266273
return list(response.json()['results'])
267274

268275
def get_target_summary_report(
@@ -281,19 +288,13 @@ def get_target_summary_report(
281288
Returns:
282289
Details of the target.
283290
"""
284-
response = _target_api_request(
285-
server_access_key=self._server_access_key,
286-
server_secret_key=self._server_secret_key,
291+
response = self._make_request(
287292
method='GET',
288293
content=b'',
289294
request_path=f'/summary/{target_id}',
290-
base_vws_url=self._base_vws_url,
291-
)
292-
293-
_raise_for_result_code(
294-
response=response,
295295
expected_result_code='Success',
296296
)
297+
297298
return dict(response.json())
298299

299300
def get_database_summary_report(self) -> Dict[str, Union[str, int]]:
@@ -306,17 +307,10 @@ def get_database_summary_report(self) -> Dict[str, Union[str, int]]:
306307
Returns:
307308
Details of the database.
308309
"""
309-
response = _target_api_request(
310-
server_access_key=self._server_access_key,
311-
server_secret_key=self._server_secret_key,
310+
response = self._make_request(
312311
method='GET',
313312
content=b'',
314313
request_path='/summary',
315-
base_vws_url=self._base_vws_url,
316-
)
317-
318-
_raise_for_result_code(
319-
response=response,
320314
expected_result_code='Success',
321315
)
322316

@@ -332,16 +326,9 @@ def delete_target(self, target_id: str) -> None:
332326
Args:
333327
target_id: The ID of the target to delete.
334328
"""
335-
response = _target_api_request(
336-
server_access_key=self._server_access_key,
337-
server_secret_key=self._server_secret_key,
329+
response = self._make_request(
338330
method='DELETE',
339331
content=b'',
340332
request_path=f'/targets/{target_id}',
341-
base_vws_url=self._base_vws_url,
342-
)
343-
344-
_raise_for_result_code(
345-
response=response,
346333
expected_result_code='Success',
347334
)

0 commit comments

Comments
 (0)