Class EnvironmentManager

mod/project/python/pyenv_tool.py:1109–1397

class EnvironmentManager:
    _REPORT_FILE = "/www/server/panel/data/python_project_env.json"

    def __init__(self):
        self._all_env = None

    @property
    def all_env(self) -> List[PythonEnvironment]:
        venv_src_map = {}
        if self._all_env is None:
            self._all_env = []
            for e in self.load_report()["environments"]:
                if not os.path.isfile(e["bin_path"]):  # 文件已不存在的，就不在再展示
                    continue
                tmp_p = PythonEnvironment.from_dict(e)
                self._all_env.append(tmp_p)
                if tmp_p.env_type == "venv":
                    venv_src_map[tmp_p.system_path] = tmp_p

        for tmp_p in self._all_env:
            if tmp_p.env_type == "system" and tmp_p.bin_path in venv_src_map:
                venv_src_map.pop(tmp_p.bin_path)

        for tmp_p in venv_src_map.values():
            self._all_env.remove(tmp_p)

        return self._all_env

    def get_env_py_path(self, python_path: str) -> Optional[PythonEnvironment]:
        python_path = python_path.rstrip("/")
        for tmp_p in self.all_env:
            if tmp_p.bin_path == python_path:
                return tmp_p
        # 兼容旧版本
        if python_path.startswith("/www/server/pyporject_evn/"):
            for tmp_p in self.all_env:
                if tmp_p.bin_path.startswith(python_path) and tmp_p.env_type == "system":
                    return tmp_p
        return None

    @staticmethod
    def load_report() -> Dict:
        with open(EnvironmentReporter.REPORT_FILE, "r") as fs:
            return json.loads(fs.read())

    @staticmethod
    def add_python_env(add_type: str, path: str) -> Optional[str]:
        p = Path(path)
        if not p.exists():
            return "Python环境不存在"
        if path.find("conda") and add_type == "system":
            tmp_path = path.rsplit("/bin")[0]
            if os.path.isfile(tmp_path + "/condabin/conda"):
                add_type = "conda"
                path = tmp_path + "/condabin/conda"
            elif os.path.isfile(tmp_path + "/../../condabin/conda"):
                add_type = "conda"
                path = os.path.realpath(tmp_path + "/../../condabin/conda")