Ensure Python venv module is installed

Author: ivankravets

pioinstaller/__main__.py

@@ -90,7 +90,7 @@ def python():
             % (platform.python_version(), util.get_pythonexe_path()),
             fg="green",
         )
-    except (exception.IncompatiblePythonError, exception.DistutilsNotFound) as e:
+    except (exception.IncompatiblePythonError, exception.PythonVenvModuleNotFound) as e:
         raise click.ClickException(
             "The Python %s (%s) interpreter is not compatible.\nReason: %s"
             % (platform.python_version(), util.get_pythonexe_path(), str(e))
@@ -123,7 +123,7 @@ def core_check(ctx, **kwargs):
             % (state.get("core_version"), state.get("platformio_exe")),
             fg="green",
         )
-    except (exception.InvalidPlatformIOCore) as e:
+    except exception.InvalidPlatformIOCore as e:
         raise click.ClickException(
             "Compatible PlatformIO Core not found.\nReason: %s" % str(e)
         )

pioinstaller/exception.py

@@ -14,7 +14,6 @@
 
 
 class PIOInstallerException(Exception):
-
     MESSAGE = None
 
     def __str__(self):  # pragma: no cover
@@ -26,15 +25,12 @@ def __str__(self):  # pragma: no cover
 
 
 class IncompatiblePythonError(PIOInstallerException):
-
     MESSAGE = "{0}"
 
 
-class DistutilsNotFound(PIOInstallerException):
-
-    MESSAGE = "Could not find distutils module"
+class PythonVenvModuleNotFound(PIOInstallerException):
+    MESSAGE = "Could not find Python `venv` module"
 
 
 class InvalidPlatformIOCore(PIOInstallerException):
-
     MESSAGE = "{0}"

pioinstaller/python.py

@@ -138,9 +138,10 @@ def check():
         raise exception.IncompatiblePythonError("Conda is not supported")
 
     try:
-        __import__("distutils.command")
+        __import__("venv")
+        # __import__("distutils.command")
     except ImportError:
-        raise exception.DistutilsNotFound()
+        raise exception.PythonVenvModuleNotFound()
 
     # portable Python 3 for macOS is not compatible with macOS < 10.13
     # https://github.com/platformio/platformio-core-installer/issues/70
@@ -224,10 +225,10 @@ def find_compatible_pythons(
                 log.debug(error)
             except UnicodeDecodeError:
                 pass
-            if error and "Could not find distutils module" in error:
+            if error and "`venv` module" in error:
                 # pylint:disable=line-too-long
                 raise click.ClickException(
-                    """Can not install PlatformIO Core due to a missed `distutils` package in your Python installation.
+                    """Can not install PlatformIO Core due to a missed `venv` module in your Python installation.
 Please install this package manually using the OS package manager. For example:
 
 $ apt-get install python3-venv