如何向 Python 交互式 shell 添加命令？

Question

对于我经常在 Python 中输入的命令，我试图节省自己的敲击次数。

在我的 python 启动脚本 中，我定义了一个名为 load 的函数，它类似于导入，但添加了一些功能。它需要一个字符串：

def load(s):
  # Do some stuff
  return something

为了调用这个函数，我必须输入

>>> load('something')

我宁愿能够简单地输入：

>>> load something

我正在运行Python readline 支持，所以我知道那里存在一些可编程性，但我不知道使用它是否可以实现这种功能。

我尝试通过使用 InteractivConsole 并在启动文件中创建它的实例来解决这个问题，如下所示：

import code, re, traceback

class LoadingInteractiveConsole(code.InteractiveConsole):
  def raw_input(self, prompt = ""):
    s = raw_input(prompt)
    match = re.match('^load\s+(.+)', s)
    if match:
      module = match.group(1)
      try:
        load(module)
        print "Loaded " + module
      except ImportError:
        traceback.print_exc()
      return ''
    else:
      return s

console = LoadingInteractiveConsole()
console.interact("")

这与我必须按 Ctrl-D 两次才能退出 python 解释器的警告一起工作：一次离开我的自定义控制台，一次离开真实的控制台。

有没有一种方法可以做到这一点，而无需编写自定义 C 程序并将解释器嵌入其中？

编辑

频道外，我建议将其附加到启动文件的末尾：

import sys
sys.exit()

它工作得很好，但我仍然对替代解决方案感兴趣。

Answer 1

你可以尝试 ipython - 它提供了一个 python shell，它允许很多事情，包括 自动括号 可以根据您的要求提供函数调用。

Answer 2

我认为你想要 cmd 模块。

请参阅此处的教程：
http://wiki.python.org/moin/CmdModule

Answer 3

我不想回答我自己的问题，但还没有一个答案适用于我使用的所有 Python 版本。除了我在问题编辑中发布的解决方案（这就是我现在正在使用的）之外，还有另一个：

编辑 .bashrc 以包含以下行：

alias python3='python3 ~/py/shellreplace.py'
alias python='python ~/py/shellreplace.py'
alias python27='python27 ~/py/shellreplace.py'

然后只需移动所有 LoadingInteractiveConsole 代码写入文件 ~/py/shellreplace.py 一旦脚本执行完毕，python 将停止执行，改进后的交互式会话将是无缝的。