Move changes back to load_module_from_name()

Author: jacobtylerwalls

astroid/manager.py

@@ -10,12 +10,9 @@
 from __future__ import annotations
 
 import collections
-import io
-import logging
 import os
 import types
 import zipimport
-from contextlib import redirect_stderr, redirect_stdout
 from importlib.util import find_spec, module_from_spec
 from typing import TYPE_CHECKING, ClassVar
 
@@ -42,9 +39,6 @@
 ZIP_IMPORT_EXTS = (".zip", ".egg", ".whl", ".pyz", ".pyzw")
 
 
-logger = logging.getLogger(__name__)
-
-
 def safe_repr(obj):
     try:
         return repr(obj)
@@ -179,29 +173,13 @@ def ast_from_module_name(self, modname, context_file=None):
                 ):
                     return self._build_stub_module(modname)
                 try:
-                    # Capture and log anything emitted during import to avoid
-                    # contaminating JSON reports in pylint
-                    with redirect_stderr(io.StringIO()) as stderr, redirect_stdout(
-                        io.StringIO()
-                    ) as stdout:
-                        module = load_module_from_name(modname)
+                    module = load_module_from_name(modname)
                 except Exception as e:
                     raise AstroidImportError(
                         "Loading {modname} failed with:\n{error}",
                         modname=modname,
                         path=found_spec.location,
                     ) from e
-                stderr_value = stderr.getvalue()
-                if stderr_value:
-                    logger.error(
-                        "Captured stderr from importing %s:\n%s", modname, stderr_value
-                    )
-                stdout_value = stdout.getvalue()
-                if stdout_value:
-                    logger.info(
-                        "Captured stdout from importing %s:\n%s", modname, stdout_value
-                    )
-
                 return self.ast_from_module(module, modname)
 
             elif found_spec.type == spec.ModuleType.PY_COMPILED:

astroid/modutils.py

@@ -19,17 +19,23 @@
 import importlib
 import importlib.machinery
 import importlib.util
+import io
 import itertools
+import logging
 import os
 import sys
 import sysconfig
 import types
+from contextlib import redirect_stderr, redirect_stdout
 from functools import lru_cache
 from pathlib import Path
 
 from astroid.const import IS_JYTHON, IS_PYPY
 from astroid.interpreter._import import spec, util
 
+logger = logging.getLogger(__name__)
+
+
 if sys.platform.startswith("win"):
     PY_SOURCE_EXTS = ("py", "pyw")
     PY_COMPILED_EXTS = ("dll", "pyd")
@@ -171,7 +177,23 @@ def load_module_from_name(dotted_name: str) -> types.ModuleType:
     except KeyError:
         pass
 
-    return importlib.import_module(dotted_name)
+    # Capture and log anything emitted during import to avoid
+    # contaminating JSON reports in pylint
+    with redirect_stderr(io.StringIO()) as stderr, redirect_stdout(
+        io.StringIO()
+    ) as stdout:
+        module = importlib.import_module(dotted_name)
+
+    stderr_value = stderr.getvalue()
+    if stderr_value:
+        logger.error(
+            "Captured stderr from importing %s:\n%s", dotted_name, stderr_value
+        )
+    stdout_value = stdout.getvalue()
+    if stdout_value:
+        logger.info("Captured stdout from importing %s:\n%s", dotted_name, stdout_value)
+
+    return module
 
 
 def load_module_from_modpath(parts):