40 lines
1.1 KiB
Python
40 lines
1.1 KiB
Python
from __future__ import absolute_import, unicode_literals
|
|
import pyperclip
|
|
|
|
from libs.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)
|