Сохранить выходные данные подпроцесса. Открыть вызов в строке
Я пытаюсь сделать системный вызов в 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")