plugins/pointofsale.py

   1 import re
   2 import platform
   3 from decimal import Decimal
   4
   5 from PyQt4.QtGui import *
   6 from PyQt4.QtCore import *
   7 import PyQt4.QtCore as QtCore
   8 import PyQt4.QtGui as QtGui
   9
  10 from electrum_gui.qrcodewidget import QRCodeWidget
  11 from electrum_gui import bmp, pyqrnative
  12
  13 from electrum_gui.i18n import _
  14
  15
  16 if platform.system() == 'Windows':
  17     MONOSPACE_FONT = 'Lucida Console'
  18 elif platform.system() == 'Darwin':
  19     MONOSPACE_FONT = 'Monaco'
  20 else:
  21     MONOSPACE_FONT = 'monospace'
  22
  23 column_index = 3
  24
  25 class QR_Window(QWidget):
  26
  27     def __init__(self, exchanger):
  28         QWidget.__init__(self)
  29         self.exchanger = exchanger
  30         self.setWindowTitle('Electrum - '+_('Invoice'))
  31         self.setMinimumSize(800, 250)
  32         self.address = ''
  33         self.label = ''
  34         self.amount = 0
  35         self.setFocusPolicy(QtCore.Qt.NoFocus)
  36
  37         self.merchant_name = config.get('merchant_name', 'Invoice')
  38         main_box = QHBoxLayout()
  39
  40         self.qrw = QRCodeWidget()
  41         main_box.addWidget(self.qrw, 1)
  42
  43         vbox = QVBoxLayout()
  44         main_box.addLayout(vbox)
  45
  46         self.address_label = QLabel("")
  47         #self.address_label.setFont(QFont(MONOSPACE_FONT))
  48         vbox.addWidget(self.address_label)
  49
  50         self.label_label = QLabel("")
  51         vbox.addWidget(self.label_label)
  52
  53         self.amount_label = QLabel("")
  54         vbox.addWidget(self.amount_label)
  55
  56         vbox.addStretch(1)
  57         self.setLayout(main_box)
  58
  59
  60     def set_content(self, addr, label, amount, currency):
  61         self.address = addr
  62         address_text = "<span style='font-size: 18pt'>%s</span>" % addr if addr else ""
  63         self.address_label.setText(address_text)
  64
  65         if currency == 'BTC': currency = None
  66         amount_text = ''
  67         if amount:
  68             if currency:
  69                 self.amount = Decimal(amount) / self.exchanger.exchange(1, currency) if currency else amount
  70             else:
  71                 self.amount = Decimal(amount)
  72             self.amount = self.amount.quantize(Decimal('1.0000'))
  73
  74             if currency:
  75                 amount_text += "<span style='font-size: 18pt'>%s %s</span><br/>" % (amount, currency)
  76             amount_text += "<span style='font-size: 21pt'>%s</span> <span style='font-size: 16pt'>BTC</span> " % str(self.amount)
  77         self.amount_label.setText(amount_text)
  78
  79         self.label = label
  80         label_text = "<span style='font-size: 21pt'>%s</span>" % label if label else ""
  81         self.label_label.setText(label_text)
  82
  83         msg = 'bitcoin:'+self.address
  84         if self.amount is not None:
  85             msg += '?amount=%s'%(str( self.amount))
  86             if self.label is not None:
  87                 msg += '&label=%s'%(self.label)
  88         elif self.label is not None:
  89             msg += '?label=%s'%(self.label)
  90
  91         self.qrw.set_addr( msg )
  92
  93
  94
  95
  96
  97
  98 def get_info():
  99     return 'Point of Sale', _('Show QR code window and amounts requested for each address. Add menu item to request amount.')
 100
 101 def init(gui):
 102     gui.requested_amounts = gui.config.get('requested_amounts',{})
 103     gui.qr_window = None
 104
 105
 106
 107 enabled = False
 108
 109 def is_enabled():
 110     return False
 111
 112 def toggle(gui):
 113     global enabled
 114     enabled = not enabled
 115     toggle_QR_window(gui, enabled)
 116
 117     if enabled:
 118         gui.expert_mode = True
 119         gui.receive_list.setHeaderLabels([ _('Address'), _('Label'), _('Balance'), _('Request')])
 120
 121         gui.set_hook('item_changed', item_changed)
 122         gui.set_hook('current_item_changed', recv_changed)
 123         gui.set_hook('receive_menu', receive_menu)
 124         gui.set_hook('update_receive_item', update_receive_item)
 125         gui.set_hook('timer_actions', timer_actions)
 126         gui.set_hook('close_main_window', close_main_window)
 127     else:
 128         gui.receive_list.setHeaderLabels([ _('Address'), _('Label'), _('Balance'), _('Tx')])
 129         gui.unset_hook('item_changed', item_changed)
 130         gui.unset_hook('current_item_changed', recv_changed)
 131         gui.unset_hook('receive_menu', receive_menu)
 132         gui.unset_hook('update_receive_item', update_receive_item)
 133         gui.unset_hook('timer_actions', timer_actions)
 134         gui.unset_hook('close_main_window', close_main_window)
 135
 136
 137     return enabled
 138
 139
 140 def toggle_QR_window(self, show):
 141     if show and not self.qr_window:
 142         self.qr_window = QR_Window(self.exchanger)
 143         self.qr_window.setVisible(True)
 144         self.qr_window_geometry = self.qr_window.geometry()
 145         item = self.receive_list.currentItem()
 146         if item:
 147             address = str(item.text(1))
 148             label = self.wallet.labels.get(address)
 149             amount, currency = self.requested_amounts.get(address, (None, None))
 150             self.qr_window.set_content( address, label, amount, currency )
 151
 152     elif show and self.qr_window and not self.qr_window.isVisible():
 153         self.qr_window.setVisible(True)
 154         self.qr_window.setGeometry(self.qr_window_geometry)
 155
 156     elif not show and self.qr_window and self.qr_window.isVisible():
 157         self.qr_window_geometry = self.qr_window.geometry()
 158         self.qr_window.setVisible(False)
 159
 160
 161
 162
 163 def item_changed(self, item, column):
 164     if column == column_index:
 165         address = str( item.text(0) )
 166         text = str( item.text(column) )
 167         try:
 168             seq = self.wallet.get_address_index(address)
 169             index = seq[-1]
 170         except:
 171             print "cannot get index"
 172             return
 173
 174         text = text.strip().upper()
 175         print text
 176         m = re.match('^(\d+(|\.\d*))\s*(|BTC|EUR|USD|GBP|CNY|JPY|RUB|BRL)$', text)
 177         if m:
 178             amount = m.group(1)
 179             currency = m.group(3)
 180             if not currency:
 181                 currency = 'BTC'
 182             else:
 183                 currency = currency.upper()
 184
 185             self.requested_amounts[address] = (amount, currency)
 186             self.wallet.config.set_key('requested_amounts', self.requested_amounts, True)
 187
 188             label = self.wallet.labels.get(address)
 189             if label is None:
 190                 label = self.merchant_name + ' - %04d'%(index+1)
 191                 self.wallet.labels[address] = label
 192
 193             if self.qr_window:
 194                 self.qr_window.set_content( address, label, amount, currency )
 195
 196         else:
 197             item.setText(column,'')
 198             if address in self.requested_amounts:
 199                 self.requested_amounts.pop(address)
 200
 201         self.update_receive_item(self.receive_list.currentItem())
 202
 203
 204 def recv_changed(self, a):
 205     if a is not None and self.qr_window and self.qr_window.isVisible():
 206         address = str(a.text(0))
 207         label = self.wallet.labels.get(address)
 208         try:
 209             amount, currency = self.requested_amounts.get(address, (None, None))
 210         except:
 211             amount, currency = None, None
 212         self.qr_window.set_content( address, label, amount, currency )
 213
 214
 215
 216 def edit_amount(self):
 217     l = self.receive_list
 218     item = l.currentItem()
 219     item.setFlags(Qt.ItemIsEditable|Qt.ItemIsSelectable | Qt.ItemIsUserCheckable | Qt.ItemIsEnabled | Qt.ItemIsDragEnabled)
 220     l.editItem( item, column_index )
 221     item.setFlags(Qt.ItemIsSelectable | Qt.ItemIsUserCheckable | Qt.ItemIsEnabled | Qt.ItemIsDragEnabled)
 222
 223 def receive_menu(self, menu):
 224     menu.addAction(_("Request amount"), lambda: edit_amount(self))
 225
 226
 227 def update_receive_item(self, address, item):
 228     try:
 229         amount, currency = self.requested_amounts.get(address, (None, None))
 230     except:
 231         print "cannot get requested amount", address, self.requested_amounts.get(address)
 232         amount, currency = None, None
 233         self.requested_amounts.pop(address)
 234
 235     amount_str = amount + (' ' + currency if currency else '') if amount is not None  else ''
 236     item.setData(column_index,0,amount_str)
 237
 238
 239 def close_main_window(self):
 240     if self.qr_window:
 241         self.qr_window.close()
 242         self.qr_window = None
 243
 244
 245 def timer_actions(self):
 246     if self.qr_window:
 247         self.qr_window.qrw.update_qr()