Viewing file:  error.py (2.52 KB) -rw-r--r--
Select action/file-type:
 (+) |  (+) |  (+) | Code (+) | Session (+) |  (+) | SDB (+) |  (+) |  (+) |  (+) |  (+) |  (+) |
from gi.repository import GLib, Gio
def register_error(name, domain, code):
"""Register and map decorated exception class to a DBus error."""
def decorated(cls):
error_registration.register_error(cls, name, domain, code)
return cls
return decorated
def map_error(error_name):
"""Map decorated exception class to a DBus error."""
def decorated(cls):
error_registration.map_error(cls, error_name)
return cls
return decorated
def map_by_default(cls):
"""Map decorated exception class to all unknown DBus errors."""
error_registration.map_by_default(cls)
return cls
class ErrorRegistration(object):
"""Class for mapping exceptions to DBus errors."""
_default = None
_map = dict()
_reversed_map = dict()
def map_by_default(self, exception_cls):
"""Set the exception class as a default."""
self._default = exception_cls
def map_error(self, exception_cls, name):
"""Map the exception class to a DBus name."""
self._map[name] = exception_cls
self._reversed_map[exception_cls] = name
def register_error(self, exception_cls, name, domain, code):
"""Map and register the exception class to a DBus name."""
self.map_error(exception_cls, name)
return Gio.DBusError.register_error(domain, code, name)
def is_registered_exception(self, obj):
"""Is the exception registered?"""
return obj.__class__ in self._reversed_map
def get_dbus_name(self, obj):
"""Get the DBus name of the exception."""
return self._reversed_map.get(obj.__class__)
def get_exception_class(self, name):
"""Get the exception class mapped to the DBus name."""
return self._map.get(name, self._default)
def transform_message(self, name, message):
"""Transform the message of the exception."""
prefix = "{}:{}: ".format("GDBus.Error", name)
if message.startswith(prefix):
return message[len(prefix):]
return message
def transform_exception(self, e):
"""Transform the remote error to the exception."""
if not isinstance(e, GLib.Error):
return e
if not Gio.DBusError.is_remote_error(e):
return e
# Get DBus name of the error.
name = Gio.DBusError.get_remote_error(e)
# Get the exception class.
exception_cls = self.get_exception_class(name)
# Return the original exception.
if not exception_cls:
return e
# Return new exception.
message = self.transform_message(name, e.message)
exception = exception_cls(message)
exception.dbus_name = name
exception.dbus_domain = e.domain
exception.dbus_code = e.code
return exception
# Default error registration.
error_registration = ErrorRegistration()
|