Обнаружение, когда сценарий python выполняется в интерактивном режиме в ipython - PullRequest
Новая
       29

Обнаружение, когда сценарий python выполняется в интерактивном режиме в ipython

7 голосов
/ 31 июля 2009

Есть ли способ для сценария Python автоматически определять, выполняется ли он в интерактивном режиме или нет? Кроме того, можно ли определить, используется ли ipython по сравнению с обычным исполняемым файлом c python?

Справочная информация: в моих скриптах на python обычно есть вызов exit (). Время от времени я запускаю сценарии в интерактивном режиме для отладки и профилирования, обычно в ipython. Когда я работаю в интерактивном режиме, я хочу подавить вызовы для выхода.

Разъяснение :

Предположим, у меня есть скрипт myscript.py, который выглядит следующим образом: 

#!/usr/bin/python
...do useful stuff...
exit(exit_status)

Иногда я хочу запустить скрипт в уже запущенном сеансе IPython, говоря что-то вроде: 

In [nnn]: %run -p -D myscript.pstats myscript.py

В конце скрипта вызов exit () приведет к зависанию ipython, в то время как он спрашивает меня, действительно ли я хочу выйти. Это незначительное раздражение при отладке (слишком незначительное для меня, чтобы позаботиться), но оно может испортить результаты профилирования: приглашение на выход включается в результаты профиля (что усложняет анализ, если я начинаю сеанс профилирования перед тем, как отправиться на обед) ,

То, что я хотел бы, это то, что позволяет мне изменить мой сценарий так, чтобы он выглядел так: 

#!/usr/bin/python
...do useful stuff...
if is_python_running_interactively():
    print "The exit_status was %d" % (exit_status,)
else:
    exit(exit_status)

Ответы [ 3 ]

13 голосов
/ 31 июля 2009

Я наткнулся на следующее, и мне кажется, это помогает мне: 

def in_ipython():
    try:
        __IPYTHON__
    except NameError:
        return False
    else:
        return True
3 голосов
/ 13 июля 2011

Документы говорят, что sys.ps1 не существует в неинтерактивном режиме. Кроме того, можно использовать sys.flags (для Python 2.6+), чтобы определить, использовали ли мы python -i <whatever>.

Этот сценарий обнаруживает, запускаем ли мы интерактивно, неинтерактивно и в посмертном режиме (что можно отнести к интерактивному режиму, если интерпретатор python вызывается неявно с использованием python -i и пользователь думает, что он приземлился в «интерактивной» консоли) : 

#!/usr/bin/python
# -*- coding: utf-8 -*-

import sys

# IPython recognition is missing; test here if __IPYTHON__ exists, etc.

if hasattr(sys, 'ps1'):
    print("Running interactively.")
else:
    print("Not running interactively...")
    if sys.flags.interactive:
        print("... but I'm in interactive postmortem mode.")

Поддержка IPython может быть добавлена, как описано г-ном Fooz.

2 голосов
/ 31 июля 2009

При интерактивном вызове python запускает скрипт в $ PYTHONSTARTUP, поэтому вы можете просто заставить эту переменную окружения вызывать скрипт, который устанавливает глобальный

...