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.
40 lines
1.1 KiB
40 lines
1.1 KiB
6 years ago
|
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)
|