ydb/contrib/python/prompt-toolkit/py2/prompt_toolkit/clipboard/in_memory.py

from .base import Clipboard, ClipboardData

from collections import deque

__all__ = (
    'InMemoryClipboard',
)


class InMemoryClipboard(Clipboard):
    """
    Default clipboard implementation.
    Just keep the data in memory.

    This implements a kill-ring, for Emacs mode.
    """
    def __init__(self, data=None, max_size=60):
        assert data is None or isinstance(data, ClipboardData)
        assert max_size >= 1

        self.max_size = max_size
        self._ring = deque()
        if data is not None:
            self.set_data(data)

    def set_data(self, data):
        assert isinstance(data, ClipboardData)
        self._ring.appendleft(data)

        while len(self._ring) > self.max_size:
            self._ring.pop()

    def get_data(self):
        if self._ring:
            return self._ring[0]
        else:
            return ClipboardData()

    def rotate(self):
        if self._ring:
            # Add the very first item at the end.
            self._ring.append(self._ring.popleft())