a6dc10dad2* Reduce and tighten comments and docstrings in tests Shorten verbose comments and docstrings across the test suite without changing any test logic. Remove narration that restates the next line, collapse long module and test docstrings to a single line, and drop banner separators. Keep regression context (issue and PR references, run ids), skip reasons, mocking and timing rationale, license headers, lint and type directives, and commented-out code. Comments and docstrings only: an AST signature check confirms no code, assertions, or string literals changed, and the suite byte-compiles cleanly. * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --------- Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
122 lines
3.8 KiB
Python
122 lines
3.8 KiB
Python
import subprocess
|
|
import sys
|
|
import os
|
|
import shutil
|
|
import importlib
|
|
|
|
|
|
def detect_package_manager():
|
|
"""Detect the available package manager"""
|
|
package_managers = {
|
|
"apt": "/usr/bin/apt",
|
|
"yum": "/usr/bin/yum",
|
|
"dnf": "/usr/bin/dnf",
|
|
"pacman": "/usr/bin/pacman",
|
|
"zypper": "/usr/bin/zypper",
|
|
}
|
|
|
|
for pm, path in package_managers.items():
|
|
if os.path.exists(path):
|
|
return pm
|
|
return None
|
|
|
|
|
|
def check_package_installed(package_name, package_manager = None):
|
|
"""Check if a package is installed using the system package manager"""
|
|
|
|
if package_manager is None:
|
|
package_manager = detect_package_manager()
|
|
|
|
if package_manager is None:
|
|
print("Warning: Could not detect package manager")
|
|
return None
|
|
|
|
try:
|
|
if package_manager == "apt":
|
|
result = subprocess.run(["dpkg", "-l", package_name], capture_output = True, text = True)
|
|
return result.returncode == 0
|
|
|
|
elif package_manager in ["yum", "dnf"]:
|
|
result = subprocess.run(["rpm", "-q", package_name], capture_output = True, text = True)
|
|
return result.returncode == 0
|
|
|
|
elif package_manager == "pacman":
|
|
result = subprocess.run(["pacman", "-Q", package_name], capture_output = True, text = True)
|
|
return result.returncode == 0
|
|
|
|
elif package_manager == "zypper":
|
|
result = subprocess.run(
|
|
["zypper", "se", "-i", package_name], capture_output = True, text = True
|
|
)
|
|
return package_name in result.stdout
|
|
|
|
except Exception as e:
|
|
print(f"Error checking package: {e}")
|
|
return None
|
|
|
|
|
|
def require_package(package_name, executable_name = None):
|
|
"""Require a package to be installed, exit if not found"""
|
|
|
|
# Executable in PATH is the most reliable signal
|
|
if executable_name:
|
|
if shutil.which(executable_name):
|
|
print(f"✓ {executable_name} is available")
|
|
return
|
|
|
|
pm = detect_package_manager()
|
|
is_installed = check_package_installed(package_name, pm)
|
|
|
|
if is_installed:
|
|
print(f"✓ Package {package_name} is installed")
|
|
return
|
|
|
|
print(f"❌ Error: {package_name} is not installed")
|
|
print(f"\nPlease install {package_name} using your system package manager:")
|
|
|
|
install_commands = {
|
|
"apt": f"sudo apt update && sudo apt install {package_name}",
|
|
"yum": f"sudo yum install {package_name}",
|
|
"dnf": f"sudo dnf install {package_name}",
|
|
"pacman": f"sudo pacman -S {package_name}",
|
|
"zypper": f"sudo zypper install {package_name}",
|
|
}
|
|
|
|
if pm and pm in install_commands:
|
|
print(f" {install_commands[pm]}")
|
|
else:
|
|
for pm_name, cmd in install_commands.items():
|
|
print(f" {pm_name}: {cmd}")
|
|
|
|
print(f"\nAlternatively, install with conda:")
|
|
print(f" conda install -c conda-forge {package_name}")
|
|
|
|
print(f"\nPlease install the required package and run the script again.")
|
|
sys.exit(1)
|
|
|
|
|
|
# Usage
|
|
# require_package("ffmpeg", "ffmpeg")
|
|
|
|
|
|
def require_python_package(
|
|
package_name,
|
|
import_name = None,
|
|
pip_name = None,
|
|
):
|
|
"""Require a Python package to be installed, exit if not found"""
|
|
if import_name is None:
|
|
import_name = package_name
|
|
if pip_name is None:
|
|
pip_name = package_name
|
|
|
|
if importlib.util.find_spec(import_name) is None:
|
|
print(f"❌ Error: Python package '{package_name}' is not installed")
|
|
print(f"\nPlease install {package_name} using pip:")
|
|
print(f" pip install {pip_name}")
|
|
print(f" # or with conda:")
|
|
print(f" conda install {pip_name}")
|
|
print(f"\nAfter installation, run this script again.")
|
|
sys.exit(1)
|
|
else:
|
|
print(f"✓ Python package '{package_name}' is installed")
|