从python内部运行交互式命令

内容来源于 Stack Overflow,并遵循CC BY-SA 3.0许可协议进行翻译与使用

  • 回答 (2)
  • 关注 (0)
  • 查看 (196)

我有一个脚本,我想在python(2.6.5)中运行它,它遵循以下逻辑:

  • 提示用户输入密码。看起来(“输入密码:”)(*注意:输入不回显)
  • 输出无关信息
  • 提示用户响应(“Blah Blah filename.txt blah blah(Y/N)?:”)

最后一个提示符行包含我需要解析的文本(filename.txt)。提供的响应并不重要(只要我能够解析行,程序实际上可以退出这里而不提供响应)。POpen行在下面

p = subprocess.Popen("cmd", shell=True, stdout=subprocess.PIPE, 
stderr=subprocess.STDOUT, stdin=subprocess.PIPE)
  • 当我在stdout上调用read()或readline()时,提示符是打印机到屏幕并挂起。
  • 如果我为stdin调用写(“Password\n”),提示将被写入屏幕并挂起。写()中的文本不是写的(我的光标没有移动新行)。
  • 如果我调用p.Communications(“Password\n”),则与write()的行为相同。

我在这里寻找一些关于输入到stdin的最佳方法的一些想法,以及如果您感觉很慷慨的话,如何解析输出中的最后一行,尽管我可能最终会弄清楚这一点。谢谢!

提问于
用户回答回答于

使用线程,对于简单的任务来说,这可能有点过分。相反,os.spawnvpe可以使用。它将生成脚本外壳作为一个进程。您将能够与脚本交互通信。在本例中,我将密码作为参数传递,显然这不是一个好主意。

import os
import sys
from getpass import unix_getpass

def cmd(cmd):
    cmd = cmd.split()
    code = os.spawnvpe(os.P_WAIT, cmd[0], cmd, os.environ)
    if code == 127:
        sys.stderr.write('{0}: command not found\n'.format(cmd[0]))
    return code

password = unix_getpass('Password: ')
cmd_run = './run.sh --password {0}'.format(password)
cmd(cmd_run)

pattern = raw_input('Pattern: ')
lines = []
with open('filename.txt', 'r') as fd:
    for line in fd:
        if pattern in line:
            lines.append(line)

# manipulate lines
用户回答回答于

从用户那里获取值,您可以始终使用原始_输入()内置以获取响应,对于密码,请使用getpass模块从您的用户获得不回显密码。然后,您可以解析这些响应并将它们写入您的子流程的stdin。

最后,我做了一些类似于以下的事情:

import sys
import subprocess
from threading  import Thread

try:
    from Queue import Queue, Empty
except ImportError:
    from queue import Queue, Empty  # python 3.x


def enqueue_output(out, queue):
    for line in iter(out.readline, b''):
        queue.put(line)
    out.close()


def getOutput(outQueue):
    outStr = ''
    try:
        while True: #Adds output from the Queue until it is empty
            outStr+=outQueue.get_nowait()

    except Empty:
        return outStr

p = subprocess.Popen("cmd", stdin=subprocess.PIPE, stdout=subprocess.PIPE, stderr=subprocess.PIPE, shell=False, universal_newlines=True)

outQueue = Queue()
errQueue = Queue()

outThread = Thread(target=enqueue_output, args=(p.stdout, outQueue))
errThread = Thread(target=enqueue_output, args=(p.stderr, errQueue))

outThread.daemon = True
errThread.daemon = True

outThread.start()
errThread.start()

try:
    someInput = raw_input("Input: ")
except NameError:
    someInput = input("Input: ")

p.stdin.write(someInput)
errors = getOutput(errQueue)
output = getOutput(outQueue)

一旦队列建立并线程启动,您就可以循环从用户获得输入、从进程中获取错误和输出,并将它们处理并显示给用户。

扫码关注云+社区

领取腾讯云代金券