【问题标题】：Python: Respond to Command Line PromptsPython：响应命令行提示
【发布时间】：2013-03-23 14:13:22
【问题描述】：

我正在尝试使用 Python 通过命令行与另一个程序进行交互。我遇到的主要问题是一个具有多个后续提示的特定呼叫。最初，命令行调用询问项目的名称，然后继续询问我是否要查看项目的任何子文件夹。我需要按顺序对每一个回答y/n，不幸的是，每一个的答案都不是y或n。此外，如果不阅读各个提示，我无法知道问题的答案，因此我无法一次发送一个“y”或“n”块。

这是命令行调用：

si viewproject

输入命令后，命令行提示：

输入项目名称：

一个示例响应是：

输入项目名称：c:/test.pj

进入项目后提示如下：

是否要递归到子项目 test_subprj.pj 中？ [ynYN](n)

此时我需要用 y 或 n 响应，具体取决于我是否需要该子项目。同样，对这个问题的回答取决于子项目。我需要能够阅读此提示中的子项目，以便用“y”或“n”来响应它

目前我需要手动输入项目和每个y和n。我的目标是使用 Python 自动化这个过程。

有没有办法自动响应这些命令行提示？

当前进展

子流程策略

 project_path = "c:/test.pj"

 with Popen(["si", "viewproject", "--project=" + project_path], 
             stdin=PIPE, stdout=PIPE, universal_newlines=True) as p:
     for line in p.stdout: 
         if line.startswith("Do you want"):
             answer = 'n'
         else:
             continue # skip it
         print(answer, file=p.stdin) # provide answer
         p.stdin.flush()

这个方法挂在 with Popen 语句之后。它永远不会出错，但永远不会进入或退出 for 语句并且永远不会完成。目前我将所有答案默认为“n”，但稍后将替换为逻辑。

Winpexpect 策略

 import re
 import sys
 from functools import partial
 import winpexpect

 project_path = "c:/test.pj"

 p = winpexpect.winspawn('si viewproject --project=' + project_path)
 p.logfile = sys.stdout
 patterns = [re.compile('ynYN'), winpexpect.EOF]

 for found in iter(partial(p.expect, patterns), 1): # until EOF
     if found == 0:
         answer = 'n'
         p.sendline(answer)

返回以下错误信息：

 Traceback (most recent call last):
   File "C:\Python33\lib\site-packages\winpexpect-1.5-py3.3.egg\winpexpect.py", line 541, in read_nonblocking
     handle, status, data = self.child_output.get(timeout=timeout)
   File "C:\Python33\lib\queue.py", line 175, in get
     raise Empty
 queue.Empty

 During handling of the above exception, another exception occurred:

 Traceback (most recent call last):
   File "C:\Python33\lib\site-packages\winpexpect-1.5-py3.3.egg\pexpect.py", line 1378, in expect_loop
     c = self.read_nonblocking (self.maxread, timeout)
   File "C:\Python33\lib\site-packages\winpexpect-1.5-py3.3.egg\winpexpect.py", line 543, in read_nonblocking
     raise TIMEOUT('Timeout exceeded in read_nonblocking().')
 pexpect.TIMEOUT: Timeout exceeded in read_nonblocking().

 During handling of the above exception, another exception occurred:

 Traceback (most recent call last):
   File "K:\eclipse_3.6.0\plugins\org.python.pydev_2.6.0.2012062818\pysrc\pydev_runfiles.py", line 432, in __get_module_from_str
     mod = __import__(modname)
   File "C:\workspace\Test_prj\Test_prj.py", line 19, in <module>
     for found in iter(partial(p.expect, patterns), 1): # until EOF
   File "C:\Python33\lib\site-packages\winpexpect-1.5-py3.3.egg\pexpect.py", line 1311, in expect
     return self.expect_list(compiled_pattern_list, timeout, searchwindowsize)
   File "C:\Python33\lib\site-packages\winpexpect-1.5-py3.3.egg\pexpect.py", line 1325, in expect_list
     return self.expect_loop(searcher_re(pattern_list), timeout, searchwindowsize)
   File "C:\Python33\lib\site-packages\winpexpect-1.5-py3.3.egg\pexpect.py", line 1409, in expect_loop
     raise TIMEOUT (str(e) + '\n' + str(self))
 pexpect.TIMEOUT: Timeout exceeded in read_nonblocking().
 <winpexpect.winspawn object at 0x0144AE50>
 version: 2.3 ($Revision: 399 $)
 command: si
 args: ['si', 'viewproject', '--project=c:/test.pj']
 searcher: searcher_re:
     0: re.compile("ynYN")
     1: EOF
 buffer (last 100 chars): 
 before (last 100 chars): 
 after: <class 'pexpect.TIMEOUT'>
 match: None
 match_index: None
 exitstatus: None
 flag_eof: False
 pid: 6448
 child_fd: 4
 closed: False
 timeout: 30
 delimiter: <class 'pexpect.EOF'>
 logfile: <_io.TextIOWrapper name='<stdout>' mode='w' encoding='Cp1252'>
 logfile_read: None
 logfile_send: None
 maxread: 2000
 ignorecase: False
 searchwindowsize: None
 delaybeforesend: 0.05
 delayafterclose: 0.1
 delayafterterminate: 0.1
 ERROR: Module: Test_prj could not be imported (file: C:\workspace\Test_prj\Test_prj.py).

安装 Winpexpect

懒人之道

Optional: Install Nose

Optional: Install Pip

第一世界问题

Python 对我来说是一门新语言，我之前从未为 Python 安装过软件包。此外，Python 3.x 与其他版本的 Python 略有不同，这使得安装模块更像是一种冒险。

所以，为了帮助其他人获得一些甜蜜甜蜜的模块操作（并帮助那些知识渊博的人看看我是否做错了什么）这里有一个即将成功的故事（希望如此）记录我如何获得和安装我的第一个模块。

设置

Python 允许第三方团体开发和分发扩展编程语言能力的模块。当然，有一种标准方法可以帮助第三方开发人员尽可能轻松地为最终用户提供模块。

对于 Python 3.x，分发模块的标准称为 Distutils。

以下是开发人员使用 Distutils 的方式： Distributing Python Modules

以下是最终用户使用 Distutils 的方式： Installing Python Modules

通常，在命令行中导航到下载模块的文件夹并运行“setup.py install”就足够了。

但是

有时生活并不那么容易，您的安装可能仍然存在问题。事实上，你可能需要别的东西。例如，您可能会收到以下错误：

“ImportError “没有名为 Setuptools 的模块””

幸运的是，有一个解决方案： Python 3: ImportError "No Module named Setuptools"

事实证明，并非所有东西都使用 distutils。一些软件包使用 setuptools。不幸的是，没有适用于 Python 3.x 的设置工具。相反，Python 3.x 使用的是 setuptools 的一个分支分发。

所以对于那些使用 Python 3.x 的人来说，这里是 Distribute：Distribute

对于那些使用 Python 2.x 的用户，这里是 Setuptools：Setuptools

在 Distribute 的安装说明中，它说如下： “下载 distribute_setup.py <http://python-distribute.org/distribute_setup.py>_ 并使用您选择的 Python 解释器执行它。”

它还说：“请注意，源版本中也提供了此文件。”

所以我下载了 Distribute 并将其保存到计算机上。将其保存到计算机后，我从源版本中运行了distribute_setup.py 并收到以下错误：

Downloading http://pypi.python.org/packages/source/d/distribute/distribute-0.6.36.tar.gz
Traceback (most recent call last):
  File "C:\Python33\lib\urllib\request.py", line 1252, in do_open
    h.request(req.get_method(), req.selector, req.data, headers)       File "C:\Python33\lib\http\client.py", line 1049, in request
    self._send_request(method, url, body, headers)
  File "C:\Python33\lib\http\client.py", line 1087, in _send_request
    self.endheaders(body)
  File "C:\Python33\lib\http\client.py", line 1045, in endheaders
    self._send_output(message_body)
  File "C:\Python33\lib\http\client.py", line 890, in _send_output
    self.send(msg)
  File "C:\Python33\lib\http\client.py", line 828, in send
    self.connect()
  File "C:\Python33\lib\http\client.py", line 806, in connect
    self.timeout, self.source_address)
  File "C:\Python33\lib\socket.py", line 406, in create_connection
    for res in getaddrinfo(host, port, 0, SOCK_STREAM):
socket.gaierror: [Errno 11001] getaddrinfo failed

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
  File "C:\workspace\PythonTest\distribute_setup.py", line 553, in <module>
    sys.exit(main())
  File "C:\workspace\PythonTest\distribute_setup.py", line 549, in main
    tarball = download_setuptools(download_base=options.download_base)
  File "C:\workspace\PythonTest\distribute_setup.py", line 204, in download_setuptools
    src = urlopen(url)
  File "C:\Python33\lib\urllib\request.py", line 160, in urlopen
    return opener.open(url, data, timeout)
  File "C:\Python33\lib\urllib\request.py", line 473, in open
    response = self._open(req, data)
  File "C:\Python33\lib\urllib\request.py", line 491, in _open
    '_open', req)
  File "C:\Python33\lib\urllib\request.py", line 451, in _call_chain
    result = func(*args)
  File "C:\Python33\lib\urllib\request.py", line 1272, in http_open
    return self.do_open(http.client.HTTPConnection, req)
  File "C:\Python33\lib\urllib\request.py", line 1255, in do_open
    raise URLError(err)
urllib.error.URLError: <urlopen error [Errno 11001] getaddrinfo failed>

那不好！老实说，我仍然不知道该错误来自何处或为什么会发生。

不管怎样，我发现以下站点运行了一个 .exe 来安装分发以及 pip。

Install Distribute

Install Pip

所以我安装了这些，然后使用以下站点设置我的计算机以更轻松地使用 easy_install：Setting Up Easy Install Made Easy

一旦我得到这个工作，我就安装了鼻子：Nose

我得到鼻子的原因是因为Winpexpect website 说： "WinPexpect 包含单元测试。要运行测试，你需要鼻子。使用以下命令运行测试：

$ python setup.py test"

听起来不错:)。现在我只是希望我知道在哪里运行该测试。我知道，如果您手动安装，则使用 setup.py install 命令，因此在线压缩目录中肯定会有 setup.py。为了查看这是否正确，我下载并保存了 winpexpect 文件，提取信息，通过命令行导航到该文件，然后运行 setup.py test。

结果如下：

running test
running build_py
running egg_info
creating c:\documents and settings\slz1fh\desktop\winpexpect\geertj-winpexpect-76df3cfcb143\build\lib\winpexpect.egg-info
writing c:\documents and settings\slz1fh\desktop\winpexpect\geertj-winpexpect-76df3cfcb143\build\lib\winpexpect.egg-info\PKG-INFO
writing dependency_links to c:\documents and settings\slz1fh\desktop\winpexpect\geertj-winpexpect-76df3cfcb143\build\lib\winpexpect.egg-info\dependency_links.txt
writing top-level names to c:\documents and settings\slz1fh\desktop\winpexpect\geertj-winpexpect-76df3cfcb143\build\lib\winpexpect.egg-info\top_level.txt
writing requirements to c:\documents and settings\slz1fh\desktop\winpexpect\geertj-winpexpect-76df3cfcb143\build\lib\winpexpect.egg-info\requires.txt
writing manifest file 'c:\documents and settings\slz1fh\desktop\winpexpect\geertj-winpexpect-76df3cfcb143\build\lib\winpexpect.egg-info\SOURCES.txt'
reading manifest file 'c:\documents and settings\slz1fh\desktop\winpexpect\geertj-winpexpect-76df3cfcb143\build\lib\winpexpect.egg-info\SOURCES.txt'
writing manifest file 'c:\documents and settings\slz1fh\desktop\winpexpect\geertj-winpexpect-76df3cfcb143\build\lib\winpexpect.egg-info\SOURCES.txt'
running build_ext
Traceback (most recent call last):
  File "C:\Documents and Settings\SLZ1FH\Desktop\winpexpect\geertj-winpexpect-76df3cfcb143\setup.py", line 35, in <module>
    use_2to3 = True
  File "C:\Python33\lib\distutils\core.py", line 148, in setup
    dist.run_commands()
  File "C:\Python33\lib\distutils\dist.py", line 917, in run_commands
    self.run_command(cmd)
  File "C:\Python33\lib\distutils\dist.py", line 936, in run_command
    cmd_obj.run()
  File "C:\Python33\lib\site-packages\distribute-0.6.36-py3.3.egg\setuptools\command\test.py", line 138, in run
    self.with_project_on_sys_path(self.run_tests)
  File "C:\Python33\lib\site-packages\distribute-0.6.36-py3.3.egg\setuptools\command\test.py", line 118, in with_project_on_sys_path
    func()
  File "C:\Python33\lib\site-packages\distribute-0.6.36-py3.3.egg\setuptools\command\test.py", line 164, in run_tests
    testLoader = cks
  File "C:\Python33\lib\unittest\main.py", line 124, in __init__
    self.parseArgs(argv)
  File "C:\Python33\lib\unittest\main.py", line 168, in parseArgs
    self.createTests()
  File "C:\Python33\lib\unittest\main.py", line 175, in createTests
    self.module)
  File "C:\Python33\lib\unittest\loader.py", line 137, in loadTestsFromNames
    suites = [self.loadTestsFromName(name, module) for name in names]
  File "C:\Python33\lib\unittest\loader.py", line 137, in <listcomp>
    suites = [self.loadTestsFromName(name, module) for name in names]
  File "C:\Python33\lib\unittest\loader.py", line 96, in loadTestsFromName
    module = __import__('.'.join(parts_copy))
  File "C:\Python33\lib\site-packages\nose-1.3.0-py3.3.egg\nose\__init__.py", line 1, in <module>
    from nose.core import collector, main, run, run_exit, runmodule
  File "C:\Python33\lib\site-packages\nose-1.3.0-py3.3.egg\nose\core.py", line 143
    print "%s version %s" % (os.path.basename(sys.argv[0]), __version__)
                    ^
SyntaxError: invalid syntax

好的，所以 Python 3.3 版本的 Nose 包含 Python 3.3 的无效语法？

print "%s version %s" % (os.path.basename(sys.argv[0]), version)...

肯定应该有括号...这让我怀疑鼻子是否真的可以在这里工作，因为它显然是为早期版本的 Python 制作的。

【问题讨论】：

你的操作系统和python版本是什么？程序是否允许输入/输出重定向，如果它以非交互方式运行，它是否使用块缓冲，例如，以下是否失败：xx viewproject < answers.txt > output.txt 其中answers.txt 包含换行符上的每个答案？
我的操作系统是 Windows XP，我的 Python 版本是 3.3。正确使用 xx viewproject output.txt 功能。但是，我对命令行提示的响应需要基于提示本身。我不知道是否需要递归到子项目，直到我可以看到它希望我递归到哪个子项目。
“错误：模块：...”消息来自哪里？你如何运行Test_prj.py？
我正在为 C/C++ 开发人员在 Eclipse IDE 中运行 Test_prj.py。
尝试直接从控制台（cmd.exe）运行它（以防万一）。注意：脚本旁边不应该有winpexpect 目录，否则它会尝试导入它而不是安装的版本。

标签： python command-line python-3.x subprocess command-prompt

【解决方案1】：

在您提到的 cmets 中，xx viewproject < answers.txt > output.txt 有效，但您不能使用它，因为答案取决于子进程的输出。

通常可以使用pexpect 类的模块，例如winpexpect（适用于Windows）。类似的东西：

import re
import sys
from functools import partial
from winpexpect import EOF, winspawn as spawn

p = spawn('xx viewproject')
p.logfile = sys.stdout
patterns = ['the project:', re.escape('? [ynYN](n)'), EOF]
for found in iter(partial(p.expect, patterns), 2): # until EOF
    if found == 0:
        p.sendline(project_name)
    elif found == 1:
        filename = get_filename_from_prompt(p.before) # a regex could be used
        answer = yes_or_no_from_subproject.get(filename, 'no') # a dict
        p.sendline(answer)

如果提示以换行符终止（并且子进程不缓冲它们）；您可以直接使用subprocess 模块逐行阅读：

from subprocess import Popen, PIPE

with Popen(["xx", "viewproject"], stdin=PIPE, stdout=PIPE, 
           universal_newlines=True) as p:
    for line in p.stdout: 
        if line.startswith("Please enter the name of the project"):
            answer = project_name
        elif line.startswith("Would you like to recurse into the subproject"):
            filename = get_filename_from_prompt(line) # a regex could be used
            answer = yes_or_no_from_subproject.get(filename, 'n') # a dict
        else:
            continue # skip it
        print(answer, file=p.stdin) # provide answer
        p.stdin.flush()

要测试您是否可以使用subprocess 从xx 中读取内容：

from subprocess import Popen, PIPE, STDOUT

with Popen(["xx", "viewproject"], bufsize=0,
           stdin=PIPE, stdout=PIPE, stderr=STDOUT) as p:
    print(repr(p.stdout.read(1)))

【讨论】：

我很好奇，为什么建议 Windows only 非标准模块？有什么好处？
@Vyktor：OP uses Windows。 pexpect 如果您需要，可以在 *nix 上工作。您不能使用.communicate()，因为答案取决于子流程输出。如果提示没有被换行符终止，那么您需要复制/重新实现类似pexpect 的代码来读取/解析它们，并且可能会出现问题，例如the subprocess might use block-buffering if it is run non-interactively。
我知道您不能使用通信，但来自原始问题（仅来自评论）并不明显，因此我添加了有关如何直接使用读/写管道的新示例 +链接如何做到这一点block-buffering-safe...我只是更喜欢可以移植到任何平台的应用程序，没有像if os.platform == 'nt'这样的黑客攻击，并且没有任何外部包（如果可能的话）的干净python安装，所以我只是好奇有什么好处。
@Vyktor: the example you provided 不处理块缓冲问题：Python 进程可能由于缓冲而看不到提示，并且没有回答子进程块。
我现在已经尝试了这两种方法，但都遇到了错误。使用 subprocess 策略，代码会在 Popen 语句之后立即挂起，并且永远不会进入 for 循环。对于 Winpexpect 策略，有一条很长的错误消息，似乎是 TIMEOUT。

【解决方案2】：

是的，首先您可以通过以下方式将子流程创建为对象：

p = subprocess.Popen('xx viewproject', shell=True, stdin=subprocess.PIPE, 
                      stdout=subprocess.PIPE, universal_newlines=True)

然后你就会有像communicate() 这样的方法可用，例如：

newline = os.linesep # [1]
commands = ['y', 'n', 'y', 'n', 'y']
p.communicate( newline.join( commands))

^{1 - os.linesep}

它会一次发送所有答案（希望这就足够了），每次都依赖于相同的问题顺序。

您也可以尝试解析p.stdout，然后写入p.stdin，但是当一个缓冲区在等待另一个缓冲区时将满时，这可能会导致死锁，因此请注意这一点。幸运的是，谷歌上有some complex examples。

简单的版本是：

p = Popen(...)
line = p.stdout.readline() # At this point, if child process will wait for stdin
                           # you have a deadlock on your hands
parse_line( line)
p.stdin.write( newline.join( commands).encode( 'utf-8'))

我也会考虑重写：

p = subprocess.Popen('si viewproject --project=d:/Projects/test.pj', shell=True, 
                      stdin=subprocess.PIPE, stdout=subprocess.PIPE)

收件人：

p = subprocess.Popen( ['si', 'viewproject', '--project=d:/Projects/test.pj'],
                      shell=False, stdin=subprocess.PIPE, stdout=subprocess.PIPE)

除非您明确需要 Shell 调用。

【讨论】：

TypeError: 'str' does not support the buffer interface Enter the project name: *** A value for "--project" is required.
哇，很抱歉没有留下对错误 O_o 的描述。哎呀。对于命令数组，不幸的是我不会立即知道所有答案，因为 y/n 响应基于所显示的子文件夹的名称，我不知道该子文件夹何时出现。至于错误，它是以下行：p.communicate(newline.join(commands)) 对于测试项目，我知道提供['n', 'n', 'n'] 的命令将充分回答提供项目名称后的所有后续提示。
当前代码如下：p = subprocess.Popen('si viewproject --project=d:/Projects/test.pj', shell = True, stdin=subprocess.PIPE, stdout=subprocess.PIPE)newline = os.linesepcommands = ['n', 'n', 'n']p.communicate(newline.join(commands))提供Popen作品中的项目名称，对于测试项目，三个'n'就足够了。无论如何，p.communicate(newline.join(commands)) 上的错误仍然存在。
@Stoating 您能否在此处粘贴触发您的错误的回溯？
@Stoating: p.communicate() 在 Python 3.3 上默认需要字节。您可以指定 Popen(..., universal_newlines=True) 来使用 Unicode 字符串。