You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
343 lines
9.0 KiB
343 lines
9.0 KiB
from __future__ import unicode_literals
|
|
import inspect
|
|
import os
|
|
import signal
|
|
import sys
|
|
import threading
|
|
import weakref
|
|
|
|
from collections import deque
|
|
from functools import partial
|
|
from six import PY2, text_type
|
|
from six.moves import range
|
|
from wcwidth import wcwidth
|
|
from .cache import memoized
|
|
|
|
__all__ = [
|
|
'Event',
|
|
'DummyContext',
|
|
'get_cwidth',
|
|
'suspend_to_background_supported',
|
|
'is_conemu_ansi',
|
|
'is_windows',
|
|
'in_main_thread',
|
|
'take_using_weights',
|
|
'test_callable_args',
|
|
'to_str',
|
|
'to_int',
|
|
'to_float',
|
|
]
|
|
|
|
|
|
class Event(object):
|
|
"""
|
|
Simple event to which event handlers can be attached. For instance::
|
|
|
|
class Cls:
|
|
def __init__(self):
|
|
# Define event. The first parameter is the sender.
|
|
self.event = Event(self)
|
|
|
|
obj = Cls()
|
|
|
|
def handler(sender):
|
|
pass
|
|
|
|
# Add event handler by using the += operator.
|
|
obj.event += handler
|
|
|
|
# Fire event.
|
|
obj.event()
|
|
"""
|
|
def __init__(self, sender, handler=None):
|
|
self.sender = sender
|
|
self._handlers = []
|
|
|
|
if handler is not None:
|
|
self += handler
|
|
|
|
def __call__(self):
|
|
" Fire event. "
|
|
for handler in self._handlers:
|
|
handler(self.sender)
|
|
|
|
def fire(self):
|
|
" Alias for just calling the event. "
|
|
self()
|
|
|
|
def add_handler(self, handler):
|
|
"""
|
|
Add another handler to this callback.
|
|
(Handler should be a callable that takes exactly one parameter: the
|
|
sender object.)
|
|
"""
|
|
# Test handler.
|
|
assert callable(handler)
|
|
if not _func_takes_one_arg(handler):
|
|
raise TypeError("%r doesn't take exactly one argument." % handler)
|
|
|
|
# Add to list of event handlers.
|
|
self._handlers.append(handler)
|
|
|
|
def remove_handler(self, handler):
|
|
"""
|
|
Remove a handler from this callback.
|
|
"""
|
|
if handler in self._handlers:
|
|
self._handlers.remove(handler)
|
|
|
|
def __iadd__(self, handler):
|
|
" `event += handler` notation for adding a handler. "
|
|
self.add_handler(handler)
|
|
return self
|
|
|
|
def __isub__(self, handler):
|
|
" `event -= handler` notation for removing a handler. "
|
|
self.remove_handler(handler)
|
|
return self
|
|
|
|
|
|
# Cache of signatures. Improves the performance of `test_callable_args`.
|
|
_signatures_cache = weakref.WeakKeyDictionary()
|
|
|
|
_inspect_signature = getattr(inspect, 'signature', None) # Only on Python 3.
|
|
|
|
|
|
def test_callable_args(func, args):
|
|
"""
|
|
Return True when this function can be called with the given arguments.
|
|
"""
|
|
assert isinstance(args, (list, tuple))
|
|
|
|
if _inspect_signature is not None:
|
|
# For Python 3, use inspect.signature.
|
|
try:
|
|
sig = _signatures_cache[func]
|
|
except KeyError:
|
|
sig = _inspect_signature(func)
|
|
_signatures_cache[func] = sig
|
|
|
|
try:
|
|
sig.bind(*args)
|
|
except TypeError:
|
|
return False
|
|
else:
|
|
return True
|
|
else:
|
|
# For older Python versions, fall back to using getargspec
|
|
# and don't check for `partial`.
|
|
if isinstance(func, partial):
|
|
return True
|
|
|
|
spec = inspect.getargspec(func)
|
|
|
|
# Drop the 'self'
|
|
def drop_self(spec):
|
|
args, varargs, varkw, defaults = spec
|
|
if args[0:1] == ['self']:
|
|
args = args[1:]
|
|
return inspect.ArgSpec(args, varargs, varkw, defaults)
|
|
|
|
spec = drop_self(spec)
|
|
|
|
# When taking *args, always return True.
|
|
if spec.varargs is not None:
|
|
return True
|
|
|
|
# Test whether the given amount of args is between the min and max
|
|
# accepted argument counts.
|
|
return len(spec.args) - len(spec.defaults or []) <= len(args) <= len(spec.args)
|
|
|
|
|
|
@memoized(maxsize=1024)
|
|
def _func_takes_one_arg(func):
|
|
"""
|
|
Test whether the given function can be called with exactly one argument.
|
|
"""
|
|
return test_callable_args(func, [None])
|
|
|
|
|
|
class DummyContext(object):
|
|
"""
|
|
(contextlib.nested is not available on Py3)
|
|
"""
|
|
def __enter__(self):
|
|
pass
|
|
|
|
def __exit__(self, *a):
|
|
pass
|
|
|
|
|
|
class _CharSizesCache(dict):
|
|
"""
|
|
Cache for wcwidth sizes.
|
|
"""
|
|
LONG_STRING_MIN_LEN = 64 # Minimum string length for considering it long.
|
|
MAX_LONG_STRINGS = 16 # Maximum number of long strings to remember.
|
|
|
|
def __init__(self):
|
|
super(_CharSizesCache, self).__init__()
|
|
# Keep track of the "long" strings in this cache.
|
|
self._long_strings = deque()
|
|
|
|
def __missing__(self, string):
|
|
# Note: We use the `max(0, ...` because some non printable control
|
|
# characters, like e.g. Ctrl-underscore get a -1 wcwidth value.
|
|
# It can be possible that these characters end up in the input
|
|
# text.
|
|
if len(string) == 1:
|
|
result = max(0, wcwidth(string))
|
|
else:
|
|
result = sum(self[c] for c in string)
|
|
|
|
# Store in cache.
|
|
self[string] = result
|
|
|
|
# Rotate long strings.
|
|
# (It's hard to tell what we can consider short...)
|
|
if len(string) > self.LONG_STRING_MIN_LEN:
|
|
long_strings = self._long_strings
|
|
long_strings.append(string)
|
|
|
|
if len(long_strings) > self.MAX_LONG_STRINGS:
|
|
key_to_remove = long_strings.popleft()
|
|
if key_to_remove in self:
|
|
del self[key_to_remove]
|
|
|
|
return result
|
|
|
|
|
|
_CHAR_SIZES_CACHE = _CharSizesCache()
|
|
|
|
|
|
def get_cwidth(string):
|
|
"""
|
|
Return width of a string. Wrapper around ``wcwidth``.
|
|
"""
|
|
return _CHAR_SIZES_CACHE[string]
|
|
|
|
|
|
def suspend_to_background_supported():
|
|
"""
|
|
Returns `True` when the Python implementation supports
|
|
suspend-to-background. This is typically `False' on Windows systems.
|
|
"""
|
|
return hasattr(signal, 'SIGTSTP')
|
|
|
|
|
|
def is_windows():
|
|
"""
|
|
True when we are using Windows.
|
|
"""
|
|
return sys.platform.startswith('win') # E.g. 'win32', not 'darwin' or 'linux2'
|
|
|
|
|
|
def is_windows_vt100_supported():
|
|
"""
|
|
True when we are using Windows, but VT100 escape sequences are supported.
|
|
"""
|
|
# Import needs to be inline. Windows libraries are not always available.
|
|
from prompt_toolkit.output.windows10 import is_win_vt100_enabled
|
|
return is_windows() and is_win_vt100_enabled()
|
|
|
|
|
|
def is_conemu_ansi():
|
|
"""
|
|
True when the ConEmu Windows console is used.
|
|
"""
|
|
return is_windows() and os.environ.get('ConEmuANSI', 'OFF') == 'ON'
|
|
|
|
|
|
def in_main_thread():
|
|
"""
|
|
True when the current thread is the main thread.
|
|
"""
|
|
return threading.current_thread().__class__.__name__ == '_MainThread'
|
|
|
|
|
|
def get_term_environment_variable():
|
|
" Return the $TERM environment variable. "
|
|
term = os.environ.get('TERM', '')
|
|
if PY2:
|
|
term = term.decode('utf-8')
|
|
return term
|
|
|
|
|
|
def take_using_weights(items, weights):
|
|
"""
|
|
Generator that keeps yielding items from the items list, in proportion to
|
|
their weight. For instance::
|
|
|
|
# Getting the first 70 items from this generator should have yielded 10
|
|
# times A, 20 times B and 40 times C, all distributed equally..
|
|
take_using_weights(['A', 'B', 'C'], [5, 10, 20])
|
|
|
|
:param items: List of items to take from.
|
|
:param weights: Integers representing the weight. (Numbers have to be
|
|
integers, not floats.)
|
|
"""
|
|
assert isinstance(items, list)
|
|
assert isinstance(weights, list)
|
|
assert all(isinstance(i, int) for i in weights)
|
|
assert len(items) == len(weights)
|
|
assert len(items) > 0
|
|
|
|
# Remove items with zero-weight.
|
|
items2 = []
|
|
weights2 = []
|
|
for i, w in zip(items, weights):
|
|
if w > 0:
|
|
items2.append(i)
|
|
weights2.append(w)
|
|
|
|
items = items2
|
|
weights = weights2
|
|
|
|
# Make sure that we have some items left.
|
|
if not items:
|
|
raise ValueError("Did't got any items with a positive weight.")
|
|
|
|
#
|
|
already_taken = [0 for i in items]
|
|
item_count = len(items)
|
|
max_weight = max(weights)
|
|
|
|
i = 0
|
|
while True:
|
|
# Each iteration of this loop, we fill up until by (total_weight/max_weight).
|
|
adding = True
|
|
while adding:
|
|
adding = False
|
|
|
|
for item_i, item, weight in zip(range(item_count), items, weights):
|
|
if already_taken[item_i] < i * weight / float(max_weight):
|
|
yield item
|
|
already_taken[item_i] += 1
|
|
adding = True
|
|
|
|
i += 1
|
|
|
|
|
|
def to_str(value):
|
|
" Turn callable or string into string. "
|
|
if callable(value):
|
|
return to_str(value())
|
|
else:
|
|
return text_type(value)
|
|
|
|
|
|
def to_int(value):
|
|
" Turn callable or int into int. "
|
|
if callable(value):
|
|
return to_int(value())
|
|
else:
|
|
return int(value)
|
|
|
|
|
|
def to_float(value):
|
|
" Turn callable or float into float. "
|
|
if callable(value):
|
|
return to_float(value())
|
|
else:
|
|
return float(value)
|