plugins: do not overload __init__, use init() instead
[electrum-nvc.git] / plugins / qrscanner.py
index e159c3a..c304464 100644 (file)
@@ -1,8 +1,15 @@
 from electrum.util import print_error
 from urlparse import urlparse, parse_qs
-from PyQt4.QtGui import QPushButton
+from PyQt4.QtGui import QPushButton, QMessageBox, QDialog, QVBoxLayout, QHBoxLayout, QGridLayout, QLabel
+from PyQt4.QtCore import Qt
 from electrum_gui.i18n import _
 
+import re
+from electrum.bitcoin import MIN_RELAY_TX_FEE, Transaction, is_valid
+from electrum_gui.qrcodewidget import QRCodeWidget
+import electrum_gui.bmp
+import json
+
 try:
     import zbar
 except ImportError:
@@ -11,10 +18,14 @@ except ImportError:
 from electrum_gui import BasePlugin
 class Plugin(BasePlugin):
 
-    def __init__(self, gui):
-        BasePlugin.__init__(self, gui, 'qrscans', 'QR scans', "QR Scans.\nInstall the zbar package (http://zbar.sourceforge.net/download.html) to enable this plugin")
+    def fullname(self): return 'QR scans'
+
+    def description(self): return "QR Scans.\nInstall the zbar package (http://zbar.sourceforge.net/download.html) to enable this plugin"
+
+    def __init__(self, gui, name):
+        BasePlugin.__init__(self, gui, name)
         self._is_available = self._init()
-        
+
     def _init(self):
         if not zbar:
             return False
@@ -34,6 +45,16 @@ class Plugin(BasePlugin):
         b = QPushButton(_("Scan QR code"))
         b.clicked.connect(self.fill_from_qr)
         grid.addWidget(b, 1, 5)
+        b2 = QPushButton(_("Scan TxQR"))
+        b2.clicked.connect(self.read_raw_qr)
+        
+        if not self.gui.wallet.seed:
+            b3 = QPushButton(_("Show unsigned TxQR"))
+            b3.clicked.connect(self.show_raw_qr)
+            grid.addWidget(b3, 7, 1)
+            grid.addWidget(b2, 7, 2)
+        else:
+            grid.addWidget(b2, 7, 1)
 
 
     def scan_qr(self):
@@ -51,11 +72,164 @@ class Plugin(BasePlugin):
             for r in proc.results:
                 if str(r.type) != 'QRCODE':
                     continue
-                return parse_uri(r.data)
+                return r.data
         
+    def show_raw_qr(self):
+        r = unicode( self.gui.payto_e.text() )
+        r = r.strip()
 
-    def fill_from_qr(self):
+        # label or alias, with address in brackets
+        m = re.match('(.*?)\s*\<([1-9A-HJ-NP-Za-km-z]{26,})\>', r)
+        to_address = m.group(2) if m else r
+
+        if not is_valid(to_address):
+            QMessageBox.warning(self.gui, _('Error'), _('Invalid Bitcoin Address') + ':\n' + to_address, _('OK'))
+            return
+
+        try:
+            amount = self.gui.read_amount(unicode( self.gui.amount_e.text()))
+        except:
+            QMessageBox.warning(self.gui, _('Error'), _('Invalid Amount'), _('OK'))
+            return
+        try:
+            fee = self.gui.read_amount(unicode( self.gui.fee_e.text()))
+        except:
+            QMessageBox.warning(self.gui, _('Error'), _('Invalid Fee'), _('OK'))
+            return
+
+        try:
+            tx = self.gui.wallet.mktx( [(to_address, amount)], None, fee, account=self.gui.current_account)
+        except BaseException, e:
+            self.gui.show_message(str(e))
+            return
+
+        if tx.requires_fee(self.gui.wallet.verifier) and fee < MIN_RELAY_TX_FEE:
+            QMessageBox.warning(self.gui, _('Error'), _("This transaction requires a higher fee, or it will not be propagated by the network."), _('OK'))
+            return
+
+        try:
+            out = {
+            "hex" : tx.hash(),
+            "complete" : "false"
+            }
+    
+            input_info = []
+
+        except BaseException, e:
+            self.gui.show_message(str(e))
+
+        try:
+            json_text = json.dumps(tx.as_dict()).replace(' ', '')
+            self.show_tx_qrcode(json_text, 'Unsigned Transaction')
+        except BaseException, e:
+            self.gui.show_message(str(e))
+
+    def show_tx_qrcode(self, data, title):
+        if not data: return
+        d = QDialog(self.gui)
+        d.setModal(1)
+        d.setWindowTitle(title)
+        d.setMinimumSize(250, 525)
+        vbox = QVBoxLayout()
+        qrw = QRCodeWidget(data)
+        vbox.addWidget(qrw, 0)
+        hbox = QHBoxLayout()
+        hbox.addStretch(1)
+
+        def print_qr(self):
+            filename = "qrcode.bmp"
+            electrum_gui.bmp.save_qrcode(qrw.qr, filename)
+            QMessageBox.information(None, _('Message'), _("QR code saved to file") + " " + filename, _('OK'))
+
+        b = QPushButton(_("Save"))
+        hbox.addWidget(b)
+        b.clicked.connect(print_qr)
+
+        b = QPushButton(_("Close"))
+        hbox.addWidget(b)
+        b.clicked.connect(d.accept)
+        b.setDefault(True)
+
+        vbox.addLayout(hbox, 1)
+        d.setLayout(vbox)
+        d.exec_()
+
+    def read_raw_qr(self):
         qrcode = self.scan_qr()
+        if qrcode:
+            tx_dict = self.gui.tx_dict_from_text(qrcode)
+            if tx_dict:
+                self.create_transaction_details_window(tx_dict)
+
+
+    def create_transaction_details_window(self, tx_dict):
+        tx = Transaction(tx_dict["hex"])
+            
+        dialog = QDialog(self.gui)
+        dialog.setMinimumWidth(500)
+        dialog.setWindowTitle(_('Process Offline transaction'))
+        dialog.setModal(1)
+
+        l = QGridLayout()
+        dialog.setLayout(l)
+
+        l.addWidget(QLabel(_("Transaction status:")), 3,0)
+        l.addWidget(QLabel(_("Actions")), 4,0)
+
+        if tx_dict["complete"] == False:
+            l.addWidget(QLabel(_("Unsigned")), 3,1)
+            if self.gui.wallet.seed :
+                b = QPushButton("Sign transaction")
+                input_info = json.loads(tx_dict["input_info"])
+                b.clicked.connect(lambda: self.sign_raw_transaction(tx, input_info, dialog))
+                l.addWidget(b, 4, 1)
+            else:
+                l.addWidget(QLabel(_("Wallet is de-seeded, can't sign.")), 4,1)
+        else:
+            l.addWidget(QLabel(_("Signed")), 3,1)
+            b = QPushButton("Broadcast transaction")
+            b.clicked.connect(lambda: self.gui.send_raw_transaction(tx, dialog))
+            l.addWidget(b,4,1)
+    
+        l.addWidget( self.gui.generate_transaction_information_widget(tx), 0,0,2,3)
+        closeButton = QPushButton(_("Close"))
+        closeButton.clicked.connect(lambda: dialog.done(0))
+        l.addWidget(closeButton, 4,2)
+
+        dialog.exec_()
+
+    def do_protect(self, func, args):
+        if self.gui.wallet.use_encryption:
+            password = self.gui.password_dialog()
+            if not password:
+                return
+        else:
+            password = None
+            
+        if args != (False,):
+            args = (self,) + args + (password,)
+        else:
+            args = (self,password)
+        apply( func, args)
+
+    def protected(func):
+        return lambda s, *args: s.do_protect(func, args)
+
+    @protected
+    def sign_raw_transaction(self, tx, input_info, dialog ="", password = ""):
+        try:
+            self.gui.wallet.signrawtransaction(tx, input_info, [], password)
+            txtext = json.dumps(tx.as_dict()).replace(' ', '')
+            self.show_tx_qrcode(txtext, 'Signed Transaction')
+        except BaseException, e:
+            self.gui.show_message(str(e))
+
+
+    def fill_from_qr(self):
+        qrcode = parse_uri(self.scan_qr())
+        if not qrcode:
+            return
+
         if 'address' in qrcode:
             self.gui.payto_e.setText(qrcode['address'])
         if 'amount' in qrcode: