Python如何执行shell脚本

Python如何执行shell脚本

自从出了Pyhon3.5之后，os模块下的system()+os.popen()，基本被废弃了。
因此如下只介绍2种方式：
一、使用commands模块，有三个方法可以使用：

（1）commands.getstatusoutput(cmd)，其以字符串的形式返回的是输出结果和状态码，即（status,output）。

（2）commands.getoutput(cmd)，返回cmd的输出结果。

（3）commands.getstatus(file)，返回ls -l file的执行结果字符串，调用了getoutput，不建议使用此方法

二、subprocess模块 : (这个是常用方法)
允许创建很多子进程，创建的时候能指定子进程和子进程的输入、输出、错误输出管道，执行后能获取输出结果和执行状态。

（1）subprocess.run()：python3.5中新增的函数， 执行指定的命令， 等待命令执行完成后返回一个包含执行结果的CompletedProcess类的实例。

语法说明：subprocess.run(args, *, stdin=None, input=None, stdout=None, stderr=None, shell=False, timeout=None, check=False, universal_newlines=False)

（2）subprocess.call()：执行指定的命令， 返回命令执行状态， 功能类似os.system（命令）。
语法说明：subprocess.call(args, *, stdin=None, stdout=None, stderr=None, shell=False, timeout=None)
　
（3）subprocess.check_call()：python2.5中新增的函数, 执行指定的命令, 如果执行成功则返回状态码， 否则抛出异常。
语法说明： subprocess.check_call(args, *, stdin=None, stdout=None, stderr=None, shell=False, timeout=None)

# coding=utf-8

import shlex
import datetime
import subprocess
import time
  
def execute_command(cmdstring, cwd=None, timeout=None, shell=False):
  """执行一个SHELL命令
      封装了subprocess的Popen方法, 支持超时判断，支持读取stdout和stderr
      参数:
    cwd: 运行命令时更改路径，如果被设定，子进程会直接先更改当前路径到cwd
    timeout: 超时时间，秒，支持小数，精度0.1秒
    shell: 是否通过shell运行
  Returns: return_code
  Raises: Exception: 执行超时
  """
  if shell:
    cmdstring_list = cmdstring
  else:
    cmdstring_list = shlex.split(cmdstring)
  if timeout:
    end_time = datetime.datetime.now() + datetime.timedelta(seconds=timeout)
   
  #没有指定标准输出和错误输出的管道，因此会打印到屏幕上；
  sub = subprocess.Popen(cmdstring_list, cwd=cwd, stdin=subprocess.PIPE,shell=shell,bufsize=4096)
   
  #subprocess.poll()方法：检查子进程是否结束了，如果结束了，设定并返回码，放在subprocess.returncode变量中 
  while sub.poll() is None:
    time.sleep(0.1)
    if timeout:
      if end_time <= datetime.datetime.now():
        raise Exception("Timeout：%s"%cmdstring)
       
  return str(sub.returncode)
  
if __name__=="__main__":
  print execute_command("ls")

也可以在Popen中指定stdin和stdout为一个变量，这样就能直接接收该输出变量值。

总结

在python中执行SHELL有时候也是很必须的，比如使用Python的线程机制启动不同的shell进程，目前subprocess是Python官方推荐的方法，其支持的功能也是最多的，推荐大家使用。