fix bug in suggest module

Author: Locked-chess-official

Lib/traceback.py

@@ -13,6 +13,7 @@
 import tokenize
 import io
 import importlib.util
+from importlib.machinery import ModuleSpec
 import pathlib
 import _colorize
 
@@ -1125,16 +1126,14 @@ def __init__(self, exc_type, exc_value, exc_traceback, *, limit=None,
                     self._str += f". Did you mean: '{suggestion}' ({suggestion!a})?"
         elif exc_type and issubclass(exc_type, ModuleNotFoundError):
             module_name = getattr(exc_value, "name", None)
+            analyse_sys_no_site = True
             if module_name in sys.stdlib_module_names:
                 message = _MISSING_STDLIB_MODULE_MESSAGES.get(
                     module_name,
                     f"Standard library module {module_name!r} was not found"
                 )
                 self._str = message
-            elif sys.flags.no_site:
-                self._str += (". Site initialization is disabled, did you forget to "
-                    + "add the site-packages directory to sys.path "
-                    + "or to enable your virtual environment?")
+                analyse_sys_no_site = False
             elif abi_tag := _find_incompatible_extension_module(module_name):
                 self._str += (
                     ". Although a module with this name was found for a "
@@ -1144,6 +1143,12 @@ def __init__(self, exc_type, exc_value, exc_traceback, *, limit=None,
                 suggestion = _compute_suggestion_error(exc_value, exc_traceback, module_name)
                 if suggestion:
                     self._str += f". Did you mean: '{suggestion}'?"
+            if analyse_sys_no_site and sys.flags.no_site:
+                if not self._str.endswith((".", "?")):
+                    self._str += "."
+                self._str += (" Site initialization is disabled, did you forget to "
+                    "add the site-packages directory to sys.path "
+                    "or to enable your virtual environment?")
         elif exc_type and issubclass(exc_type, AttributeError) and \
                 getattr(exc_value, "name", None) is not None:
             wrong_name = getattr(exc_value, "name", None)
@@ -1758,7 +1763,10 @@ def _compute_suggestion_error(exc_value, tb, wrong_name):
             d = []
             for finder in sys.meta_path:
                 if discover := getattr(finder, 'discover', None):
-                    d += [spec.name for spec in discover(parent)]
+                    try:
+                        d += [spec.name for spec in discover(parent) if isinstance(spec, ModuleSpec)]
+                    except Exception:
+                        continue
         except Exception:
             return None
     elif isinstance(exc_value, ImportError):