36 lines
1.0 KiB
Python
36 lines
1.0 KiB
Python
#!/usr/bin/env python3
|
||
# -*- coding: utf-8 -*-
|
||
"""
|
||
统一启动 gRPC 客户端脚本
|
||
自动设置 PYTHONPATH，让 generated 下的 cmvr 包可用
|
||
用法：
|
||
python run_client.py get_joint_state_client
|
||
python run_client.py movej_client
|
||
"""
|
||
|
||
import os
|
||
import sys
|
||
import subprocess
|
||
|
||
PROJECT_ROOT = os.path.dirname(os.path.abspath(__file__))
|
||
GENERATED_DIR = os.path.join(PROJECT_ROOT, "generated")
|
||
CLIENTS_DIR = os.path.join(PROJECT_ROOT, "clients")
|
||
|
||
# 添加 generated 到 PYTHONPATH
|
||
sys.path.insert(0, GENERATED_DIR)
|
||
os.environ["PYTHONPATH"] = GENERATED_DIR + os.pathsep + os.environ.get("PYTHONPATH", "")
|
||
|
||
if len(sys.argv) < 2:
|
||
print("请指定要运行的客户端脚本名，例如：get_joint_state_client")
|
||
sys.exit(1)
|
||
|
||
client_name = sys.argv[1]
|
||
client_script = os.path.join(CLIENTS_DIR, f"{client_name}.py")
|
||
|
||
if not os.path.isfile(client_script):
|
||
print(f"客户端脚本不存在: {client_script}")
|
||
sys.exit(1)
|
||
|
||
# 支持传递额外参数给客户端
|
||
subprocess.run([sys.executable, client_script] + sys.argv[2:])
|