| import os.path |
| import sys |
| import json |
|
|
| if "/www/server/panel/class" not in sys.path: |
| sys.path.insert(0, "/www/server/panel/class") |
|
|
| if "/www/server/panel" not in sys.path: |
| sys.path.insert(0, "/www/server/panel") |
|
|
| from mod.project.python.pyenv_tool import EnvironmentManager, _SYS_BIN_PATH, python_manager_path, pyenv_path |
| from mod.base import json_response |
| import public |
|
|
|
|
| class main: |
|
|
| @staticmethod |
| def create_python_env(get): |
| try: |
| venv_name = get.venv_name.strip() |
| src_python_bin = get.python_bin.strip() |
| if "ps" in get: |
| ps = get.ps.strip() |
| else: |
| ps = "" |
| except: |
| return json_response(False, '参数错误') |
|
|
| ws_send = None |
| if "_ws" in get: |
| ws_send = lambda x: get._ws.send(json.dumps({"callback": "create_python_env", "result": {"log": x}})) |
| res = EnvironmentManager().create_python_env(venv_name, src_python_bin, ps, ws_send) |
| if isinstance(res, str): |
| return json_response(False, res) |
| return json_response(True, msg="创建成功") |
|
|
| @staticmethod |
| def list_environment(get): |
| if get and get.get("sort_not_use/s") in ("1", "true"): |
| sort_not_use = True |
| else: |
| sort_not_use = False |
|
|
| em = EnvironmentManager() |
| all_project_map = em.all_python_project_map() |
| env_map = { |
| i.bin_path: i.to_dict( |
| project_name=all_project_map.get(i.bin_path, []), |
| can_remove=i.can_remove, |
| can_create=i.can_create, |
| can_set_default=i.can_set_default, |
| path_name=i.path_name, |
| from_panel=any((i.bin_path.startswith(x) for x in (python_manager_path(), pyenv_path()))) and i.env_type == "system" |
| ) for i in em.all_env |
| } |
| data = list(env_map.values()) |
| if sort_not_use: |
| data.sort(key=lambda x: ( |
| len(x["project_name"]), |
| ("venv", "conda", "system").index(x["type"]), |
| any(i == os.path.dirname(x["bin_path"]) for i in _SYS_BIN_PATH) |
| )) |
| else: |
| data.sort(key=lambda x: ( |
| 0 - len(x["project_name"]), |
| ("venv", "conda", "system").index(x["type"]), |
| any(i == os.path.dirname(x["bin_path"]) for i in _SYS_BIN_PATH) |
| )) |
| for i in data[::-1]: |
| i["name"] = i["venv_name"] or i["path_name"] or i["version"] |
| if i["type"] == "venv": |
| i["system_data"] = env_map[i["system_path"]] |
| i["can_remove"] = False if i["project_name"] else i["can_remove"] |
| now_env = em.get_default_python_env() |
| if now_env: |
| now_env = env_map[now_env.bin_path] |
| now_env["can_set_default"] = False |
|
|
| return json_response(True, data={"env_list": data, "now_env": now_env}) |
|
|
| @staticmethod |
| def add_environment(get): |
| try: |
| add_type = get.add_type.strip() |
| path = get.path.strip() |
| except: |
| return json_response(False, '参数错误') |
| res = EnvironmentManager().add_python_env(add_type, path) |
| if isinstance(res, str): |
| return json_response(False, res) |
| return json_response(True, msg="添加成功") |
|
|
| @staticmethod |
| def remove_environment(get): |
| try: |
| path_data = get.path_data.strip() |
| except: |
| return json_response(False, '参数错误') |
| if not path_data: |
| return json_response(False, '参数错误') |
| em = EnvironmentManager() |
| if "," in path_data: |
| path_list = path_data.split(",") |
| res = em.multi_remove_env(*path_list) |
| else: |
| res = em.multi_remove_env(path_data) |
| return json_response(True, data=res) |
|
|
| @staticmethod |
| def set_environment_default(get): |
| try: |
| path = get.path.strip() |
| except: |
| return json_response(False, '参数错误') |
| if not path or path == "close": |
| path = "" |
| em = EnvironmentManager() |
| res = em.set_python2env(path) |
| if not path: |
| if not res: |
| return json_response(True, msg="关闭成功") |
| else: |
| return json_response(False, res) |
| else: |
| if not res: |
| return json_response(True, msg="设置成功") |
| else: |
| return json_response(False, res) |
|
|
| @staticmethod |
| def set_environment_ps(get): |
| try: |
| path = get.path.strip() |
| ps = get.ps.strip() |
| except: |
| return json_response(False, '参数错误') |
| if not path or not ps: |
| return json_response(False, '参数错误') |
| em = EnvironmentManager() |
| res = em.set_python_env_ps(path, ps) |
| if not res: |
| return json_response(True, msg="设置成功") |
| else: |
| return json_response(False, res) |
|
|
