zydronium/manage.get.gov
1
0
Fork 0
mirror of https://github.com/cisagov/manage.get.gov.git synced 2025-08-21 16:51:08 +02:00
Code Issues Projects Releases Packages Wiki Activity

wip

Browse source
This commit is contained in:
David Kennedy 2025-03-24 13:16:20 -04:00
parent b4b6565f81
commit d31af94a15
No known key found for this signature in database
GPG key ID: 6528A5386E66B96B
2 changed files with 13 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

11
src/epplibwrapper/client.py
View file

@ -100,7 +100,7 @@ class EPPLibWrapper:
response = self._client.send(self._login) # type: ignore
if response.code >= 2000: # type: ignore
self._client.close() # type: ignore
raise LoginError(response.msg) # type: ignore
raise LoginError(response.msg, response=response) # type: ignore
def _disconnect(self) -> None:
"""Close the connection. Sends a logout command and closes the connection."""
@ -144,14 +144,14 @@ class EPPLibWrapper:
text = "failed to execute due to a registry login error."
message = f"{cmd_type} {text}"
logger.error(f"{message} Error: {err}")
raise RegistryError(message) from err
raise LoginError(message, response=err.response) from err
except Exception as err:
message = f"{cmd_type} failed to execute due to an unknown error."
logger.error(f"{message} Error: {err}")
raise RegistryError(message) from err
else:
if response.code >= 2000:
raise RegistryError(response.msg, code=response.code)
raise RegistryError(response.msg, code=response.code, response=response)
else:
return response
@ -182,7 +182,10 @@ class EPPLibWrapper:
or err.should_retry()
):
message = f"{cmd_type} failed and will be retried"
logger.info(f"{message} Error: {err}")
info_message = f"{message} Error: {err}"
if err.response:
info_message = f"{info_message}. cltrid is {err.response.cl_tr_id} svtrid is {err.response.sv_tr_id}"
logger.info(f"{info_message}")
return self._retry(command)
else:
raise err