From 90ba4b61e6f7f35d0814a97dfd19673e38383e66 Mon Sep 17 00:00:00 2001 From: Robin Krahl Date: Wed, 8 Nov 2023 15:43:03 +0100 Subject: [PATCH] Use generated client for set_key_certificate Previously, we manually implemented the PUT /keys/{KeyID}/cert API call because it used content types that were not supported by the generated API client. Now it only accepts application/octet-stream, so we can use the generated client instead of manually performing the request. --- nethsm/__init__.py | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) diff --git a/nethsm/__init__.py b/nethsm/__init__.py index f30d477..55a358f 100644 --- a/nethsm/__init__.py +++ b/nethsm/__init__.py @@ -1031,12 +1031,13 @@ def set_certificate(self, cert: BufferedReader) -> None: def set_key_certificate(self, key_id: str, cert: BufferedReader) -> None: try: - self.request( - "PUT", - f"keys/{key_id}/cert", - data=cert, - mime_type="application/octet-stream", + from .client.paths.keys_key_id_cert.put.path_parameters import ( + PathParametersDict, ) + + path_params = PathParametersDict(KeyID=key_id) + + self.get_api().keys_key_id_cert_put(body=cert, path_params=path_params) except Exception as e: _handle_exception( e,