python - 在python中嵌入bash

Question

我正在编写 Python 脚本，但时间不多了。我需要在 bash 中做一些我非常熟悉的事情，所以我只是想知道如何将一些 bash 行嵌入到 Python 脚本中。

谢谢

score 33 · Accepted Answer

理想的方法：

def run_script(script, stdin=None):
    """Returns (stdout, stderr), raises error on non-zero return code"""
    import subprocess
    # Note: by using a list here (['bash', ...]) you avoid quoting issues, as the 
    # arguments are passed in exactly this order (spaces, quotes, and newlines won't
    # cause problems):
    proc = subprocess.Popen(['bash', '-c', script],
        stdout=subprocess.PIPE, stderr=subprocess.PIPE,
        stdin=subprocess.PIPE)
    stdout, stderr = proc.communicate()
    if proc.returncode:
        raise ScriptException(proc.returncode, stdout, stderr, script)
    return stdout, stderr

class ScriptException(Exception):
    def __init__(self, returncode, stdout, stderr, script):
        self.returncode = returncode
        self.stdout = stdout
        self.stderr = stderr
        Exception().__init__('Error in script')

你也可以添加一个不错的__str__方法ScriptException（你肯定需要它来调试你的脚本）——但我把它留给读者。

如果您不使用stdout=subprocess.PIPEetc，脚本将直接附加到控制台。例如，如果您有来自 ssh 的密码提示，这将非常方便。因此，您可能需要添加标志来控制是否要捕获标准输出、标准错误和标准输入。

score 14 · Accepted Answer

14

如果要调用系统命令，请使用subprocess模块。

于 2010-04-16T09:49:17.163 回答

score 6 · Accepted Answer

假设主机系统支持该命令：

import os
os.system('command')

如果您有一个长命令或一组命令。你可以使用变量。例如：

# this simple line will capture column five of file.log
# and then removed blanklines, and gives output in filtered_content.txt.

import os

filter = "cat file.log | awk '{print $5}'| sed '/^$/d' > filtered_content.txt"

os.system(filter)

score 6 · Accepted Answer

是

import os
os.system ("bash -c 'echo $0'")

会为你做吗？

编辑：关于可读性

是的，当然，你可以让它更具可读性

import os
script = """
echo $0
ls -l
echo done
"""
os.system("bash -c '%s'" % script)

EDIT2：关于宏，据我所知，没有python没有走得那么远，但介于

import os
def sh(script):
    os.system("bash -c '%s'" % script)

sh("echo $0")
sh("ls -l")
sh("echo done")

和前面的例子一样，你基本上得到了你想要的（但你必须允许一些辩证的限制）

score 3 · Accepted Answer

当 bash 命令很简单且没有括号、逗号和引号时，subprocess 和 os.system() 可以正常工作。嵌入复杂 bash 参数的简单方法是在 python 脚本的末尾添加带有唯一字符串注释的 bash 脚本，并使用简单的 os.system() 命令尾随并转换为 bash 文件。

#!/usr/bin/python
## name this file  "file.py"
import os
def get_xred(xx,yy,zz):
    xred=[]
####gaur###
    xred.append([     zz[9] ,  zz[19] ,  zz[29]     ])
    xred.append([     zz[9] ,  xx[9]  ,  yy[9]      ])
    xred.append([     zz[10],  zz[20] ,  zz[30]     ])
    xred.append([     zz[10],  xx[10] ,  yy[10]     ])
###nitai###
    xred=np.array(xred)
    return xred
## following 3 lines executes last 6 lines of this file.
os.system("tail -n 6 file.py >tmpfile1")
os.system("sed 's/###123//g' tmpfile1>tmpfile2")
os.system("bash tmpfile2")
###### Here ###123 is a unique string to be removed
###123#!/bin/sh
###123awk '/###gaur/{flag=1;next}/###nitai/{flag=0} flag{print}' file.py >tmp1
###123cat tmp1 | awk '{gsub("xred.append\\(\\[","");gsub("\\]\\)","");print}' >tmp2
###123awk 'NF >0' tmp2 > tmp3
###123sed '$d' tmp3 |sed '$d' | sed '$d' >rotation ; rm tmp*

score 1 · Accepted Answer

我创建 Sultan 正是为了解决您想要做的事情。它没有任何外部依赖项，并尝试尽可能轻，并为 Bash 提供 Pythonic 接口。

https://github.com/aeroxis/sultan

score 0 · Accepted Answer

还有一个commands模块可以更好地控制输出： https ://docs.python.org/2/library/commands.html

score 0 · Accepted Answer

如前所述，您可以使用 os.system(); 它又快又脏，但它易于使用并且适用于大多数情况。它实际上是对 C system() 函数的映射。

http://docs.python.org/2/library/os.html#os.system

http://www.cplusplus.com/reference/cstdlib/system/

score 0 · Accepted Answer

@Ian Bicking的答案很有用，但它只允许我们运行脚本。相反，我们可以提出一个更灵活的代码，我们也可以在其中运行命令。我和他的方法不同。

#!/usr/bin/env python3

from subprocess import Popen, PIPE


class BashCommandsException(Exception):
    def __init__(self, returncode, output, error_msg):
        self.returncode = returncode
        self.output = output
        self.error_msg = error_msg
        Exception.__init__('Error in executed command')


def popen_communicate(cmd, stdout_file=None):
    """Acts similir to lib.run(cmd) but also returns the output message captures on
    during the run stdout_file is not None in case of nohup process writes its
    results into a file
    """
    cmd = list(map(str, cmd))  # all items should be string
    if stdout_file is None:
        p = Popen(cmd, stdout=PIPE, stderr=PIPE)
    else:
        with open(stdout_file, "w") as outfile:
            # output written into file, error will be returned
            p = Popen(cmd, stdout=outfile, stderr=PIPE, universal_newlines=True)
            output, error = p.communicate()
            p.wait()
            return p, output, error

    output, error = p.communicate()
    output = output.strip().decode("utf-8")
    error = error.decode("utf-8")
    return p, output, error


def run(cmd):
    log_file = "/tmp/log.txt"
    # if log_file is not provided returned output will be stored in output
    p, output, error_msg = popen_communicate(cmd, log_file)
    if p.returncode != 0:
        raise BashCommandsException(p.returncode, output, error_msg, str(cmd))
    return output


if __name__ == "__main__":
    # This could be any command you want to execute as you were in bash
    cmd = ["bash", "script_to_run.sh"]
    try:
        run(cmd)
    except Exception as e:
        print(e)

python - 在python中嵌入bash

9 回答 9

Related

Reference