def exec_shell(name ,cmd_str ,timeout=5 * 60): ret_values = [128] start_time = time.time() result = subprocess.Popen(cmd_str, shell=True) isrun = True while result.poll() is None and isrun: now = time.time() if now - start_time >= timeout and result.poll() is None: break time.sleep(1) ret = os.system('ssh -o ConnectTimeout=3 '+ name +' \"cd\"') if ret != 0: logging.info('%s Connect Timeout, ERROR: %d' % (name,ret)) isrun = False if result.poll() is None: result.terminate() # os.killpg(os.getpgid(result.pid), signal.SIGTERM) # os.kill(result.pid, signal.SIGKILL) kill_process(result.pid) ret_values[0] = 65535 else: ret_values[0] = result.returncode return ret_values[0]代码问题
时间: 2023-12-17 21:04:45 浏览: 229
这段代码看起来是用于执行 shell 命令的函数,但是其中有一些问题:
1. 在函数内部执行了一个无限循环,如果命令执行时间超过了指定的 timeout 时间,就会退出循环。但是这个循环会一直占用 CPU 资源,不太合理。可以使用 Python 的 `subprocess.call()` 函数来执行命令,并且可以通过 `timeout` 参数设置超时时间。
2. 在循环内部还执行了一个 `os.system()` 命令,这个命令会新开一个进程去执行 `ssh` 命令,这样会增加系统负担。可以使用 `paramiko` 库来实现 SSH 连接。
3. 函数最后返回的 `ret_values[0]` 只有一个元素,而且这个元素在函数内部已经被修改了,使用一个变量就可以了。
下面是修改后的代码:
```python
import subprocess
import time
import paramiko
def exec_shell(name, cmd_str, timeout=5*60):
start_time = time.time()
try:
result = subprocess.run(cmd_str, shell=True, check=True, timeout=timeout)
ret = result.returncode
except subprocess.TimeoutExpired:
ret = 65535
except subprocess.CalledProcessError as e:
ret = e.returncode
except Exception as e:
ret = 128
# 这里可以根据返回值做相应的处理,比如记录日志等
return ret
def ssh_connect(name):
ssh = paramiko.SSHClient()
ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
try:
ssh.connect(name, timeout=3)
return True
except:
return False
def exec_ssh_cmd(name, cmd_str, timeout=5*60):
start_time = time.time()
while True:
if ssh_connect(name):
break
now = time.time()
if now - start_time >= timeout:
logging.info('%s Connect Timeout' % name)
return 65535
time.sleep(1)
stdin, stdout, stderr = ssh.exec_command(cmd_str, timeout=timeout)
ret = stdout.channel.recv_exit_status()
# 这里可以根据返回值做相应的处理,比如记录日志等
return ret
```
其中 `exec_shell()` 函数使用了 Python 内置的 `subprocess.run()` 函数来执行 shell 命令,可以设置超时时间和检查命令是否执行成功的选项,这样就不需要使用循环了。
另外,我还添加了一个 `ssh_connect()` 函数来判断是否能够连接到远程服务器,以及一个 `exec_ssh_cmd()` 函数来执行 SSH 命令。这两个函数使用了 `paramiko` 库来实现 SSH 连接和命令执行,比较方便和安全。
阅读全文