mirror of
https://github.com/cisagov/manage.get.gov.git
synced 2025-07-24 11:38:39 +02:00
44 lines
1.2 KiB
Python
44 lines
1.2 KiB
Python
from enum import IntEnum
|
|
|
|
|
|
class PoolErrorCodes(IntEnum):
|
|
"""Used in the PoolError class for
|
|
error mapping.
|
|
|
|
Overview of contact error codes:
|
|
- 2000 KILL_ALL_FAILED
|
|
- 2001 NEW_CONNECTION_FAILED
|
|
- 2002 KEEP_ALIVE_FAILED
|
|
"""
|
|
|
|
KILL_ALL_FAILED = 2000
|
|
NEW_CONNECTION_FAILED = 2001
|
|
KEEP_ALIVE_FAILED = 2002
|
|
|
|
|
|
class PoolError(Exception):
|
|
"""
|
|
Overview of contact error codes:
|
|
- 2000 KILL_ALL_FAILED
|
|
- 2001 NEW_CONNECTION_FAILED
|
|
- 2002 KEEP_ALIVE_FAILED
|
|
"""
|
|
|
|
# For linter
|
|
kill_failed = "Could not kill all connections."
|
|
conn_failed = "Failed to execute due to a registry login error."
|
|
alive_failed = "Failed to keep the connection alive."
|
|
_error_mapping = {
|
|
PoolErrorCodes.KILL_ALL_FAILED: kill_failed,
|
|
PoolErrorCodes.NEW_CONNECTION_FAILED: conn_failed,
|
|
PoolErrorCodes.KEEP_ALIVE_FAILED: alive_failed,
|
|
}
|
|
|
|
def __init__(self, *args, code=None, **kwargs):
|
|
super().__init__(*args, **kwargs)
|
|
self.code = code
|
|
if self.code in self._error_mapping:
|
|
self.message = self._error_mapping.get(self.code)
|
|
|
|
def __str__(self):
|
|
return f"{self.message}"
|