mirror of https://github.com/mongodb/mongo
87 lines
3.3 KiB
Python
87 lines
3.3 KiB
Python
# -*- mode: python; -*-
|
|
|
|
# Try to keep this modules imports minimum and only
|
|
# import python standard modules, because this module
|
|
# should be used for finding such external modules or
|
|
# missing dependencies.
|
|
import platform
|
|
import re
|
|
import subprocess
|
|
import sys
|
|
|
|
|
|
class MissingRequirements(Exception):
|
|
"""Raised when when verify_requirements() detects missing requirements."""
|
|
pass
|
|
|
|
|
|
def verify_requirements(silent: bool = False, executable=sys.executable):
|
|
"""Check if the modules in a pip requirements file are installed.
|
|
This allows for a more friendly user message with guidance on how to
|
|
resolve the missing dependencies.
|
|
Args:
|
|
requirements_file: path to a pip requirements file.
|
|
silent: True if the function should print.
|
|
Raises:
|
|
MissingRequirements if any requirements are missing
|
|
"""
|
|
|
|
def verbose(*args, **kwargs):
|
|
if not silent:
|
|
print(*args, **kwargs)
|
|
|
|
def raiseSuggestion(ex, pip_pkg):
|
|
raise MissingRequirements(f"{ex}\n"
|
|
f"Try running:\n"
|
|
f" {executable} -m pip install {pip_pkg}") from ex
|
|
|
|
# Import poetry. If this fails then we know the next function will fail.
|
|
# This is so the user will have an easier time diagnosing the problem
|
|
try:
|
|
import poetry
|
|
except ModuleNotFoundError as ex:
|
|
raiseSuggestion(ex, "'poetry==2.0.0'")
|
|
|
|
verbose("Checking required python packages...")
|
|
|
|
try:
|
|
extras = []
|
|
if platform.machine() in set(["s390x", "ppc64le"]) and ".el9" not in platform.release():
|
|
extras = ["--extras", "oldcrypt"]
|
|
poetry_dry_run_proc = subprocess.run(
|
|
[executable, "-m", "poetry", "install", "--no-root", "--sync", "--dry-run", *extras],
|
|
check=True,
|
|
text=True,
|
|
capture_output=True,
|
|
errors="backslashreplace",
|
|
)
|
|
except subprocess.CalledProcessError as exc:
|
|
print("ERROR: poetry packages verification failed.")
|
|
print(exc.stdout)
|
|
print(exc.stderr)
|
|
raise MissingRequirements(
|
|
"Detected one or more packages are out of date. "
|
|
"Try running:\n"
|
|
" export PYTHON_KEYRING_BACKEND=keyring.backends.null.Keyring\n"
|
|
" python3 -m poetry install --no-root --sync")
|
|
|
|
# String match should look like the following
|
|
# Package operations: 2 installs, 3 updates, 0 removals, 165 skipped
|
|
match = re.search(r"Package operations: (\d+) \w+, (\d+) \w+, (\d+) \w+, (\d+) \w+",
|
|
poetry_dry_run_proc.stdout)
|
|
verbose("Requirements list:")
|
|
verbose(poetry_dry_run_proc.stdout)
|
|
installs = int(match[1])
|
|
updates = int(match[2])
|
|
if updates == 1 and sys.platform == 'win32' and "Updating pywin32" in poetry_dry_run_proc.stdout:
|
|
# We have no idea why pywin32 thinks it needs to be updated
|
|
# We could use some more investigation into this
|
|
verbose(
|
|
"Windows detected a single update to pywin32 which is known to be buggy. Continuing.")
|
|
elif installs + updates > 0:
|
|
raise MissingRequirements(
|
|
f"Detected one or more packages are out of date. "
|
|
f"Try running:\n"
|
|
f" export PYTHON_KEYRING_BACKEND=keyring.backends.null.Keyring\n"
|
|
f" {executable} -m poetry install --no-root --sync")
|