panyy
/
UnisKB
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

refactor: delete .SANDBOX_BANNED_HOSTS file if SANDBOX_PYTHON_BANNED_KEYWORDS is empty/

v3.2
liqiang-fit2cloud 2025-11-10 09:39:54 +08:00
parent da73582093
commit ad1742fd0f
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
apps/common/utils/tool_code.py
View File

@ -28,14 +28,14 @@ class ToolExecutor:
os.system(f"chown -R {self.user}:root {self.sandbox_path}") os.system(f"chown -R {self.user}:root {self.sandbox_path}")
self.banned_keywords = CONFIG.get("SANDBOX_PYTHON_BANNED_KEYWORDS", 'nothing_is_banned').split(','); self.banned_keywords = CONFIG.get("SANDBOX_PYTHON_BANNED_KEYWORDS", 'nothing_is_banned').split(',');
try: try:
banned_hosts_file_path = f'{self.sandbox_path}/.SANDBOX_BANNED_HOSTS'
if os.path.exists(banned_hosts_file_path):
os.remove(banned_hosts_file_path)
banned_hosts = CONFIG.get("SANDBOX_PYTHON_BANNED_HOSTS", '').strip() banned_hosts = CONFIG.get("SANDBOX_PYTHON_BANNED_HOSTS", '').strip()
if banned_hosts: if banned_hosts:
hostname = socket.gethostname() hostname = socket.gethostname()
local_ip = socket.gethostbyname(hostname) local_ip = socket.gethostbyname(hostname)
banned_hosts = f"{banned_hosts},{hostname},{local_ip}" banned_hosts = f"{banned_hosts},{hostname},{local_ip}"
banned_hosts_file_path = f'{self.sandbox_path}/.SANDBOX_BANNED_HOSTS'
if os.path.exists(banned_hosts_file_path):
os.remove(banned_hosts_file_path)
with open(banned_hosts_file_path, "w") as f: with open(banned_hosts_file_path, "w") as f:
f.write(banned_hosts) f.write(banned_hosts)
os.chmod(banned_hosts_file_path, 0o644) os.chmod(banned_hosts_file_path, 0o644)