Ipython自定义提示以显示单元运行时 [英] Ipython customize prompt to display cell run time

问题描述

我想知道如何配置Ipython，以便将最后一个命令的运行时间（以毫秒/秒为单位）添加到正确的命令提示符。这可以在ZSH / Bash shell中完成，如图所示 https://coderwall.com/p/kmchbw

i am wondering how to configure Ipython so that it adds the run time of the last command in milliseconds/seconds to the right command prompt. This could be done in ZSH/Bash shells as illustrated here https://coderwall.com/p/kmchbw

我应该怎么做？

推荐答案

这是一个代码片段，对每个语句进行计时并在下一个提示之前将其打印正确，并且还可以通过名称texc访问该值。

This is a code snippet that times each statement and prints it right adjusted before the next prompt, and also makes the value accessible by name 'texc'.

# Assumes from __future__ import print_function
from time import time
import blessings  # Not a necessary requirement
class ExecTimer(object):
    def __init__(self, ip):
        self.shell = ip
        self.t_pre = time()
        self.texc = 0
        self.prev_texc = 0
        self.term = blessings.Terminal()

    def pre_execute(self):
        self.t_pre = time()

    def post_execute(self):
        self.prev_texc = self.texc
        self.texc = round(time() - self.t_pre, 4)
        print(self.term.bold_blue(
            '{} s'.format(self.texc).rjust(self.term.width - 1)
        ))
        # Only add or update user namespace var if it is safe to do so
        if 'texc' not in self.shell.user_ns or \
                self.shell.user_ns['texc'] == self.prev_texc:
            self.shell.push({'texc': self.texc})
        else:
            pass

    def register(self):
        self.shell.events.register('pre_execute', self.pre_execute)
        self.shell.events.register('post_execute', self.post_execute)

ExecTimer(get_ipython()).register()

要在输入提示上方打印它，请删除打印，然后输入ipython_config.py设置：

To print it above the in-prompt instead, remove the print, and in ipython_config.py set:

c.PromptManager.in_template = '{texc} s\nIn[\\#]: '

或在同一文件（startup.py）中使用

or in the same file (startup.py) use

get_ipython().run_line_magic(
    'config',
    r"PromptManager.in_template = '{texc} s\nIn[\\#]: '"
)

这篇关于Ipython自定义提示以显示单元运行时的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！