check for message attribute (#81)

* check for message

* test

* comment

Author: neelasha23

src/ploomber_core/exceptions.py

@@ -87,6 +87,11 @@ def _add_community_link(e):
     elif COMMUNITY not in e.args[0]:
         message = e.args[0] + COMMUNITY
         e.args = (message,)
+        # Certain exceptions like ClickException have message
+        # as their first parameter. In that case args is not
+        # accessed while raising the error
+        if hasattr(e, "message"):
+            e.message = message
 
     return e
 

tests/test_exceptions.py

@@ -54,6 +54,23 @@ def crash():
     assert exceptions.get_community_link() in str(excinfo.value)
 
 
+def test_modify_exceptions_with_message_attribute():
+    class CustomException(Exception):
+        def __init__(self, message: str) -> None:
+            super().__init__(message)
+            self.message = message
+            self.modify_exception = True
+
+    @exceptions.modify_exceptions
+    def crash():
+        raise CustomException("some message")
+
+    with pytest.raises(CustomException) as excinfo:
+        crash()
+
+    assert exceptions.get_community_link() in str(excinfo.value)
+
+
 def test_modify_exceptions_value_error():
     @exceptions.modify_exceptions
     def crash():