X-Git-Url: https://git.novaco.in/?a=blobdiff_plain;f=plugins%2Fqrscanner.py;h=0ff1fe936c7c83959800a0e3594f3853b547000a;hb=81d1e67253c8ca581e821a6cd9e5ee1502fffd08;hp=cc164843a0173522c34ffa1f5931f5b664de0a3b;hpb=b6afa2455c62a91df7e00ee92d203ef72036bf8e;p=electrum-nvc.git diff --git a/plugins/qrscanner.py b/plugins/qrscanner.py index cc16484..0ff1fe9 100644 --- a/plugins/qrscanner.py +++ b/plugins/qrscanner.py @@ -1,123 +1,160 @@ from electrum.util import print_error from urlparse import urlparse, parse_qs -from PyQt4.QtGui import QPushButton -from electrum_gui.i18n import _ +from PyQt4.QtGui import QPushButton, QMessageBox, QDialog, QVBoxLayout, QHBoxLayout, QGridLayout, QLabel, QLineEdit, QComboBox +from PyQt4.QtCore import Qt + +from electrum.i18n import _ +import re +import os +from electrum import Transaction +from electrum.bitcoin import MIN_RELAY_TX_FEE, is_valid +from electrum_gui.qt.qrcodewidget import QRCodeWidget +from electrum import bmp +from electrum_gui.qt import HelpButton, EnterButton +import json try: import zbar except ImportError: zbar = None +from electrum import BasePlugin +class Plugin(BasePlugin): + def fullname(self): return 'QR scans' -def init(gui): - if is_enabled(): - gui.set_hook('create_send_tab', create_send_tab) - else: - gui.unset_hook('create_send_tab', create_send_tab) + def description(self): return "QR Scans.\nInstall the zbar package to enable this plugin.\nOn linux, type: 'apt-get install python-zbar'" -def get_info(): - return 'QR scans', "QR Scans.\nInstall the zbar package to enable this plugin" + def __init__(self, gui, name): + BasePlugin.__init__(self, gui, name) + self._is_available = self._init() -def is_enabled(): - return is_available() - -def toggle(gui): - return is_enabled() + def _init(self): + if not zbar: + return False + try: + proc = zbar.Processor() + proc.init(video_device=self.video_device()) + except zbar.UnsupportedError: + return False + except zbar.SystemError: + # Cannot open video device + pass + #return False + return True -def is_available(): - if not zbar: - return False + def init(self): + self.win = self.gui.main_window - try: - proc = zbar.Processor() - proc.init() - except zbar.SystemError: - # Cannot open video device - return False + def is_available(self): + return self._is_available - return True + def is_enabled(self): + return True -def scan_qr(): - proc = zbar.Processor() - proc.init() - proc.visible = True - - while True: + def scan_qr_hook(self): + proc = zbar.Processor() try: - proc.process_one() - except: - # User closed the preview window - return {} - - for r in proc.results: - if str(r.type) != 'QRCODE': - continue - - return parse_uri(r.data) - -def parse_uri(uri): - if ':' not in uri: - # It's just an address (not BIP21) - return {'address': uri} - - if '//' not in uri: - # Workaround for urlparse, it don't handle bitcoin: URI properly - uri = uri.replace(':', '://') - - uri = urlparse(uri) - result = {'address': uri.netloc} - - if uri.path.startswith('?'): - params = parse_qs(uri.path[1:]) - else: - params = parse_qs(uri.path) - - for k,v in params.items(): - if k in ('amount', 'label', 'message'): - result[k] = v[0] + proc.init(video_device=self.video_device()) + except zbar.SystemError, e: + QMessageBox.warning(self.win, _('Error'), _(e), _('OK')) + return + + proc.visible = True + + while True: + try: + proc.process_one() + except Exception: + # User closed the preview window + return {} + + for r in proc.results: + if str(r.type) != 'QRCODE': + continue + return r.data - return result + def video_device(self): + device = self.config.get("video_device", "default") + if device == 'default': + device = '' + return device + def requires_settings(self): + return True -def fill_from_qr(self): - qrcode = scan_qr() - if 'address' in qrcode: - self.payto_e.setText(qrcode['address']) - if 'amount' in qrcode: - self.amount_e.setText(str(qrcode['amount'])) - if 'label' in qrcode: - self.message_e.setText(qrcode['label']) - if 'message' in qrcode: - self.message_e.setText("%s (%s)" % (self.message_e.text(), qrcode['message'])) - - -def create_send_tab(gui, grid): - if is_available(): - b = QPushButton(_("Scan QR code")) - b.clicked.connect(lambda: fill_from_qr(gui)) - grid.addWidget(b, 1, 5) - - - -if __name__ == '__main__': - # Run some tests - - assert(parse_uri('1Marek48fwU7mugmSe186do2QpUkBnpzSN') == - {'address': '1Marek48fwU7mugmSe186do2QpUkBnpzSN'}) - - assert(parse_uri('bitcoin://1Marek48fwU7mugmSe186do2QpUkBnpzSN') == - {'address': '1Marek48fwU7mugmSe186do2QpUkBnpzSN'}) - - assert(parse_uri('bitcoin:1Marek48fwU7mugmSe186do2QpUkBnpzSN') == - {'address': '1Marek48fwU7mugmSe186do2QpUkBnpzSN'}) - - assert(parse_uri('bitcoin:1Marek48fwU7mugmSe186do2QpUkBnpzSN?amount=10') == - {'amount': '10', 'address': '1Marek48fwU7mugmSe186do2QpUkBnpzSN'}) - - assert(parse_uri('bitcoin:1Marek48fwU7mugmSe186do2QpUkBnpzSN?amount=10&label=slush&message=Small%20tip%20to%20slush') == - {'amount': '10', 'label': 'slush', 'message': 'Small tip to slush', 'address': '1Marek48fwU7mugmSe186do2QpUkBnpzSN'}) - + def settings_widget(self, window): + return EnterButton(_('Settings'), self.settings_dialog) + def _find_system_cameras(self): + device_root = "/sys/class/video4linux" + devices = {} # Name -> device + if os.path.exists(device_root): + for device in os.listdir(device_root): + name = open(os.path.join(device_root, device, 'name')).read() + devices[name] = os.path.join("/dev",device) + return devices + + def settings_dialog(self): + system_cameras = self._find_system_cameras() + + d = QDialog() + layout = QGridLayout(d) + layout.addWidget(QLabel("Choose a video device:"),0,0) + + # Create a combo box with the available video devices: + combo = QComboBox() + + # on change trigger for video device selection, makes the + # manual device selection only appear when needed: + def on_change(x): + combo_text = str(combo.itemText(x)) + combo_data = combo.itemData(x) + if combo_text == "Manually specify a device": + custom_device_label.setVisible(True) + self.video_device_edit.setVisible(True) + if self.config.get("video_device") == "default": + self.video_device_edit.setText("") + else: + self.video_device_edit.setText(self.config.get("video_device",'')) + else: + custom_device_label.setVisible(False) + self.video_device_edit.setVisible(False) + self.video_device_edit.setText(combo_data.toString()) + + # on save trigger for the video device selection window, + # stores the chosen video device on close. + def on_save(): + device = str(self.video_device_edit.text()) + self.config.set_key("video_device", device) + d.accept() + + custom_device_label = QLabel("Video device: ") + custom_device_label.setVisible(False) + layout.addWidget(custom_device_label,1,0) + self.video_device_edit = QLineEdit() + self.video_device_edit.setVisible(False) + layout.addWidget(self.video_device_edit, 1,1,2,2) + combo.currentIndexChanged.connect(on_change) + + combo.addItem("Default","default") + for camera, device in system_cameras.items(): + combo.addItem(camera, device) + combo.addItem("Manually specify a device",self.config.get("video_device")) + + # Populate the previously chosen device: + index = combo.findData(self.config.get("video_device")) + combo.setCurrentIndex(index) + + layout.addWidget(combo,0,1) + + self.accept = QPushButton(_("Done")) + self.accept.clicked.connect(on_save) + layout.addWidget(self.accept,4,2) + + if d.exec_(): + return True + else: + return False