File size: 5,051 Bytes
a757bd3 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 | 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)
|