python环境检测

基于python代码实现的python环境检测：

包含电脑系统

python版本

依赖包检查

环境变量检查

当然啦，用户可以基于自己的需求增加多种项目检查，如自定义多种依赖包。

import sys import platform import os from packaging.version import parse # 替换弃用的pkg_resources try: from colorama import init, Fore, Style HAS_COLORAMA = True except ImportError: HAS_COLORAMA = False # 初始化颜色输出 if HAS_COLORAMA: init(autoreset=True) # 配置检查项 REQUIRED_PYTHON_VERSION = (3, 6) REQUIRED_PACKAGES = { "numpy": "1.18.0", "pandas": "1.0.0", "requests": "2.24.0" } REQUIRED_ENV_VARS = ["HOME", "PATH"] class EnvironmentChecker: def __init__(self): self.report = { "python_version": {}, "operating_system": {}, "packages": {}, "environment_vars": {} } def check_python_version(self): """检查Python版本""" result = { "required": ".".join(map(str, REQUIRED_PYTHON_VERSION)), "installed": platform.python_version(), "status": "OK" } if sys.version_info < REQUIRED_PYTHON_VERSION: result["status"] = "ERROR" elif sys.version_info[0] != REQUIRED_PYTHON_VERSION[0]: result["status"] = "ERROR" else: result["status"] = "OK" self.report["python_version"] = result return result def check_operating_system(self): """检查操作系统""" os_info = { "system": platform.system(), "release": platform.release(), "version": platform.version(), "status": "OK" # 操作系统检查通常不会失败，但保留状态字段 } self.report["operating_system"] = os_info return os_info def check_packages(self): """检查安装的包""" for pkg, req_version in REQUIRED_PACKAGES.items(): try: __import__(pkg) installed_version = sys.modules[pkg].__version__ status = "OK" if parse(installed_version) >= parse(req_version) else "WARNING" except ImportError: installed_version = "Not installed" status = "ERROR" except AttributeError: installed_version = "Unknown version" status = "WARNING" self.report["packages"][pkg] = { "required": req_version, "installed": installed_version, "status": status } return self.report["packages"] def check_environment_vars(self): """检查环境变量""" for var in REQUIRED_ENV_VARS: exists = var in os.environ self.report["environment_vars"][var] = { "exists": exists, "status": "OK" if exists else "WARNING" } return self.report["environment_vars"] def generate_report(self, save_to_file=False): """生成检查报告""" self.check_python_version() self.check_operating_system() self.check_packages() self.check_environment_vars() # 控制台输出 self._print_colored("Python版本检查:", self.report["python_version"]) self._print_colored("\n操作系统信息:", self.report["operating_system"]) print("\n依赖包检查:") for pkg, info in self.report["packages"].items(): self._print_colored(f"- {pkg}", info) print("\n环境变量检查:") for var, info in self.report["environment_vars"].items(): self._print_colored(f"- {var}", info) # 保存到文件 if save_to_file: with open("environment_report.txt", "w") as f: import json json.dump(self.report, f, indent=2) def _print_colored(self, title, data): """带颜色输出""" if "status" in data: status = data["status"] color = Fore.GREEN if status == "WARNING": color = Fore.YELLOW elif status == "ERROR": color = Fore.RED details = [] for k, v in data.items(): if k != "status": details.append(f"{k}={v}") if HAS_COLORAMA: print(f"{color}{title.ljust(20)}{Style.RESET_ALL} {', '.join(details)}") else: print(f"{title.ljust(20)} [{status}] {', '.join(details)}") else: details = [f"{k}={v}" for k, v in data.items()] print(f"{title.ljust(20)} {', '.join(details)}") if __name__ == "__main__": checker = EnvironmentChecker() print("=" * 60) print("Python环境检查报告".center(60)) print("=" * 60) checker.generate_report(save_to_file=True) print("\n" + "=" * 60) # 修正后的错误检查逻辑 has_errors = False # 检查Python版本状态 if checker.report["python_version"]["status"] == "ERROR": has_errors = True # 检查所有包的安装状态 for pkg_info in checker.report["packages"].values(): if pkg_info["status"] == "ERROR": has_errors = True # 检查环境变量状态 for env_info in checker.report["environment_vars"].values(): if env_info["status"] == "ERROR": # 注意这里原代码中没有ERROR状态，只有WARNING has_errors = True if has_errors: sys.exit(1) else: sys.exit(0)