fix history export (bug #338)
[electrum-nvc.git] / gui / qt / lite_window.py
1 import sys
2
3 # Let's do some dep checking and handle missing ones gracefully
4 try:
5     from PyQt4.QtCore import *
6     from PyQt4.QtGui import *
7     from PyQt4.Qt import Qt
8     import PyQt4.QtCore as QtCore
9
10 except ImportError:
11     print "You need to have PyQT installed to run Electrum in graphical mode."
12     print "If you have pip installed try 'sudo pip install pyqt' if you are on Debian/Ubuntu try 'sudo apt-get install python-qt4'."
13     sys.exit(0)
14
15 from decimal import Decimal as D
16 from electrum.util import get_resource_path as rsrc
17 from electrum.bitcoin import is_valid
18 from electrum.i18n import _
19 import decimal
20 import json
21 import os.path
22 import random
23 import re
24 import time
25 from electrum.wallet import Wallet, WalletStorage
26 import webbrowser
27 import history_widget
28 import receiving_widget
29 from electrum import util
30 import csv 
31 import datetime
32
33 from electrum.version import ELECTRUM_VERSION as electrum_version
34 from electrum.util import format_satoshis, age
35
36 from main_window import ElectrumWindow
37 import shutil
38
39 from util import *
40
41 bitcoin = lambda v: v * 100000000
42
43 def IconButton(filename, parent=None):
44     pixmap = QPixmap(filename)
45     icon = QIcon(pixmap)
46     return QPushButton(icon, "", parent)
47
48 class Timer(QThread):
49     def run(self):
50         while True:
51             self.emit(SIGNAL('timersignal'))
52             time.sleep(0.5)
53
54 def resize_line_edit_width(line_edit, text_input):
55     metrics = QFontMetrics(qApp.font())
56     # Create an extra character to add some space on the end
57     text_input += "A"
58     line_edit.setMinimumWidth(metrics.width(text_input))
59
60 def load_theme_name(theme_path):
61     try:
62         with open(os.path.join(theme_path, "name.cfg")) as name_cfg_file:
63             return name_cfg_file.read().rstrip("\n").strip()
64     except IOError:
65         return None
66
67
68 def theme_dirs_from_prefix(prefix):
69     if not os.path.exists(prefix):
70         return []
71     theme_paths = {}
72     for potential_theme in os.listdir(prefix):
73         theme_full_path = os.path.join(prefix, potential_theme)
74         theme_css = os.path.join(theme_full_path, "style.css")
75         if not os.path.exists(theme_css):
76             continue
77         theme_name = load_theme_name(theme_full_path)
78         if theme_name is None:
79             continue
80         theme_paths[theme_name] = prefix, potential_theme
81     return theme_paths
82
83 def load_theme_paths():
84     theme_paths = {}
85     prefixes = (util.local_data_dir(), util.appdata_dir())
86     for prefix in prefixes:
87         theme_paths.update(theme_dirs_from_prefix(prefix))
88     return theme_paths
89
90
91 def csv_transaction(wallet):
92     try:
93         select_export = _('Select file to export your wallet transactions to')
94         fileName = QFileDialog.getSaveFileName(QWidget(), select_export, os.path.expanduser('~/electrum-history.csv'), "*.csv")
95         if fileName:
96             with open(fileName, "w+") as csvfile:
97                 transaction = csv.writer(csvfile)
98                 transaction.writerow(["transaction_hash","label", "confirmations", "value", "fee", "balance", "timestamp"])
99                 for item in wallet.get_tx_history():
100                     tx_hash, confirmations, is_mine, value, fee, balance, timestamp = item
101                     if confirmations:
102                         if timestamp is not None:
103                             try:
104                                 time_string = datetime.datetime.fromtimestamp(timestamp).isoformat(' ')[:-3]
105                             except [RuntimeError, TypeError, NameError] as reason:
106                                 time_string = "unknown"
107                                 pass
108                         else:
109                           time_string = "unknown"
110                     else:
111                         time_string = "pending"
112
113                     if value is not None:
114                         value_string = format_satoshis(value, True)
115                     else:
116                         value_string = '--'
117
118                     if fee is not None:
119                         fee_string = format_satoshis(fee, True)
120                     else:
121                         fee_string = '0'
122
123                     if tx_hash:
124                         label, is_default_label = wallet.get_label(tx_hash)
125                     else:
126                       label = ""
127
128                     balance_string = format_satoshis(balance, False)
129                     transaction.writerow([tx_hash, label, confirmations, value_string, fee_string, balance_string, time_string])
130                 QMessageBox.information(None,"CSV Export created", "Your CSV export has been successfully created.")
131     except (IOError, os.error), reason:
132         export_error_label = _("Electrum was unable to produce a transaction export.")
133         QMessageBox.critical(None,_("Unable to create csv"), export_error_label + "\n" + str(reason))
134
135
136
137 class TransactionWindow(QDialog):
138
139     def set_label(self):
140         label = unicode(self.label_edit.text())
141         self.parent.wallet.labels[self.tx_id] = label
142
143         super(TransactionWindow, self).accept() 
144
145     def __init__(self, transaction_id, parent):
146         super(TransactionWindow, self).__init__()
147
148         self.tx_id = str(transaction_id)
149         self.parent = parent
150
151         self.setModal(True)
152         self.resize(200,100)
153         self.setWindowTitle(_("Transaction successfully sent"))
154
155         self.layout = QGridLayout(self)
156         history_label = "%s\n%s" % (_("Your transaction has been sent."), _("Please enter a label for this transaction for future reference."))
157         self.layout.addWidget(QLabel(history_label))
158
159         self.label_edit = QLineEdit()
160         self.label_edit.setPlaceholderText(_("Transaction label"))
161         self.label_edit.setObjectName("label_input")
162         self.label_edit.setAttribute(Qt.WA_MacShowFocusRect, 0)
163         self.label_edit.setFocusPolicy(Qt.ClickFocus)
164         self.layout.addWidget(self.label_edit)
165
166         self.save_button = QPushButton(_("Save"))
167         self.layout.addWidget(self.save_button)
168         self.save_button.clicked.connect(self.set_label)
169
170         self.exec_()
171
172 class MiniWindow(QDialog):
173
174     def __init__(self, actuator, expand_callback, config):
175         super(MiniWindow, self).__init__()
176
177         self.actuator = actuator
178         self.config = config
179         self.btc_balance = None
180         self.quote_currencies = ["BRL", "CNY", "EUR", "GBP", "RUB", "USD"]
181         self.actuator.set_configured_currency(self.set_quote_currency)
182
183         # Needed because price discovery is done in a different thread
184         # which needs to be sent back to this main one to update the GUI
185         self.connect(self, SIGNAL("refresh_balance()"), self.refresh_balance)
186
187         self.balance_label = BalanceLabel(self.change_quote_currency, self)
188         self.balance_label.setObjectName("balance_label")
189
190
191         # Bitcoin address code
192         self.address_input = QLineEdit()
193         self.address_input.setPlaceholderText(_("Enter a Bitcoin address or contact"))
194         self.address_input.setObjectName("address_input")
195
196         self.address_input.setFocusPolicy(Qt.ClickFocus)
197
198         self.address_input.textChanged.connect(self.address_field_changed)
199         resize_line_edit_width(self.address_input,
200                                "1BtaFUr3qVvAmwrsuDuu5zk6e4s2rxd2Gy")
201
202         self.address_completions = QStringListModel()
203         address_completer = QCompleter(self.address_input)
204         address_completer.setCaseSensitivity(False)
205         address_completer.setModel(self.address_completions)
206         self.address_input.setCompleter(address_completer)
207
208         address_layout = QHBoxLayout()
209         address_layout.addWidget(self.address_input)
210
211         self.amount_input = QLineEdit()
212         self.amount_input.setPlaceholderText(_("... and amount") + " (%s)"%self.actuator.g.base_unit())
213         self.amount_input.setObjectName("amount_input")
214
215         self.amount_input.setFocusPolicy(Qt.ClickFocus)
216         # This is changed according to the user's displayed balance
217         self.amount_validator = QDoubleValidator(self.amount_input)
218         self.amount_validator.setNotation(QDoubleValidator.StandardNotation)
219         self.amount_validator.setDecimals(8)
220         self.amount_input.setValidator(self.amount_validator)
221
222         # This removes the very ugly OSX highlighting, please leave this in :D
223         self.address_input.setAttribute(Qt.WA_MacShowFocusRect, 0)
224         self.amount_input.setAttribute(Qt.WA_MacShowFocusRect, 0)
225         self.amount_input.textChanged.connect(self.amount_input_changed)
226
227         #if self.actuator.g.wallet.seed:
228         self.send_button = QPushButton(_("&Send"))
229         #else:
230         #    self.send_button = QPushButton(_("&Create"))
231
232         self.send_button.setObjectName("send_button")
233         self.send_button.setDisabled(True);
234         self.send_button.clicked.connect(self.send)
235
236         # Creating the receive button
237         self.switch_button = QPushButton( QIcon(":icons/switchgui.png"),'' )
238         self.switch_button.setMaximumWidth(25)
239         self.switch_button.setFlat(True)
240         self.switch_button.clicked.connect(expand_callback)
241
242         main_layout = QGridLayout(self)
243
244         main_layout.addWidget(self.balance_label, 0, 0, 1, 3)
245         main_layout.addWidget(self.switch_button, 0, 3)
246
247         main_layout.addWidget(self.address_input, 1, 0, 1, 4)
248         main_layout.addWidget(self.amount_input, 2, 0, 1, 2)
249         main_layout.addWidget(self.send_button, 2, 2, 1, 2)
250
251         self.send_button.setMaximumWidth(125)
252
253         self.history_list = history_widget.HistoryWidget()
254         self.history_list.setObjectName("history")
255         self.history_list.hide()
256         self.history_list.setAlternatingRowColors(True)
257
258         main_layout.addWidget(self.history_list, 3, 0, 1, 4)
259
260         self.receiving = receiving_widget.ReceivingWidget(self)
261         self.receiving.setObjectName("receiving")
262
263         # Add to the right side 
264         self.receiving_box = QGroupBox(_("Select a receiving address"))
265         extra_layout = QGridLayout()
266
267         # Checkbox to filter used addresses
268         hide_used = QCheckBox(_('Hide used addresses'))
269         hide_used.setChecked(True)
270         hide_used.stateChanged.connect(self.receiving.toggle_used)
271
272         # Events for receiving addresses
273         self.receiving.clicked.connect(self.receiving.copy_address)
274         self.receiving.itemDoubleClicked.connect(self.receiving.edit_label)
275         self.receiving.itemChanged.connect(self.receiving.update_label)
276
277
278         # Label
279         extra_layout.addWidget( QLabel(_('Selecting an address will copy it to the clipboard.') + '\n' + _('Double clicking the label will allow you to edit it.') ),0,0)
280
281         extra_layout.addWidget(self.receiving, 1,0)
282         extra_layout.addWidget(hide_used, 2,0)
283         extra_layout.setColumnMinimumWidth(0,200)
284
285         self.receiving_box.setLayout(extra_layout)
286         main_layout.addWidget(self.receiving_box,0,4,-1,3)
287         self.receiving_box.hide()
288
289         self.main_layout = main_layout
290
291         quit_shortcut = QShortcut(QKeySequence("Ctrl+Q"), self)
292         quit_shortcut.activated.connect(self.close)
293         close_shortcut = QShortcut(QKeySequence("Ctrl+W"), self)
294         close_shortcut.activated.connect(self.close)
295
296         g = self.config.get("winpos-lite",[4, 25, 351, 149])
297         self.setGeometry(g[0], g[1], g[2], g[3])
298
299         show_hist = self.config.get("gui_show_history",False)
300         self.show_history(show_hist)
301         show_hist = self.config.get("gui_show_receiving",False)
302         self.toggle_receiving_layout(show_hist)
303         
304         self.setWindowIcon(QIcon(":icons/electrum.png"))
305         self.setWindowTitle("Electrum")
306         self.setWindowFlags(Qt.Window|Qt.MSWindowsFixedSizeDialogHint)
307         self.layout().setSizeConstraint(QLayout.SetFixedSize)
308         self.setObjectName("main_window")
309
310
311     def context_menu(self):
312         view_menu = QMenu()
313         themes_menu = view_menu.addMenu(_("&Themes"))
314         selected_theme = self.actuator.selected_theme()
315         theme_group = QActionGroup(self)
316         for theme_name in self.actuator.theme_names():
317             theme_action = themes_menu.addAction(theme_name)
318             theme_action.setCheckable(True)
319             if selected_theme == theme_name:
320                 theme_action.setChecked(True)
321             class SelectThemeFunctor:
322                 def __init__(self, theme_name, toggle_theme):
323                     self.theme_name = theme_name
324                     self.toggle_theme = toggle_theme
325                 def __call__(self, checked):
326                     if checked:
327                         self.toggle_theme(self.theme_name)
328             delegate = SelectThemeFunctor(theme_name, self.toggle_theme)
329             theme_action.toggled.connect(delegate)
330             theme_group.addAction(theme_action)
331         view_menu.addSeparator()
332
333         show_receiving = view_menu.addAction(_("Show Receiving addresses"))
334         show_receiving.setCheckable(True)
335         show_receiving.toggled.connect(self.toggle_receiving_layout)
336         show_receiving.setChecked(self.config.get("gui_show_receiving",False))
337
338         show_history = view_menu.addAction(_("Show History"))
339         show_history.setCheckable(True)
340         show_history.toggled.connect(self.show_history)
341         show_history.setChecked(self.config.get("gui_show_history",False))
342
343         return view_menu
344
345
346
347     def toggle_theme(self, theme_name):
348         old_path = QDir.currentPath()
349         self.actuator.change_theme(theme_name)
350         # Recompute style globally
351         qApp.style().unpolish(self)
352         qApp.style().polish(self)
353         QDir.setCurrent(old_path)
354
355     def closeEvent(self, event):
356         g = self.geometry()
357         self.config.set_key("winpos-lite", [g.left(),g.top(),g.width(),g.height()],True)
358         
359         super(MiniWindow, self).closeEvent(event)
360         qApp.quit()
361
362     def set_payment_fields(self, dest_address, amount):
363         self.address_input.setText(dest_address)
364         self.address_field_changed(dest_address)
365         self.amount_input.setText(amount)
366
367     def activate(self):
368         pass
369
370     def deactivate(self):
371         pass
372
373     def set_quote_currency(self, currency):
374         """Set and display the fiat currency country."""
375         if currency not in self.quote_currencies:
376             return
377         self.quote_currencies.remove(currency)
378         self.quote_currencies.insert(0, currency)
379         self.refresh_balance()
380
381     def change_quote_currency(self, forward=True):
382         if forward:
383             self.quote_currencies = \
384                 self.quote_currencies[1:] + self.quote_currencies[0:1]
385         else:
386             self.quote_currencies = \
387                 self.quote_currencies[-1:] + self.quote_currencies[0:-1]
388         self.actuator.set_config_currency(self.quote_currencies[0])
389         self.refresh_balance()
390
391     def refresh_balance(self):
392         if self.btc_balance is None:
393             # Price has been discovered before wallet has been loaded
394             # and server connect... so bail.
395             return
396         self.set_balances(self.btc_balance)
397         self.amount_input_changed(self.amount_input.text())
398
399     def set_balances(self, btc_balance):
400         """Set the bitcoin balance and update the amount label accordingly."""
401         self.btc_balance = btc_balance
402         quote_text = self.create_quote_text(btc_balance)
403         if quote_text:
404             quote_text = "(%s)" % quote_text
405
406         amount = self.actuator.g.format_amount(btc_balance)
407         unit = self.actuator.g.base_unit()
408
409         self.balance_label.set_balance_text(amount, unit, quote_text)
410         self.setWindowTitle("Electrum %s - %s %s" % (electrum_version, amount, unit))
411
412     def amount_input_changed(self, amount_text):
413         """Update the number of bitcoins displayed."""
414         self.check_button_status()
415
416         try:
417             amount = D(str(amount_text)) * (10**self.actuator.g.decimal_point)
418         except decimal.InvalidOperation:
419             self.balance_label.show_balance()
420         else:
421             quote_text = self.create_quote_text(amount)
422             if quote_text:
423                 self.balance_label.set_amount_text(quote_text)
424                 self.balance_label.show_amount()
425             else:
426                 self.balance_label.show_balance()
427
428     def create_quote_text(self, btc_balance):
429         """Return a string copy of the amount fiat currency the 
430         user has in bitcoins."""
431         from electrum.plugins import run_hook
432         r = {}
433         run_hook('set_quote_text', btc_balance, r)
434         return r.get(0,'')
435
436     def send(self):
437         if self.actuator.send(self.address_input.text(),
438                               self.amount_input.text(), self):
439             self.address_input.setText("")
440             self.amount_input.setText("")
441
442     def check_button_status(self):
443         """Check that the bitcoin address is valid and that something
444         is entered in the amount before making the send button clickable."""
445         try:
446             value = D(str(self.amount_input.text())) * (10**self.actuator.g.decimal_point)
447         except decimal.InvalidOperation:
448             value = None
449         # self.address_input.property(...) returns a qVariant, not a bool.
450         # The == is needed to properly invoke a comparison.
451         if (self.address_input.property("isValid") == True and
452             value is not None and 0 < value <= self.btc_balance):
453             self.send_button.setDisabled(False)
454         else:
455             self.send_button.setDisabled(True)
456
457     def address_field_changed(self, address):
458         # label or alias, with address in brackets
459         match2 = re.match("(.*?)\s*\<([1-9A-HJ-NP-Za-km-z]{26,})\>",
460                           address)
461         if match2:
462           address = match2.group(2)
463           self.address_input.setText(address)
464
465         if is_valid(address):
466             self.check_button_status()
467             self.address_input.setProperty("isValid", True)
468             self.recompute_style(self.address_input)
469         else:
470             self.send_button.setDisabled(True)
471             self.address_input.setProperty("isValid", False)
472             self.recompute_style(self.address_input)
473
474         if len(address) == 0:
475             self.address_input.setProperty("isValid", None)
476             self.recompute_style(self.address_input)
477
478     def recompute_style(self, element):
479         self.style().unpolish(element)
480         self.style().polish(element)
481
482     def copy_address(self):
483         receive_popup = ReceivePopup(self.receive_button)
484         self.actuator.copy_address(receive_popup)
485
486     def update_completions(self, completions):
487         self.address_completions.setStringList(completions)
488  
489
490     def update_history(self, tx_history):
491
492         self.history_list.empty()
493
494         for item in tx_history[-10:]:
495             tx_hash, conf, is_mine, value, fee, balance, timestamp = item
496             label = self.actuator.g.wallet.get_label(tx_hash)[0]
497             v_str = self.actuator.g.format_amount(value, True)
498             self.history_list.append(label, v_str, age(timestamp))
499
500
501     def the_website(self):
502         webbrowser.open("http://electrum.org")
503
504
505     def toggle_receiving_layout(self, toggle_state):
506         if toggle_state:
507             self.receiving_box.show()
508         else:
509             self.receiving_box.hide()
510         self.config.set_key("gui_show_receiving", toggle_state)
511
512     def show_history(self, toggle_state):
513         if toggle_state:
514             self.main_layout.setRowMinimumHeight(3,200)
515             self.history_list.show()
516         else:
517             self.main_layout.setRowMinimumHeight(3,0)
518             self.history_list.hide()
519         self.config.set_key("gui_show_history", toggle_state)
520
521 class BalanceLabel(QLabel):
522
523     SHOW_CONNECTING = 1
524     SHOW_BALANCE = 2
525     SHOW_AMOUNT = 3
526
527     def __init__(self, change_quote_currency, parent=None):
528         super(QLabel, self).__init__(_("Connecting..."), parent)
529         self.change_quote_currency = change_quote_currency
530         self.state = self.SHOW_CONNECTING
531         self.balance_text = ""
532         self.amount_text = ""
533         self.parent = parent
534
535     def mousePressEvent(self, event):
536         """Change the fiat currency selection if window background is clicked."""
537         if self.state != self.SHOW_CONNECTING:
538             if event.button() == Qt.LeftButton:
539                 self.change_quote_currency()
540             else:
541                 position = event.globalPos()
542                 menu = self.parent.context_menu()
543                 menu.exec_(position)
544                 
545
546     def set_balance_text(self, amount, unit, quote_text):
547         """Set the amount of bitcoins in the gui."""
548         if self.state == self.SHOW_CONNECTING:
549             self.state = self.SHOW_BALANCE
550
551         self.balance_text = "<span style='font-size: 18pt'>%s</span>"%amount\
552             + " <span style='font-size: 10pt'>%s</span>" % unit \
553             + " <span style='font-size: 10pt'>%s</span>" % quote_text
554
555         if self.state == self.SHOW_BALANCE:
556             self.setText(self.balance_text)
557
558     def set_amount_text(self, quote_text):
559         self.amount_text = "<span style='font-size: 10pt'>%s</span>" % quote_text
560         if self.state == self.SHOW_AMOUNT:
561             self.setText(self.amount_text)
562
563     def show_balance(self):
564         if self.state == self.SHOW_AMOUNT:
565             self.state = self.SHOW_BALANCE
566             self.setText(self.balance_text)
567
568     def show_amount(self):
569         if self.state == self.SHOW_BALANCE:
570             self.state = self.SHOW_AMOUNT
571             self.setText(self.amount_text)
572
573 def ok_cancel_buttons(dialog):
574     row_layout = QHBoxLayout()
575     row_layout.addStretch(1)
576     ok_button = QPushButton(_("OK"))
577     row_layout.addWidget(ok_button)
578     ok_button.clicked.connect(dialog.accept)
579     cancel_button = QPushButton(_("Cancel"))
580     row_layout.addWidget(cancel_button)
581     cancel_button.clicked.connect(dialog.reject)
582     return row_layout
583
584 class PasswordDialog(QDialog):
585
586     def __init__(self, parent):
587         super(QDialog, self).__init__(parent)
588
589         self.setModal(True)
590
591         self.password_input = QLineEdit()
592         self.password_input.setEchoMode(QLineEdit.Password)
593
594         main_layout = QVBoxLayout(self)
595         message = _('Please enter your password')
596         main_layout.addWidget(QLabel(message))
597
598         grid = QGridLayout()
599         grid.setSpacing(8)
600         grid.addWidget(QLabel(_('Password')), 1, 0)
601         grid.addWidget(self.password_input, 1, 1)
602         main_layout.addLayout(grid)
603
604         main_layout.addLayout(ok_cancel_buttons(self))
605         self.setLayout(main_layout) 
606
607     def run(self):
608         if not self.exec_():
609             return
610         return unicode(self.password_input.text())
611
612 class ReceivePopup(QDialog):
613
614     def leaveEvent(self, event):
615         self.close()
616
617     def setup(self, address):
618         label = QLabel(_("Copied your Bitcoin address to the clipboard!"))
619         address_display = QLineEdit(address)
620         address_display.setReadOnly(True)
621         resize_line_edit_width(address_display, address)
622
623         main_layout = QVBoxLayout(self)
624         main_layout.addWidget(label)
625         main_layout.addWidget(address_display)
626
627         self.setMouseTracking(True)
628         self.setWindowTitle("Electrum - " + _("Receive Bitcoin payment"))
629         self.setWindowFlags(Qt.Window|Qt.FramelessWindowHint|
630                             Qt.MSWindowsFixedSizeDialogHint)
631         self.layout().setSizeConstraint(QLayout.SetFixedSize)
632         #self.setFrameStyle(QFrame.WinPanel|QFrame.Raised)
633         #self.setAlignment(Qt.AlignCenter)
634
635     def popup(self):
636         parent = self.parent()
637         top_left_pos = parent.mapToGlobal(parent.rect().bottomLeft())
638         self.move(top_left_pos)
639         center_mouse_pos = self.mapToGlobal(self.rect().center())
640         QCursor.setPos(center_mouse_pos)
641         self.show()
642
643 class MiniActuator:
644     """Initialize the definitions relating to themes and 
645     sending/receiving bitcoins."""
646     
647     
648     def __init__(self, main_window):
649         """Retrieve the gui theme used in previous session."""
650         self.g = main_window
651         self.theme_name = self.g.config.get('litegui_theme','Cleanlook')
652         self.themes = load_theme_paths()
653
654     def load_theme(self):
655         """Load theme retrieved from wallet file."""
656         try:
657             theme_prefix, theme_path = self.themes[self.theme_name]
658         except KeyError:
659             util.print_error("Theme not found!", self.theme_name)
660             return
661         QDir.setCurrent(os.path.join(theme_prefix, theme_path))
662         with open(rsrc("style.css")) as style_file:
663             qApp.setStyleSheet(style_file.read())
664
665     def theme_names(self):
666         """Sort themes."""
667         return sorted(self.themes.keys())
668     
669     def selected_theme(self):
670         """Select theme."""
671         return self.theme_name
672
673     def change_theme(self, theme_name):
674         """Change theme."""
675         self.theme_name = theme_name
676         self.g.config.set_key('litegui_theme',theme_name)
677         self.load_theme()
678     
679     def set_configured_currency(self, set_quote_currency):
680         """Set the inital fiat currency conversion country (USD/EUR/GBP) in 
681         the GUI to what it was set to in the wallet."""
682         currency = self.g.config.get('currency')
683         # currency can be none when Electrum is used for the first
684         # time and no setting has been created yet.
685         if currency is not None:
686             set_quote_currency(currency)
687
688     def set_config_currency(self, conversion_currency):
689         """Change the wallet fiat currency country."""
690         self.g.config.set_key('currency',conversion_currency,True)
691         self.g.update_status()
692
693     def copy_address(self, receive_popup):
694         """Copy the wallet addresses into the client."""
695         addrs = [addr for addr in self.g.wallet.addresses(True)
696                  if not self.g.wallet.is_change(addr)]
697         # Select most recent addresses from gap limit
698         addrs = addrs[-self.g.wallet.gap_limit:]
699         copied_address = random.choice(addrs)
700         qApp.clipboard().setText(copied_address)
701         receive_popup.setup(copied_address)
702         receive_popup.popup()
703
704     def waiting_dialog(self, f):
705         s = Timer()
706         s.start()
707         w = QDialog()
708         w.resize(200, 70)
709         w.setWindowTitle('Electrum')
710         l = QLabel(_('Sending transaction, please wait.'))
711         vbox = QVBoxLayout()
712         vbox.addWidget(l)
713         w.setLayout(vbox)
714         w.show()
715         def ff():
716             s = f()
717             if s: l.setText(s)
718             else: w.close()
719         w.connect(s, QtCore.SIGNAL('timersignal'), ff)
720         w.exec_()
721         w.destroy()
722
723
724     def send(self, address, amount, parent_window):
725         """Send bitcoins to the target address."""
726         dest_address = self.fetch_destination(address)
727
728         if dest_address is None or not is_valid(dest_address):
729             QMessageBox.warning(parent_window, _('Error'), 
730                 _('Invalid Bitcoin Address') + ':\n' + address, _('OK'))
731             return False
732
733         amount = D(unicode(amount)) * (10*self.g.decimal_point)
734         print "amount", amount
735         return
736
737         if self.g.wallet.use_encryption:
738             password_dialog = PasswordDialog(parent_window)
739             password = password_dialog.run()
740             if not password:
741                 return
742         else:
743             password = None
744
745         fee = 0
746         # 0.1 BTC = 10000000
747         if amount < bitcoin(1) / 10:
748             # 0.001 BTC
749             fee = bitcoin(1) / 1000
750
751         try:
752             tx = self.g.wallet.mktx([(dest_address, amount)], password, fee)
753         except BaseException as error:
754             QMessageBox.warning(parent_window, _('Error'), str(error), _('OK'))
755             return False
756
757         if tx.is_complete:
758             h = self.g.wallet.send_tx(tx)
759
760             self.waiting_dialog(lambda: False if self.g.wallet.tx_event.isSet() else _("Sending transaction, please wait..."))
761               
762             status, message = self.g.wallet.receive_tx(h)
763
764             if not status:
765                 import tempfile
766                 dumpf = tempfile.NamedTemporaryFile(delete=False)
767                 dumpf.write(tx)
768                 dumpf.close()
769                 print "Dumped error tx to", dumpf.name
770                 QMessageBox.warning(parent_window, _('Error'), message, _('OK'))
771                 return False
772           
773             TransactionWindow(message, self)
774         else:
775             filename = 'unsigned_tx_%s' % (time.mktime(time.gmtime()))
776             try:
777                 fileName = QFileDialog.getSaveFileName(QWidget(), _("Select a transaction filename"), os.path.expanduser('~/%s' % (filename)))
778                 with open(fileName,'w') as f:
779                     f.write(json.dumps(tx.as_dict(),indent=4) + '\n')
780                 QMessageBox.information(QWidget(), _('Unsigned transaction created'), _("Unsigned transaction was saved to file:") + " " +fileName, _('OK'))
781             except BaseException as e:
782                 QMessageBox.warning(QWidget(), _('Error'), _('Could not write transaction to file: %s' % e), _('OK'))
783         return True
784
785     def fetch_destination(self, address):
786         recipient = unicode(address).strip()
787
788         # alias
789         match1 = re.match("^(|([\w\-\.]+)@)((\w[\w\-]+\.)+[\w\-]+)$",
790                           recipient)
791
792         # label or alias, with address in brackets
793         match2 = re.match("(.*?)\s*\<([1-9A-HJ-NP-Za-km-z]{26,})\>",
794                           recipient)
795         
796         if match1:
797             dest_address = \
798                 self.g.wallet.get_alias(recipient, True, 
799                                       self.show_message, self.question)
800             return dest_address
801         elif match2:
802             return match2.group(2)
803         else:
804             return recipient
805
806
807         
808
809
810 class MiniDriver(QObject):
811
812     INITIALIZING = 0
813     CONNECTING = 1
814     SYNCHRONIZING = 2
815     READY = 3
816
817     def __init__(self, main_window, mini_window):
818         super(QObject, self).__init__()
819
820         self.g = main_window
821         self.network = main_window.network
822         self.window = mini_window
823
824         self.network.register_callback('updated',self.update_callback)
825         self.network.register_callback('connected', self.update_callback)
826         self.network.register_callback('disconnected', self.update_callback)
827
828         self.state = None
829
830         self.initializing()
831         self.connect(self, SIGNAL("updatesignal()"), self.update)
832         self.update_callback()
833
834     # This is a hack to workaround that Qt does not like changing the
835     # window properties from this other thread before the runloop has
836     # been called from.
837     def update_callback(self):
838         self.emit(SIGNAL("updatesignal()"))
839
840     def update(self):
841         if not self.network.interface:
842             self.initializing()
843         elif not self.network.interface.is_connected:
844             self.connecting()
845
846         if self.g.wallet is None:
847             self.ready()
848         elif not self.g.wallet.up_to_date:
849             self.synchronizing()
850         else:
851             self.ready()
852             self.update_balance()
853             self.update_completions()
854             self.update_history()
855             self.window.receiving.update_list()
856
857
858     def initializing(self):
859         if self.state == self.INITIALIZING:
860             return
861         self.state = self.INITIALIZING
862         self.window.deactivate()
863
864     def connecting(self):
865         if self.state == self.CONNECTING:
866             return
867         self.state = self.CONNECTING
868         self.window.deactivate()
869
870     def synchronizing(self):
871         if self.state == self.SYNCHRONIZING:
872             return
873         self.state = self.SYNCHRONIZING
874         self.window.deactivate()
875
876     def ready(self):
877         if self.state == self.READY:
878             return
879         self.state = self.READY
880         self.window.activate()
881
882     def update_balance(self):
883         conf_balance, unconf_balance = self.g.wallet.get_balance()
884         balance = D(conf_balance + unconf_balance)
885         self.window.set_balances(balance)
886
887     def update_completions(self):
888         completions = []
889         for addr, label in self.g.wallet.labels.items():
890             if addr in self.g.wallet.addressbook:
891                 completions.append("%s <%s>" % (label, addr))
892         self.window.update_completions(completions)
893
894     def update_history(self):
895         tx_history = self.g.wallet.get_tx_history()
896         self.window.update_history(tx_history)
897
898
899 if __name__ == "__main__":
900     app = QApplication(sys.argv)
901     with open(rsrc("style.css")) as style_file:
902         app.setStyleSheet(style_file.read())
903     mini = MiniWindow()
904     sys.exit(app.exec_())
905