aboutsummaryrefslogtreecommitdiffstats
path: root/contrib/python/prompt-toolkit/py2/prompt_toolkit/clipboard/pyperclip.py
blob: 2dc49ae8cba8cdb0cdee125d6818f826b0c348fa (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
from __future__ import absolute_import, unicode_literals 
import pyperclip 
 
from prompt_toolkit.selection import SelectionType 
from .base import Clipboard, ClipboardData 
 
__all__ = ( 
    'PyperclipClipboard', 
) 
 
 
class PyperclipClipboard(Clipboard): 
    """ 
    Clipboard that synchronizes with the Windows/Mac/Linux system clipboard, 
    using the pyperclip module. 
    """ 
    def __init__(self): 
        self._data = None 
 
    def set_data(self, data): 
        assert isinstance(data, ClipboardData) 
        self._data = data 
        pyperclip.copy(data.text) 
 
    def get_data(self): 
        text = pyperclip.paste() 
 
        # When the clipboard data is equal to what we copied last time, reuse 
        # the `ClipboardData` instance. That way we're sure to keep the same 
        # `SelectionType`. 
        if self._data and self._data.text == text: 
            return self._data 
 
        # Pyperclip returned something else. Create a new `ClipboardData` 
        # instance. 
        else: 
            return ClipboardData( 
                text=text, 
                type=SelectionType.LINES if '\n' in text else SelectionType.LINES)