issue #125: handle frozen classes as exception

noxdafox · noxdafox · commit b42bfc426af4 · 2024-03-20T23:55:38.000+02:00
Seriously, who is the nutcrack which thought this was a good idea?

Signed-off-by: Matteo Cafasso &lt;noxdafox@gmail.com&gt;
diff --git a/pebble/common.py b/pebble/common.py
@@ -122,7 +122,11 @@ def __reduce__(self):
 
 
 def rebuild_exception(exception, traceback):
-    exception.__cause__ = RemoteTraceback(traceback)
+    try:
+        exception.traceback = traceback
+        exception.__cause__ = RemoteTraceback(traceback)
+    except AttributeError:  # Frozen exception
+        pass
 
     return exception
 
@@ -165,7 +169,11 @@ def execute(function, *args, **kwargs):
     try:
         return Result(SUCCESS, function(*args, **kwargs))
     except BaseException as error:
-        error.traceback = format_exc()
+        try:
+            error.traceback = format_exc()
+        except AttributeError:  # Frozen exception
+            pass
+
         return Result(FAILURE, error)
 
 
@@ -174,17 +182,15 @@ def process_execute(function, *args, **kwargs):
     try:
         return Result(SUCCESS, function(*args, **kwargs))
     except BaseException as error:
-        error.traceback = format_exc()
-        return Result(FAILURE, RemoteException(error, error.traceback))
+        return Result(FAILURE, RemoteException(error, format_exc()))
 
 
 def send_result(pipe, data):
     """Send result handling pickling and communication errors."""
     try:
         pipe.send(data)
     except (pickle.PicklingError, TypeError) as error:
-        error.traceback = format_exc()
-        pipe.send(Result(ERROR, RemoteException(error, error.traceback)))
+        pipe.send(Result(ERROR, RemoteException(error, format_exc())))
 
 
 Result = namedtuple('Result', ('status', 'value'))
