เก็บเอาต์พุตของ subprocess.Popen เปิดในสตริง


300

ฉันพยายามโทรออกด้วยระบบใน Python และเก็บเอาท์พุทเป็นสตริงที่ฉันสามารถจัดการได้ในโปรแกรม Python

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

ฉันลองทำบางสิ่งรวมถึงคำแนะนำที่นี่:

การดึงผลลัพธ์ของ subprocess.call ()

แต่ไม่มีโชค


3
มันเป็นการดีเสมอที่จะโพสต์รหัสที่คุณใช้จริงและการย้อนกลับหรือ bahaviour ที่ไม่คาดคิดสำหรับคำถามที่เป็นรูปธรรมเช่นนี้ ตัวอย่างเช่นฉันไม่ทราบว่าคุณพยายามทำอะไรเพื่อให้ได้ผลลัพธ์และฉันคิดว่าคุณไม่ได้เริ่มต้นด้วยการเริ่มต้น - คุณจะได้รับข้อผิดพลาดเกี่ยวกับการไม่ค้นหาไฟล์"ntpq -p"ซึ่งเป็นส่วนที่แตกต่างกันของ ปัญหามากกว่าที่คุณถาม
Mike Graham

คำตอบ:


467

ใน 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"]-


3
ในกรณีนี้ไพ ธ อนจะรอให้การเรียกระบบนี้เสร็จสิ้นหรือไม่ หรือจำเป็นต้องเรียกใช้ฟังก์ชัน wait / waitpid อย่างชัดเจนหรือไม่
NoneType

7
@NoneType Popen.communicateไม่ส่งคืนจนกว่ากระบวนการจะสิ้นสุด
Mike Graham

10
ถ้าคุณต้องการที่จะได้รับข้อผิดพลาดกระแสเพิ่ม stderr:p = subprocess.Popen(["ntpq", "-p"], stdout=subprocess.PIPE, stderr=subprocess.PIPE)
Timofey

82
ระวังsubprocess.check_output()จะส่งคืนวัตถุไม่ได้เป็นbytes strหากสิ่งที่คุณต้องการทำคือพิมพ์ผลลัพธ์นั่นจะไม่ทำให้เกิดความแตกต่าง แต่ถ้าคุณต้องการใช้วิธีเช่นmyString.split("\n")ผลลัพธ์คุณจะต้องถอดรหัสbytesวัตถุก่อนsubprocess.check_output(myParams).decode("utf-8")เช่น
TanguyP

17
การเพิ่มuniversal_newlines=Trueเป็นพารามิเตอร์ช่วยฉันใน Python 3 เพื่อรับวัตถุสตริง หาก universal_newlines เป็น True จะเปิดในโหมดข้อความพร้อมการเข้ารหัสเริ่มต้น มิฉะนั้นจะถูกเปิดเป็นสตรีมไบนารี
Jonathan Komar

38

สิ่งนี้ทำงานให้ฉันสำหรับการเปลี่ยนเส้นทาง stdout (stderr สามารถจัดการในทำนองเดียวกัน):

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

หากไม่ได้ผลโปรดระบุปัญหาที่คุณพบ


3
สิ่งนี้จะสร้างวัตถุแปลก ๆ \nเมื่อฉันแปลงเป็นสตริงก็หนีช่องว่างเช่น
Tomáš Zato - Reinstate Monica

1
โปรดทราบว่านี่ไม่ได้ตรวจสอบว่ากระบวนการย่อยทำงานอย่างถูกต้องหรือไม่ คุณอาจต้องการตรวจสอบว่าpipe.returncode == 0หลังจากบรรทัดสุดท้ายด้วย
thakis

เหตุผลที่งานนี้เป็นเพราะPopenผลตอบแทน tuple ของstdoutและstderrดังนั้นเมื่อคุณเข้าถึงคุณเพียงแค่โลภ[0] stdoutคุณสามารถทำได้text, err = pipe.communicate()และจากนั้นtextจะมีสิ่งที่คุณคาดหวัง
Jona

23

สมมติว่าpwdเป็นเพียงตัวอย่างนี่คือวิธีที่คุณทำได้:

import subprocess

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

ดูเอกสารประกอบย่อยกระบวนการสำหรับตัวอย่างอื่นและข้อมูลเพิ่มเติม


22

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

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 ถ้าเชลล์เป็น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()

น่ากลัวนี่คือสิ่งที่ฉันกำลังมองหา
kRazzy R

14

สำหรับ Python 2.7+ คำตอบที่ใช้คือ subprocess.check_output()

คุณควรจดบันทึกการจัดการอาร์กิวเมนต์เมื่อเรียกใช้กระบวนการย่อยเนื่องจากอาจทำให้สับสนเล็กน้อย ....

หาก args เป็นเพียงคำสั่งเดียวโดยไม่มี args ของตัวเอง (หรือคุณได้shell=Trueตั้งค่า) มันอาจเป็นสตริง มิฉะนั้นจะต้องเป็นรายการ

ตัวอย่างเช่น ... เพื่อเรียกใช้lsคำสั่งนี่เป็นเรื่องปกติ:

from subprocess import check_call
check_call('ls')

ดังนั้นนี่คือ:

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

อย่างไรก็ตามหากคุณต้องการส่ง args ไปยังคำสั่ง 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'))

ห้าปีต่อมาคำถามนี้ยังคงได้รับความรักมากมาย ขอบคุณสำหรับการอัปเดต 2.7+, Corey!
มาร์ค

11

มันทำงานได้อย่างสมบูรณ์แบบสำหรับฉัน:

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 

9

นี่คือที่สมบูรณ์แบบสำหรับฉัน คุณจะได้รับรหัสส่งคืน stdout และ stderr ใน tuple

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]

6

คำตอบที่ได้รับการยอมรับนั้นยังดีอยู่เพียงแค่พูดถึงคุณสมบัติใหม่ ๆ ตั้งแต่หลาม 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 docs :

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

4
 import os   
 list = os.popen('pwd').read()

ในกรณีนี้คุณจะมีองค์ประกอบเดียวในรายการ


7
os.popenถูกเลิกใช้ในความโปรดปรานของsubprocessโมดูล
Mike Graham

3
สิ่งนี้มีประโยชน์มากสำหรับผู้ดูแลระบบในกล่องเก่าโดยใช้ 2.2.X series ของ Python
Neil McF

4

ฉันเขียนฟังก์ชันเล็กน้อยตามคำตอบอื่น ๆ ที่นี่:

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))

4

ในหลาม 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'

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

นี่เป็นวิธีแก้ปัญหาหนึ่งบรรทัด


0

ต่อไปนี้รวบรวม 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)

0

สำหรับ 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")

0

ใช้ckeck_outputวิธีการsubprocess

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

ในที่สุดการแยกสตริง

for line in res.splitlines():

หวังว่าจะช่วยให้การเขียนโปรแกรมมีความสุข

โดยการใช้ไซต์ของเรา หมายความว่าคุณได้อ่านและทำความเข้าใจนโยบายคุกกี้และนโยบายความเป็นส่วนตัวของเราแล้ว
Licensed under cc by-sa 3.0 with attribution required.