Move log capture lower level

Author: pllim

jwst/msaflagopen/msaflag_open.py

@@ -10,7 +10,13 @@
 from stdatamodels.jwst import datamodels
 from stdatamodels.jwst.transforms.models import Slit
 
-from jwst.assign_wcs.nirspec import slitlets_wcs, _nrs_wcs_set_input_lite, _get_transforms
+from jwst.assign_wcs.nirspec import (
+    slitlets_wcs,
+    _nrs_wcs_set_input_lite,
+    _get_transforms,
+    log as nirspec_log,
+)
+from jwst.lib.basic_utils import LoggingContext
 
 log = logging.getLogger(__name__)
 log.setLevel(logging.DEBUG)
@@ -82,7 +88,8 @@ def flag(input_datamodel, failed_slitlets, wcs_refnames):
         Science data with DQ flags modified.
     """
     # Use the machinery in assign_wcs to create a WCS object for the bad shutters
-    pipeline = slitlets_wcs(input_datamodel, wcs_refnames, failed_slitlets)
+    with LoggingContext(nirspec_log, level=logging.WARNING):
+        pipeline = slitlets_wcs(input_datamodel, wcs_refnames, failed_slitlets)
     wcs = WCS(pipeline)
 
     # Create output as a copy of the input science data model so we can overwrite

jwst/msaflagopen/msaflagopen_step.py

@@ -1,12 +1,9 @@
-import logging
-
 from stdatamodels.jwst import datamodels
+
 from stpipe.crds_client import reference_uri_to_cache_path
 
 from jwst.stpipe import Step
 from jwst.assign_wcs import AssignWcsStep
-from jwst.assign_wcs.nirspec import log as nirspec_log
-from jwst.lib.basic_utils import LoggingContext
 from jwst.msaflagopen import msaflag_open
 
 __all__ = ["MSAFlagOpenStep"]
@@ -53,10 +50,7 @@ def process(self, input_data):
             wcs_reffile_names = create_reference_filename_dictionary(input_model)
 
             # Do the DQ flagging
-            with LoggingContext(nirspec_log, level=logging.WARNING):
-                result = msaflag_open.do_correction(
-                    input_model, self.reference_name, wcs_reffile_names
-                )
+            result = msaflag_open.do_correction(input_model, self.reference_name, wcs_reffile_names)
 
             # set the step status to complete
             result.meta.cal_step.msa_flagging = "COMPLETE"