put error/warning message in variable for reuse

Brian-Taylor8 · Jan 8, 2023 · d5fdbeb · d5fdbeb
1 parent d8b5f1d
commit d5fdbeb
Showing 1 changed file with 4 additions and 8 deletions.
diff --git a/rt_utils/rtstruct_builder.py b/rt_utils/rtstruct_builder.py
@@ -79,13 +79,9 @@ def validate_contour_image_in_series_data(
  return
 
  # ReferencedSOPInstanceUID is NOT available
- if warning_only:
- warnings.warn(
- f"Loaded RTStruct references image(s) that are not contained in input series data. "
+ msg = f"Loaded RTStruct references image(s) that are not contained in input series data. "
  + f"Problematic image has SOP Instance Id: {contour_image.ReferencedSOPInstanceUID}"
- )
+ if warning_only:
+ warnings.warn(msg)
  else:
- raise Exception(
- f"Loaded RTStruct references image(s) that are not contained in input series data. "
- + f"Problematic image has SOP Instance Id: {contour_image.ReferencedSOPInstanceUID}"
- )
+ raise Exception(msg)