36 lines
921 B
Python
36 lines
921 B
Python
import docker, sys, os
|
|
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 (
|
|
SANDBOX_CONTRAINER_NAME, CONTRAINER_NAME, SANDBOX_SERVER, DOCKER_SERVICE
|
|
)
|
|
|
|
from start import check_docker, check_process
|
|
|
|
try:
|
|
client = docker.from_env()
|
|
except:
|
|
client = None
|
|
|
|
|
|
def stop_main():
|
|
#
|
|
check_docker(client, SANDBOX_CONTRAINER_NAME, do_stop=True, )
|
|
check_process(f"port={SANDBOX_SERVER['port']}", do_stop=True)
|
|
check_process(f"port=5050", do_stop=True)
|
|
|
|
#
|
|
check_docker(client, CONTRAINER_NAME, do_stop=True, )
|
|
check_process("api.py", do_stop=True)
|
|
check_process("sdfile_api.py", do_stop=True)
|
|
check_process("llm_api.py", do_stop=True)
|
|
check_process("webui.py", do_stop=True)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
stop_main() |