mirror of https://github.com/conqp/rcon
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
44 lines
1.2 KiB
44 lines
1.2 KiB
"""Common errors handler."""
|
|
|
|
from logging import Logger
|
|
from socket import timeout
|
|
|
|
from rcon.exceptions import ConfigReadError, RequestIdMismatch, WrongPassword
|
|
|
|
|
|
__all__ = ['ErrorHandler']
|
|
|
|
|
|
class ErrorHandler:
|
|
"""Handles common errors and exits."""
|
|
|
|
__slots__ = ('logger', 'exit_code')
|
|
|
|
def __init__(self, logger: Logger):
|
|
"""Sets the logger."""
|
|
self.logger = logger
|
|
self.exit_code = 0
|
|
|
|
def __enter__(self):
|
|
return self
|
|
|
|
def __exit__(self, _, value, __):
|
|
"""Checks for connection errors and exits respectively."""
|
|
if isinstance(value, ConnectionRefusedError):
|
|
self.logger.error('Connection refused.')
|
|
self.exit_code = 3
|
|
elif isinstance(value, (TimeoutError, timeout)):
|
|
self.logger.error('Connection timed out.')
|
|
self.exit_code = 4
|
|
elif isinstance(value, WrongPassword):
|
|
self.logger.error('Wrong password.')
|
|
self.exit_code = 5
|
|
elif isinstance(value, RequestIdMismatch):
|
|
self.logger.error('Session timed out.')
|
|
self.exit_code = 6
|
|
elif isinstance(value, ConfigReadError):
|
|
self.exit_code = value.exit_code
|
|
else:
|
|
return None
|
|
|
|
return True
|
|
|