79 lines
2.3 KiB
Python
79 lines
2.3 KiB
Python
|
# coding=utf-8
|
||
|
|
"""
|
||
|
|
ipython.py - sopel ipython console!
|
||
|
|
Copyright © 2014, Elad Alfassa <elad@fedoraproject.org>
|
||
|
|
Licensed under the Eiffel Forum License 2.
|
||
|
|
|
||
|
|
Sopel: http://sopel.chat/
|
||
|
|
"""
|
||
|
|
from __future__ import unicode_literals, absolute_import, print_function, division
|
||
|
|
import sys
|
||
|
|
import module
|
||
|
|
if sys.version_info.major >= 3:
|
||
|
|
# Backup stderr/stdout wrappers
|
||
|
|
old_stdout = sys.stdout
|
||
|
|
old_stderr = sys.stderr
|
||
|
|
|
||
|
|
# IPython wants actual stderr and stdout. In Python 2, it only needed that
|
||
|
|
# when actually starting the console, but in Python 3 it seems to need that
|
||
|
|
# on import as well
|
||
|
|
sys.stdout = sys.__stdout__
|
||
|
|
sys.stderr = sys.__stderr__
|
||
|
|
try:
|
||
|
|
import IPython
|
||
|
|
if hasattr(IPython, 'terminal'):
|
||
|
|
from IPython.terminal.embed import InteractiveShellEmbed
|
||
|
|
else:
|
||
|
|
from IPython.frontend.terminal.embed import InteractiveShellEmbed
|
||
|
|
finally:
|
||
|
|
if sys.version_info.major >= 3:
|
||
|
|
# Restore stderr/stdout wrappers
|
||
|
|
sys.stdout = old_stdout
|
||
|
|
sys.stderr = old_stderr
|
||
|
|
|
||
|
|
console = None
|
||
|
|
|
||
|
|
|
||
|
|
@module.commands('console')
|
||
|
|
def interactive_shell(bot, trigger):
|
||
|
|
"""
|
||
|
|
Starts an interactive IPython console
|
||
|
|
"""
|
||
|
|
global console
|
||
|
|
if not trigger.admin:
|
||
|
|
bot.say('Only admins can start the interactive console')
|
||
|
|
return
|
||
|
|
if 'iconsole_running' in bot.memory and bot.memory['iconsole_running']:
|
||
|
|
bot.say('Console already running')
|
||
|
|
return
|
||
|
|
if not sys.__stdout__.isatty():
|
||
|
|
bot.say('A tty is required to start the console')
|
||
|
|
return
|
||
|
|
if bot._daemon:
|
||
|
|
bot.say('Can\'t start console when running as a daemon')
|
||
|
|
return
|
||
|
|
|
||
|
|
# Backup stderr/stdout wrappers
|
||
|
|
old_stdout = sys.stdout
|
||
|
|
old_stderr = sys.stderr
|
||
|
|
|
||
|
|
# IPython wants actual stderr and stdout
|
||
|
|
sys.stdout = sys.__stdout__
|
||
|
|
sys.stderr = sys.__stderr__
|
||
|
|
|
||
|
|
banner1 = 'Sopel interactive shell (embedded IPython)'
|
||
|
|
banner2 = '`bot` and `trigger` are available. To exit, type exit'
|
||
|
|
exitmsg = 'Interactive shell closed'
|
||
|
|
|
||
|
|
console = InteractiveShellEmbed(banner1=banner1, banner2=banner2,
|
||
|
|
exit_msg=exitmsg)
|
||
|
|
|
||
|
|
bot.memory['iconsole_running'] = True
|
||
|
|
bot.say('console started')
|
||
|
|
console()
|
||
|
|
bot.memory['iconsole_running'] = False
|
||
|
|
|
||
|
|
# Restore stderr/stdout wrappers
|
||
|
|
sys.stdout = old_stdout
|
||
|
|
sys.stderr = old_stderr
|