Adding more input context to all error messages.

montepy/errors.py

@@ -189,3 +189,28 @@ class LineExpansionWarning(Warning):
     def __init__(self, message):
         self.message = message
         super().__init__(self.message)
+
+
+def add_line_number_to_exception(error, broken_robot):
+    """ """
+    try:
+        input_obj = broken_robot._input
+        assert input_obj is not None
+    except (AttributeError, AssertionError) as e:
+        lineno = "unknown"
+        file = "unknown"
+        lines = []
+    else:
+        lineno = input_obj.line_number
+        file = str(input_obj.input_file)
+        lines = input_obj.input_lines
+    args = e.args
+    if len(args) > 0:
+        message = args[0]
+    else:
+        message = ""
+    message = f"{message}\nError came from line: {lineno} of {file}.\n"
+    f"{'\n'.join(lines)}"
+    args = (message,) + args[1:]
+    e.args = args
+    raise e

montepy/input_parser/input_file.py

@@ -180,4 +180,4 @@ def write(self, to_write):
             return self._fh.write(to_write)
 
     def __str__(self):
-        return self.name
+        return str(self.name)

montepy/mcnp_problem.py

@@ -385,7 +385,19 @@ def parse_input(self, check_input=False, replace=True):
                         if isinstance(obj, transform.Transform):
                             self._transforms.append(obj, False)
                     if trailing_comment is not None and last_obj is not None:
-                        obj._grab_beginning_comment(trailing_comment)
+                        try:
+                            obj._grab_beginning_comment(trailing_comment)
+                        except Exception as e:
+                            args = e.args
+                            if len(args) > 0:
+                                message = args[0]
+                            else:
+                                message = ""
+                            message = f"{message}\nError came from line: {obj._input.line_number} of {obj._input.input_file}.\n"
+                            message += '\n'.join(obj._input.input_lines)
+                            args = (message,) + args[1:]
+                            e.args = args
+                            raise e
                         last_obj._delete_trailing_comment()
                     trailing_comment = obj.trailing_comment
                     last_obj = obj