72 lines
3.0 KiB
Python
72 lines
3.0 KiB
Python
|
|
|||
|
|
import subprocess
|
|||
|
|
import time
|
|||
|
|
import os
|
|||
|
|
import psutil
|
|||
|
|
from loguru import logger
|
|||
|
|
|
|||
|
|
# 定义要启动的程序和命令
|
|||
|
|
binary_command = "cd /home/evan/Workplace/project_capture/build/; ./capture" # 替换为你的二进制程序路径
|
|||
|
|
conda_envs = [
|
|||
|
|
{"env": "conda_venv", "path": "/home/evan/Workplace/project_infer/lane_server/", "script": "python lane_infer_server.py"},
|
|||
|
|
{"env": "conda_venv", "path": "/home/evan/Workplace/project_infer/yolo_server/", "script": "python yolo_infer_server.py"},
|
|||
|
|
]
|
|||
|
|
|
|||
|
|
def kill_processes_by_name(process_name):
|
|||
|
|
"""
|
|||
|
|
终止所有具有指定名称的进程。
|
|||
|
|
"""
|
|||
|
|
for proc in psutil.process_iter(['pid', 'name', 'cmdline']):
|
|||
|
|
try:
|
|||
|
|
# 检查进程名是否包含给定的字符串
|
|||
|
|
if process_name.lower() in proc.info['name'].lower() or process_name.lower() in ' '.join(proc.info['cmdline']).lower():
|
|||
|
|
logger.warning(f"Killing {proc.info['name']} (PID: {proc.info['pid']})")
|
|||
|
|
proc.kill()
|
|||
|
|
except (psutil.NoSuchProcess, psutil.AccessDenied, psutil.ZombieProcess):
|
|||
|
|
pass
|
|||
|
|
|
|||
|
|
# 启动函数，用于启动并监控一个进程
|
|||
|
|
def start_and_monitor_process(cmd, env=None):
|
|||
|
|
process = subprocess.Popen(cmd, shell=True, env=env)
|
|||
|
|
# while process.poll() is None: # 等待进程结束
|
|||
|
|
# time.sleep(1) # 可以根据需要调整等待时间
|
|||
|
|
# return process.returncode # 返回进程退出码
|
|||
|
|
|
|||
|
|
# 主程序
|
|||
|
|
def main():
|
|||
|
|
# 启动二进制程序
|
|||
|
|
|
|||
|
|
logger.error("hi")
|
|||
|
|
|
|||
|
|
kill_processes_by_name("capture")
|
|||
|
|
binary_returncode = start_and_monitor_process(binary_command)
|
|||
|
|
|
|||
|
|
time.sleep(2)
|
|||
|
|
|
|||
|
|
logger.info("start python script")
|
|||
|
|
# 为每个 conda 环境启动 Python 脚本
|
|||
|
|
conda_procs = []
|
|||
|
|
for env_info in conda_envs:
|
|||
|
|
logger.info(f"start {env_info['script']}")
|
|||
|
|
# 激活 conda 环境并运行脚本（这里假设你使用的是 bash shell）
|
|||
|
|
activate_cmd = f"cd {env_info['path']}; {env_info['script']}"
|
|||
|
|
# activate_cmd = f"conda activate {env_info['env']}; {env_info['script']}"
|
|||
|
|
# 注意：在某些系统上，可能需要使用 `conda activate` 而不是 `source activate`
|
|||
|
|
# 并且可能需要设置 CONDA_PREFIX 等环境变量
|
|||
|
|
env = os.environ.copy()
|
|||
|
|
# 如果有需要传递给子 shell 的环境变量，可以在这里设置
|
|||
|
|
# 例如：env["CONDA_PREFIX"] = ...
|
|||
|
|
conda_proc = subprocess.Popen(["bash", "-c", activate_cmd], env=env)
|
|||
|
|
conda_procs.append(conda_proc)
|
|||
|
|
|
|||
|
|
# 等待所有 conda 环境中的 Python 脚本完成
|
|||
|
|
for proc in conda_procs:
|
|||
|
|
proc.wait() # 等待进程结束，但不会轮询检查
|
|||
|
|
|
|||
|
|
# 打印提示信息
|
|||
|
|
print(f"Binary program exited with code {binary_returncode}")
|
|||
|
|
for idx, proc in enumerate(conda_procs):
|
|||
|
|
print(f"{env_info['script']} script exited with code {proc.returncode}")
|
|||
|
|
|
|||
|
|
if __name__ == "__main__":
|
|||
|
|
main()
|