90 lines
3.6 KiB
Python
90 lines
3.6 KiB
Python
# Code borrowed from ptpython
|
|
# https://github.com/jonathanslenders/ptpython/blob/86b71a89626114b18898a0af463978bdb32eeb70/ptpython/eventloop.py
|
|
|
|
# Copyright (c) 2015, Jonathan Slenders
|
|
# All rights reserved.
|
|
#
|
|
# Redistribution and use in source and binary forms, with or without modification,
|
|
# are permitted provided that the following conditions are met:
|
|
#
|
|
# * Redistributions of source code must retain the above copyright notice, this
|
|
# list of conditions and the following disclaimer.
|
|
#
|
|
# * Redistributions in binary form must reproduce the above copyright notice, this
|
|
# list of conditions and the following disclaimer in the documentation and/or
|
|
# other materials provided with the distribution.
|
|
#
|
|
# * Neither the name of the {organization} nor the names of its
|
|
# contributors may be used to endorse or promote products derived from
|
|
# this software without specific prior written permission.
|
|
#
|
|
# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
|
|
# ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
|
|
# WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
|
|
# DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR
|
|
# ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
|
|
# (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
|
|
# LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON
|
|
# ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
|
# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
|
|
# SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
|
|
|
"""
|
|
Wrapper around the eventloop that gives some time to the Tkinter GUI to process
|
|
events when it's loaded and while we are waiting for input at the REPL. This
|
|
way we don't block the UI of for instance ``turtle`` and other Tk libraries.
|
|
|
|
(Normally Tkinter registers it's callbacks in ``PyOS_InputHook`` to integrate
|
|
in readline. ``prompt-toolkit`` doesn't understand that input hook, but this
|
|
will fix it for Tk.)
|
|
"""
|
|
import time
|
|
|
|
import _tkinter
|
|
import tkinter
|
|
|
|
def inputhook(inputhook_context):
|
|
"""
|
|
Inputhook for Tk.
|
|
Run the Tk eventloop until prompt-toolkit needs to process the next input.
|
|
"""
|
|
# Get the current TK application.
|
|
root = tkinter._default_root
|
|
|
|
def wait_using_filehandler():
|
|
"""
|
|
Run the TK eventloop until the file handler that we got from the
|
|
inputhook becomes readable.
|
|
"""
|
|
# Add a handler that sets the stop flag when `prompt-toolkit` has input
|
|
# to process.
|
|
stop = [False]
|
|
def done(*a):
|
|
stop[0] = True
|
|
|
|
root.createfilehandler(inputhook_context.fileno(), _tkinter.READABLE, done)
|
|
|
|
# Run the TK event loop as long as we don't receive input.
|
|
while root.dooneevent(_tkinter.ALL_EVENTS):
|
|
if stop[0]:
|
|
break
|
|
|
|
root.deletefilehandler(inputhook_context.fileno())
|
|
|
|
def wait_using_polling():
|
|
"""
|
|
Windows TK doesn't support 'createfilehandler'.
|
|
So, run the TK eventloop and poll until input is ready.
|
|
"""
|
|
while not inputhook_context.input_is_ready():
|
|
while root.dooneevent(_tkinter.ALL_EVENTS | _tkinter.DONT_WAIT):
|
|
pass
|
|
# Sleep to make the CPU idle, but not too long, so that the UI
|
|
# stays responsive.
|
|
time.sleep(.01)
|
|
|
|
if root is not None:
|
|
if hasattr(root, 'createfilehandler'):
|
|
wait_using_filehandler()
|
|
else:
|
|
wait_using_polling()
|