omnata-plugin-runtime 0.1.4__tar.gz → 0.1.6__tar.gz

{omnata_plugin_runtime-0.1.4 → omnata_plugin_runtime-0.1.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: omnata-plugin-runtime
-Version: 0.1.4
+Version: 0.1.6
 Summary: A development kit to assist with building, testing and publishing Omnata Plugins
 Author: James Weakley
 Author-email: james.weakley@omnata.com

{omnata_plugin_runtime-0.1.4 → omnata_plugin_runtime-0.1.6}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "omnata-plugin-runtime"
-version = "0.1.4"
+version = "0.1.6"
 description = "A development kit to assist with building, testing and publishing Omnata Plugins"
 authors = ["James Weakley <james.weakley@omnata.com>"]
 readme = "README.md"

{omnata_plugin_runtime-0.1.4 → omnata_plugin_runtime-0.1.6}/src/omnata_plugin_runtime/logging.py

@@ -10,7 +10,7 @@ import pandas
 from snowflake.snowpark import Session
 from pydantic import ValidationError
 
-class CustomLogger(logging.Logger):
+class CustomLogger(logging.getLoggerClass()):
     """
     Custom logger that can handle pydantic validation errors.
     Without this, you get "Object of type ErrorWrapper is not JSON serializable"
@@ -18,7 +18,12 @@ class CustomLogger(logging.Logger):
     """
     def error(self, msg, *args, **kwargs):
         if isinstance(msg, ValidationError):
-            msg = msg.errors()
+            # If the exc_info or stack_info flags are set, unset them and manually
+            # add the error information to the log message.
+            exc_info = kwargs.pop('exc_info', False)
+            stack_info = kwargs.pop('stack_info', False)
+            if exc_info or stack_info:
+                msg = f"{msg.errors()} (originally raised at {msg.__traceback__.tb_frame.f_code.co_filename}:{msg.__traceback__.tb_lineno})"
         super().error(msg, *args, **kwargs)
 
 class OmnataPluginLogHandler(logging.handlers.BufferingHandler):