Сохранить выходные данные подпроцесса. Открыть вызов в строке

Я пытаюсь сделать системный вызов в Python и сохранить вывод в строку, которой я могу манипулировать в программе Python.

#!/usr/bin/python
import subprocess
p2 = subprocess.Popen("ntpq -p")

Я попробовал несколько вещей, включая некоторые из предложений здесь:

Получение выходных данных subprocess.call()

но без удачи.

9 ответов

Решение

В Python 2.7 или Python 3

Вместо того чтобы делать Popen объект, вы можете использовать subprocess.check_output() функция для хранения вывода команды в строку:

from subprocess import check_output
out = check_output(["ntpq", "-p"])

В Python 2.4-2.6

Использовать communicate метод.

import subprocess
p = subprocess.Popen(["ntpq", "-p"], stdout=subprocess.PIPE)
out, err = p.communicate()

out это то, что вы хотите.

Важное замечание о других ответах

Обратите внимание, как я прошел в команде. "ntpq -p" пример поднимает другое дело. поскольку Popen не вызывает оболочку, вы должны использовать список команд и опций: ["ntpq", "-p"],

В Python 3.7 новый аргумент ключевого слова capture_output был введен для subprocess.run. Включение короткого и простого:

import subprocess

p = subprocess.run("echo 'hello world!'", capture_output=True, shell=True, encoding="utf8")
assert p.stdout == 'hello world!\n'

Это сработало для меня для перенаправления stdout (stderr может обрабатываться аналогично):

from subprocess import Popen, PIPE
pipe = Popen(path, stdout=PIPE)
text = pipe.communicate()[0]

Если это не работает для вас, пожалуйста, укажите точно, какая у вас проблема.

subprocess.Popen: http://docs.python.org/2/library/subprocess.html

import subprocess

command = "ntpq -p"  # the shell command
process = subprocess.Popen(command, stdout=subprocess.PIPE, stderr=None, shell=True)

#Launch the shell command:
output = process.communicate()

print output[0]

В конструкторе Popen, если shell равен True, вы должны передавать команду как строку, а не как последовательность. В противном случае просто разбейте команду на список:

command = ["ntpq", "-p"]  # the shell command
process = subprocess.Popen(command, stdout=subprocess.PIPE, stderr=None)

Если вам нужно прочитать также стандартную ошибку, в инициализации Popen вы можете установить для stderr значение subprocess.PIPE или subprocess.STDOUT:

import subprocess

command = "ntpq -p"  # the shell command
process = subprocess.Popen(command, stdout=subprocess.PIPE, stderr=subprocess.PIPE, shell=True)

#Launch the shell command:
output, error = process.communicate()

При условии, что pwd это просто пример, вот как вы можете это сделать:

import subprocess

p = subprocess.Popen("pwd", stdout=subprocess.PIPE)
result = p.communicate()[0]
print result

См. Документацию подпроцесса для другого примера и дополнительной информации.

для Python 2.7+ идиоматический ответ заключается в использовании subprocess.check_output()

Вы также должны обратить внимание на обработку аргументов при вызове подпроцесса, так как это может немного сбить с толку....

Если args - это просто одна команда без собственных аргументов (или у вас есть shell=True установить), это может быть строка. В противном случае это должен быть список.

например... чтобы вызвать ls команда, это хорошо:

from subprocess import check_call
check_call('ls')

так это:

from subprocess import check_call
check_call(['ls',])

однако, если вы хотите передать некоторые аргументы команде shell, вы не можете сделать это:

from subprocess import check_call
check_call('ls -al')

вместо этого вы должны передать его в виде списка:

from subprocess import check_call
check_call(['ls', '-al'])

shlex.split() Функция иногда может быть полезна для разбиения строки на оболочечный синтаксис перед созданием подпроцессов... вот так:

from subprocess import check_call
import shlex
check_call(shlex.split('ls -al'))

Это прекрасно работает для меня:

import subprocess
try:
    #prints results and merges stdout and std
    result = subprocess.check_output("echo %USERNAME%", stderr=subprocess.STDOUT, shell=True)
    print result
    #causes error and merges stdout and stderr
    result = subprocess.check_output("copy testfds", stderr=subprocess.STDOUT, shell=True)
except subprocess.CalledProcessError, ex: # error code <> 0 
    print "--------error------"
    print ex.cmd
    print ex.message
    print ex.returncode
    print ex.output # contains stdout and stderr together 

Это было идеально для меня. Вы получите код возврата, stdout и stderr в кортеже.

from subprocess import Popen, PIPE

def console(cmd):
    p = Popen(cmd, shell=True, stdout=PIPE)
    out, err = p.communicate()
    return (p.returncode, out, err)

Например:

result = console('ls -l')
print 'returncode: %s' % result[0]
print 'output: %s' % result[1]
print 'error: %s' % result[2]

Принятый ответ по-прежнему хорош, всего несколько замечаний о новых функциях. Начиная с python 3.6, вы можете обрабатывать кодирование непосредственно вcheck_outputсм. документацию. Теперь это возвращает строковый объект:

import subprocess 
out = subprocess.check_output(["ls", "-l"], encoding="utf-8")

В python 3.7 параметр capture_outputбыл добавлен в subprocess.run(), который выполняет за нас часть обработки Popen/PIPE, см. документацию по python:

import subprocess 
p2 = subprocess.run(["ls", "-l"], capture_output=True, encoding="utf-8")
p2.stdout

Я написал небольшую функцию на основе других ответов здесь:

def pexec(*args):
    return subprocess.Popen(args, stdout=subprocess.PIPE).communicate()[0].rstrip()

Использование:

changeset = pexec('hg','id','--id')
branch = pexec('hg','id','--branch')
revnum = pexec('hg','id','--num')
print('%s : %s (%s)' % (revnum, changeset, branch))
 import os   
 list = os.popen('pwd').read()

В этом случае у вас будет только один элемент в списке.

import subprocess
output = str(subprocess.Popen("ntpq -p",shell = True,stdout = subprocess.PIPE, 
stderr = subprocess.STDOUT).communicate()[0])

Это однолинейное решение

Использовать ckeck_output метод subprocess

import subprocess
address = 192.168.x.x
res = subprocess.check_output(['ping', address, '-c', '3'])

Наконец, проанализируйте строку

for line in res.splitlines():

Надеюсь, это поможет, удачного кодирования

Следующее захватывает stdout и stderr процесса в одной переменной. Это Python 2 и 3 совместимы:

from subprocess import check_output, CalledProcessError, STDOUT

command = ["ls", "-l"]
try:
    output = check_output(command, stderr=STDOUT).decode()
    success = True 
except CalledProcessError as e:
    output = e.output.decode()
    success = False

Если ваша команда является строкой, а не массивом, добавьте к ней префикс:

import shlex
command = shlex.split(command)

Для python 3.5 я добавил функцию на основе предыдущего ответа. Лог можно удалить, подумал, что неплохо иметь

import shlex
from subprocess import check_output, CalledProcessError, STDOUT


def cmdline(command):
    log("cmdline:{}".format(command))
    cmdArr = shlex.split(command)
    try:
        output = check_output(cmdArr,  stderr=STDOUT).decode()
        log("Success:{}".format(output))
    except (CalledProcessError) as e:
        output = e.output.decode()
        log("Fail:{}".format(output))
    except (Exception) as e:
        output = str(e);
        log("Fail:{}".format(e))
    return str(output)


def log(msg):
    msg = str(msg)
    d_date = datetime.datetime.now()
    now = str(d_date.strftime("%Y-%m-%d %H:%M:%S"))
    print(now + " " + msg)
    if ("LOG_FILE" in globals()):
        with open(LOG_FILE, "a") as myfile:
            myfile.write(now + " " + msg + "\n")
Другие вопросы по тегам