72 lines
2.1 KiB
Python
72 lines
2.1 KiB
Python
import subprocess
|
||
from typing import Union, Tuple
|
||
from .system_utils import SystemUtils
|
||
|
||
|
||
class CommandUtils:
|
||
"""命令执行工具类"""
|
||
|
||
@staticmethod
|
||
def execute(command: Union[str, list], timeout: int = None) -> Tuple[int, str, str]:
|
||
"""
|
||
执行系统命令并返回结果（兼容Windows/macOS/Linux）
|
||
|
||
参数:
|
||
command: 要执行的命令，可以是字符串或列表
|
||
timeout: 超时时间（秒）
|
||
|
||
返回:
|
||
元组: (return_code, stdout, stderr)
|
||
"""
|
||
# 预处理命令
|
||
processed_cmd = CommandUtils._prepare_command(command)
|
||
|
||
try:
|
||
# 执行命令
|
||
result = subprocess.run(
|
||
processed_cmd,
|
||
shell=isinstance(command, str),
|
||
stdout=subprocess.PIPE,
|
||
stderr=subprocess.PIPE,
|
||
universal_newlines=True,
|
||
timeout=timeout
|
||
)
|
||
return result.returncode, result.stdout, result.stderr
|
||
except subprocess.TimeoutExpired:
|
||
return -1, "", "Command timed out"
|
||
except Exception as e:
|
||
return -1, "", str(e)
|
||
|
||
@staticmethod
|
||
def _prepare_command(command: Union[str, list]) -> Union[str, list]:
|
||
"""预处理命令"""
|
||
if SystemUtils.is_windows() and isinstance(command, str):
|
||
return ['cmd', '/c'] + command.split()
|
||
return command
|
||
|
||
@staticmethod
|
||
def execute_with_real_time_output(command: Union[str, list], timeout: int = None) -> int:
|
||
"""
|
||
执行命令并实时输出（不捕获输出，直接打印到控制台）
|
||
|
||
参数:
|
||
command: 要执行的命令
|
||
timeout: 超时时间（秒）
|
||
|
||
返回:
|
||
返回状态码
|
||
"""
|
||
processed_cmd = CommandUtils._prepare_command(command)
|
||
|
||
try:
|
||
result = subprocess.run(
|
||
processed_cmd,
|
||
shell=isinstance(command, str),
|
||
timeout=timeout
|
||
)
|
||
return result.returncode
|
||
except subprocess.TimeoutExpired:
|
||
return -1
|
||
except Exception as e:
|
||
return -1
|