98 lines
3.1 KiB
Python
98 lines
3.1 KiB
Python
from PyQt5.QtWidgets import QFileDialog
|
|
|
|
from electrum.i18n import _
|
|
from electrum.plugin import run_hook
|
|
from electrum.simple_config import SimpleConfig
|
|
from electrum.util import UserFacingException
|
|
from electrum.logging import Logger
|
|
|
|
from .util import ButtonsTextEdit, MessageBoxMixin, ColorScheme, getOpenFileName
|
|
from .qrreader import scan_qrcode
|
|
|
|
|
|
class ShowQRTextEdit(ButtonsTextEdit):
|
|
|
|
def __init__(self, text=None, *, config: SimpleConfig):
|
|
ButtonsTextEdit.__init__(self, text)
|
|
self.config = config
|
|
self.setReadOnly(True)
|
|
icon = "qrcode_white.png" if ColorScheme.dark_scheme else "qrcode.png"
|
|
self.addButton(icon, self.qr_show, _("Show as QR code"))
|
|
|
|
run_hook('show_text_edit', self)
|
|
|
|
def qr_show(self):
|
|
from .qrcodewidget import QRDialog
|
|
try:
|
|
s = str(self.toPlainText())
|
|
except:
|
|
s = self.toPlainText()
|
|
QRDialog(
|
|
data=s,
|
|
parent=self,
|
|
config=self.config,
|
|
).exec_()
|
|
|
|
def contextMenuEvent(self, e):
|
|
m = self.createStandardContextMenu()
|
|
m.addAction(_("Show as QR code"), self.qr_show)
|
|
m.exec_(e.globalPos())
|
|
|
|
|
|
class ScanQRTextEdit(ButtonsTextEdit, MessageBoxMixin, Logger):
|
|
|
|
def __init__(self, text="", allow_multi=False, *, config: SimpleConfig):
|
|
ButtonsTextEdit.__init__(self, text)
|
|
Logger.__init__(self)
|
|
self.allow_multi = allow_multi
|
|
self.config = config
|
|
self.setReadOnly(False)
|
|
self.addButton("file.png", self.file_input, _("Read file"))
|
|
icon = "camera_white.png" if ColorScheme.dark_scheme else "camera_dark.png"
|
|
self.addButton(icon, self.qr_input, _("Read QR code"))
|
|
run_hook('scan_text_edit', self)
|
|
|
|
def file_input(self):
|
|
fileName = getOpenFileName(
|
|
parent=self,
|
|
title='select file',
|
|
config=self.config,
|
|
)
|
|
if not fileName:
|
|
return
|
|
try:
|
|
try:
|
|
with open(fileName, "r") as f:
|
|
data = f.read()
|
|
except UnicodeError as e:
|
|
with open(fileName, "rb") as f:
|
|
data = f.read()
|
|
data = data.hex()
|
|
except BaseException as e:
|
|
self.show_error(_('Error opening file') + ':\n' + repr(e))
|
|
else:
|
|
self.setText(data)
|
|
|
|
def qr_input(self, *, callback=None) -> None:
|
|
def cb(success: bool, error: str, data):
|
|
if not success:
|
|
if error:
|
|
self.show_error(error)
|
|
return
|
|
if not data:
|
|
data = ''
|
|
if self.allow_multi:
|
|
new_text = self.text() + data + '\n'
|
|
else:
|
|
new_text = data
|
|
self.setText(new_text)
|
|
if callback and success:
|
|
callback(data)
|
|
|
|
scan_qrcode(parent=self.top_level_window(), config=self.config, callback=cb)
|
|
|
|
def contextMenuEvent(self, e):
|
|
m = self.createStandardContextMenu()
|
|
m.addAction(_("Read QR code"), self.qr_input)
|
|
m.exec_(e.globalPos())
|