33 lines
988 B
Python
33 lines
988 B
Python
|
import docker, sys, os, time, requests
|
||
|
|
|
||
|
|
from loguru import logger
|
||
|
|
|
||
|
|
src_dir = os.path.join(
|
||
|
|
os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
|
||
|
|
)
|
||
|
|
sys.path.append(src_dir)
|
||
|
|
|
||
|
|
from configs.server_config import CONTRAINER_NAME, SANDBOX_SERVER
|
||
|
|
|
||
|
|
|
||
|
|
if SANDBOX_SERVER["do_remote"]:
|
||
|
|
# stop and remove the container
|
||
|
|
client = docker.from_env()
|
||
|
|
for i in client.containers.list(all=True):
|
||
|
|
if i.name == CONTRAINER_NAME:
|
||
|
|
container = i
|
||
|
|
container.stop()
|
||
|
|
container.remove()
|
||
|
|
break
|
||
|
|
else:
|
||
|
|
# stop local
|
||
|
|
import psutil
|
||
|
|
for process in psutil.process_iter(["pid", "name", "cmdline"]):
|
||
|
|
# check process name contains "jupyter" and port=xx
|
||
|
|
if f"port={SANDBOX_SERVER['port']}" in str(process.info["cmdline"]).lower() and \
|
||
|
|
"jupyter" in process.info['name'].lower():
|
||
|
|
|
||
|
|
logger.warning(f"port={SANDBOX_SERVER['port']}, {process.info}")
|
||
|
|
# 关闭进程
|
||
|
|
process.terminate()
|