更改enroll命名,添加了注释,向get_error_msg中添加了一些错误代码

venv/Lib/site-packages/_distutils_hack/__init__.py

@@ -1,11 +1,19 @@
-# don't import any costly modules
 import sys
 import os
+import re
+import importlib
+import warnings
+import contextlib
 
 
 is_pypy = '__pypy__' in sys.builtin_module_names
 
 
+warnings.filterwarnings('ignore',
+                        r'.+ distutils\b.+ deprecated',
+                        DeprecationWarning)
+
+
 def warn_distutils_present():
     if 'distutils' not in sys.modules:
         return
@@ -13,7 +21,6 @@ def warn_distutils_present():
         # PyPy for 3.6 unconditionally imports distutils, so bypass the warning
         # https://foss.heptapod.net/pypy/pypy/-/blob/be829135bc0d758997b3566062999ee8b23872b4/lib-python/3/site.py#L250
         return
-    import warnings
     warnings.warn(
         "Distutils was imported before Setuptools, but importing Setuptools "
         "also replaces the `distutils` module in `sys.modules`. This may lead "
@@ -26,12 +33,8 @@ def warn_distutils_present():
 def clear_distutils():
     if 'distutils' not in sys.modules:
         return
-    import warnings
     warnings.warn("Setuptools is replacing distutils.")
-    mods = [
-        name for name in sys.modules
-        if name == "distutils" or name.startswith("distutils.")
-    ]
+    mods = [name for name in sys.modules if re.match(r'distutils\b', name)]
     for name in mods:
         del sys.modules[name]
 
@@ -45,7 +48,6 @@ def enabled():
 
 
 def ensure_local_distutils():
-    import importlib
     clear_distutils()
 
     # With the DistutilsMetaFinder in place,
@@ -57,7 +59,6 @@ def ensure_local_distutils():
     # check that submodules load as expected
     core = importlib.import_module('distutils.core')
     assert '_distutils' in core.__file__, core.__file__
-    assert 'setuptools._distutils.log' not in sys.modules
 
 
 def do_override():
@@ -72,14 +73,6 @@ def do_override():
         ensure_local_distutils()
 
 
-class _TrivialRe:
-    def __init__(self, *patterns):
-        self._patterns = patterns
-
-    def match(self, string):
-        return all(pat in string for pat in self._patterns)
-
-
 class DistutilsMetaFinder:
     def find_spec(self, fullname, path, target=None):
         if path is not None:
@@ -90,10 +83,6 @@ class DistutilsMetaFinder:
         return method()
 
     def spec_for_distutils(self):
-        if self.is_cpython():
-            return
-
-        import importlib
         import importlib.abc
         import importlib.util
 
@@ -113,23 +102,12 @@ class DistutilsMetaFinder:
         class DistutilsLoader(importlib.abc.Loader):
 
             def create_module(self, spec):
-                mod.__name__ = 'distutils'
                 return mod
 
             def exec_module(self, module):
                 pass
 
-        return importlib.util.spec_from_loader(
-            'distutils', DistutilsLoader(), origin=mod.__file__
-        )
-
-    @staticmethod
-    def is_cpython():
-        """
-        Suppress supplying distutils for CPython (build and tests).
-        Ref #2965 and #3007.
-        """
-        return os.path.isfile('pybuilddir.txt')
+        return importlib.util.spec_from_loader('distutils', DistutilsLoader())
 
     def spec_for_pip(self):
         """
@@ -168,11 +146,12 @@ def add_shim():
     DISTUTILS_FINDER in sys.meta_path or insert_shim()
 
 
-class shim:
-    def __enter__(self):
-        insert_shim()
-
-    def __exit__(self, exc, value, tb):
+@contextlib.contextmanager
+def shim():
+    insert_shim()
+    try:
+        yield
+    finally:
         remove_shim()