Exchange rate update
[electrum-nvc.git] / plugins / exchange_rate.py
index 9f850ff..64b0ec7 100644 (file)
@@ -6,14 +6,19 @@ import decimal
 import httplib
 import json
 import threading
+import time
 import re
 from decimal import Decimal
 from electrum.plugins import BasePlugin
 from electrum.i18n import _
 from electrum_gui.qt.util import *
+from electrum_gui.qt.amountedit import AmountEdit
 
 
 EXCHANGES = ["BitcoinAverage",
+             "BitcoinVenezuela",
+             "Bitcurex",
+             "Bitmarket",
              "BitPay",
              "Blockchain",
              "BTCChina",
@@ -22,7 +27,7 @@ EXCHANGES = ["BitcoinAverage",
              "CoinDesk",
              "LocalBitcoins",
              "Winkdex"]
-             
+
 
 class Exchanger(threading.Thread):
 
@@ -32,9 +37,10 @@ class Exchanger(threading.Thread):
         self.parent = parent
         self.quote_currencies = None
         self.lock = threading.Lock()
+        self.query_rates = threading.Event()
         self.use_exchange = self.parent.config.get('use_exchange', "Blockchain")
         self.parent.exchanges = EXCHANGES
-        self.parent.currencies = ["EUR","GBP","USD"]
+        self.parent.currencies = ["EUR","GBP","USD","PLN"]
         self.parent.win.emit(SIGNAL("refresh_exchanges_combo()"))
         self.parent.win.emit(SIGNAL("refresh_currencies_combo()"))
         self.is_running = False
@@ -68,7 +74,7 @@ class Exchanger(threading.Thread):
             except Exception:
                 return
             return btc_amount * decimal.Decimal(str(resp_rate["bpi"][str(quote_currency)]["rate_float"]))
-        return btc_amount * decimal.Decimal(quote_currencies[quote_currency])
+        return btc_amount * decimal.Decimal(str(quote_currencies[quote_currency]))
 
     def stop(self):
         self.is_running = False
@@ -77,6 +83,9 @@ class Exchanger(threading.Thread):
         self.use_exchange = self.parent.config.get('use_exchange', "Blockchain")
         update_rates = {
             "BitcoinAverage": self.update_ba,
+            "BitcoinVenezuela": self.update_bv,
+            "Bitcurex": self.update_bx,
+            "Bitmarket": self.update_bm,
             "BitPay": self.update_bp,
             "Blockchain": self.update_bc,
             "BTCChina": self.update_CNY,
@@ -94,8 +103,9 @@ class Exchanger(threading.Thread):
     def run(self):
         self.is_running = True
         while self.is_running:
+            self.query_rates.clear()
             self.update_rate()
-            time.sleep(150)
+            self.query_rates.wait(150)
 
 
     def update_cd(self):
@@ -110,7 +120,7 @@ class Exchanger(threading.Thread):
         with self.lock:
             self.quote_currencies = quote_currencies
         self.parent.set_currencies(quote_currencies)
-    
+
     def update_wd(self):
         try:
             winkresp = self.get_json('winkdex.com', "/static/data/0_600_288.json")
@@ -122,13 +132,13 @@ class Exchanger(threading.Thread):
         lenprices = len(winkresp["prices"])
         usdprice = winkresp["prices"][lenprices-1]["y"]
         try:
-            quote_currencies["USD"] = decimal.Decimal(usdprice)
+            quote_currencies["USD"] = decimal.Decimal(str(usdprice))
             with self.lock:
                 self.quote_currencies = quote_currencies
         except KeyError:
             pass
         self.parent.set_currencies(quote_currencies)
-            
+
     def update_cv(self):
         try:
             jsonresp = self.get_json('www.cavirtex.com', "/api/CAD/ticker.json")
@@ -137,7 +147,37 @@ class Exchanger(threading.Thread):
         quote_currencies = {"CAD": 0.0}
         cadprice = jsonresp["last"]
         try:
-            quote_currencies["CAD"] = decimal.Decimal(cadprice)
+            quote_currencies["CAD"] = decimal.Decimal(str(cadprice))
+            with self.lock:
+                self.quote_currencies = quote_currencies
+        except KeyError:
+            pass
+        self.parent.set_currencies(quote_currencies)
+
+    def update_bm(self):
+        try:
+            jsonresp = self.get_json('www.bitmarket.pl', "/json/BTCPLN/ticker.json")
+        except Exception:
+            return
+        quote_currencies = {"PLN": 0.0}
+        pln_price = jsonresp["last"]
+        try:
+            quote_currencies["PLN"] = decimal.Decimal(str(pln_price))
+            with self.lock:
+                self.quote_currencies = quote_currencies
+        except KeyError:
+            pass
+        self.parent.set_currencies(quote_currencies)
+
+    def update_bx(self):
+        try:
+            jsonresp = self.get_json('pln.bitcurex.com', "/data/ticker.json")
+        except Exception:
+            return
+        quote_currencies = {"PLN": 0.0}
+        pln_price = jsonresp["last"]
+        try:
+            quote_currencies["PLN"] = decimal.Decimal(str(pln_price))
             with self.lock:
                 self.quote_currencies = quote_currencies
         except KeyError:
@@ -152,7 +192,7 @@ class Exchanger(threading.Thread):
         quote_currencies = {"CNY": 0.0}
         cnyprice = jsonresp["ticker"]["last"]
         try:
-            quote_currencies["CNY"] = decimal.Decimal(cnyprice)
+            quote_currencies["CNY"] = decimal.Decimal(str(cnyprice))
             with self.lock:
                 self.quote_currencies = quote_currencies
         except KeyError:
@@ -222,7 +262,23 @@ class Exchanger(threading.Thread):
         except KeyError:
             pass
         self.parent.set_currencies(quote_currencies)
-                
+
+
+    def update_bv(self):
+        try:
+            jsonresp = self.get_json('api.bitcoinvenezuela.com', "/")
+        except Exception:
+            return
+        quote_currencies = {}
+        try:
+            for r in jsonresp["BTC"]:
+                quote_currencies[r] = Decimal(jsonresp["BTC"][r])
+            with self.lock:
+                self.quote_currencies = quote_currencies
+        except KeyError:
+            pass
+        self.parent.set_currencies(quote_currencies)
+
 
     def update_ba(self):
         try:
@@ -271,6 +327,7 @@ class Plugin(BasePlugin):
     def init(self):
         self.win = self.gui.main_window
         self.win.connect(self.win, SIGNAL("refresh_currencies()"), self.win.update_status)
+        self.btc_rate = Decimal("0.0")
         # Do price discovery
         self.exchanger = Exchanger(self)
         self.exchanger.start()
@@ -281,17 +338,41 @@ class Plugin(BasePlugin):
         self.win.emit(SIGNAL("refresh_currencies()"))
         self.win.emit(SIGNAL("refresh_currencies_combo()"))
 
-
-    def set_quote_text(self, btc_balance, r):
-        r[0] = self.create_quote_text(Decimal(btc_balance) / 100000000)
-
-    def create_quote_text(self, btc_balance):
+    def get_fiat_balance_text(self, btc_balance, r):
+        # return balance as: 1.23 USD
+        r[0] = self.create_fiat_balance_text(Decimal(btc_balance) / 100000000)
+
+    def get_fiat_price_text(self, r):
+        # return BTC price as: 123.45 USD
+        r[0] = self.create_fiat_balance_text(1)
+        quote = r[0]
+        if quote:
+            r[0] = "%s"%quote
+
+    def get_fiat_status_text(self, btc_balance, r2):
+        # return status as:   (1.23 USD)    1 BTC~123.45 USD
+        text = ""
+        r = {}
+        self.get_fiat_price_text(r)
+        quote = r.get(0)
+        if quote:
+            price_text = "1 BTC~%s"%quote
+            fiat_currency = quote[-3:]
+            btc_price = self.btc_rate
+            fiat_balance = Decimal(btc_price) * (Decimal(btc_balance)/100000000)
+            balance_text = "(%.2f %s)" % (fiat_balance,fiat_currency)
+            text = "  " + balance_text + "     " + price_text + " "
+        r2[0] = text
+
+    def create_fiat_balance_text(self, btc_balance):
         quote_currency = self.config.get("currency", "EUR")
         self.exchanger.use_exchange = self.config.get("use_exchange", "Blockchain")
-        quote_balance = self.exchanger.exchange(btc_balance, quote_currency)
-        if quote_balance is None:
+        cur_rate = self.exchanger.exchange(Decimal("1.0"), quote_currency)
+        if cur_rate is None:
             quote_text = ""
         else:
+            quote_balance = btc_balance * Decimal(cur_rate)
+            self.btc_rate = cur_rate
             quote_text = "%.2f %s" % (quote_balance, quote_currency)
         return quote_text
 
@@ -301,9 +382,9 @@ class Plugin(BasePlugin):
         for item in self.wallet.get_tx_history(self.wallet.storage.get("current_account", None)):
             tx_hash, conf, is_mine, value, fee, balance, timestamp = item
             tx_list[tx_hash] = {'value': value, 'timestamp': timestamp, 'balance': balance}
-            
+
         self.tx_list = tx_list
-        
+
 
     def requires_settings(self):
         return True
@@ -312,6 +393,11 @@ class Plugin(BasePlugin):
     def toggle(self):
         out = BasePlugin.toggle(self)
         self.win.update_status()
+        if self.config.get('use_exchange_rate'):
+            try:
+                self.fiat_button
+            except:
+                self.gui.main_window.show_message(_("To see fiat amount when sending bitcoin, please restart Electrum to activate the new GUI settings."))
         return out
 
 
@@ -320,14 +406,42 @@ class Plugin(BasePlugin):
 
     def history_tab_update(self):
         if self.config.get('history_rates', 'unchecked') == "checked":
-            tx_list = self.tx_list
-            
-            mintimestr = datetime.datetime.fromtimestamp(int(min(tx_list.items(), key=lambda x: x[1]['timestamp'])[1]['timestamp'])).strftime('%Y-%m-%d')
-            maxtimestr = datetime.datetime.now().strftime('%Y-%m-%d')
+            cur_exchange = self.config.get('use_exchange', "Blockchain")
+            try:
+                tx_list = self.tx_list
+            except Exception:
+                return
+
             try:
-                resp_hist = self.exchanger.get_json('api.coindesk.com', "/v1/bpi/historical/close.json?start=" + mintimestr + "&end=" + maxtimestr)
+                mintimestr = datetime.datetime.fromtimestamp(int(min(tx_list.items(), key=lambda x: x[1]['timestamp'])[1]['timestamp'])).strftime('%Y-%m-%d')
             except Exception:
                 return
+            maxtimestr = datetime.datetime.now().strftime('%Y-%m-%d')
+
+            if cur_exchange == "CoinDesk":
+                try:
+                    resp_hist = self.exchanger.get_json('api.coindesk.com', "/v1/bpi/historical/close.json?start=" + mintimestr + "&end=" + maxtimestr)
+                except Exception:
+                    return
+            elif cur_exchange == "Winkdex":
+                try:
+                    resp_hist = self.exchanger.get_json('winkdex.com', "/static/data/0_86400_730.json")['prices']
+                except Exception:
+                    return
+            elif cur_exchange == "BitcoinVenezuela":
+                cur_currency = self.config.get('currency', "EUR")
+                if cur_currency == "VEF":
+                    try:
+                        resp_hist = self.exchanger.get_json('api.bitcoinvenezuela.com', "/historical/index.php")['VEF_BTC']
+                    except Exception:
+                        return
+                elif cur_currency == "ARS":
+                    try:
+                        resp_hist = self.exchanger.get_json('api.bitcoinvenezuela.com', "/historical/index.php")['ARS_BTC']
+                    except Exception:
+                        return
+                else:
+                    return
 
             self.gui.main_window.is_edit = True
             self.gui.main_window.history_list.setColumnCount(6)
@@ -341,15 +455,36 @@ class Plugin(BasePlugin):
                 except Exception:
                     newtx = self.wallet.get_tx_history()
                     v = newtx[[x[0] for x in newtx].index(str(item.data(0, Qt.UserRole).toPyObject()))][3]
-                   
-                    tx_info = {'timestamp':int(datetime.datetime.now().strftime("%s")), 'value': v }
+
+                    tx_info = {'timestamp':int(time.time()), 'value': v }
                     pass
                 tx_time = int(tx_info['timestamp'])
-                tx_time_str = datetime.datetime.fromtimestamp(tx_time).strftime('%Y-%m-%d')
-                tx_USD_val = "%.2f %s" % (Decimal(tx_info['value']) / 100000000 * Decimal(resp_hist['bpi'][tx_time_str]), "USD")
-
-                item.setText(5, tx_USD_val)
-                if Decimal(tx_info['value']) < 0:
+                if cur_exchange == "CoinDesk":
+                    tx_time_str = datetime.datetime.fromtimestamp(tx_time).strftime('%Y-%m-%d')
+                    try:
+                        tx_USD_val = "%.2f %s" % (Decimal(str(tx_info['value'])) / 100000000 * Decimal(resp_hist['bpi'][tx_time_str]), "USD")
+                    except KeyError:
+                        tx_USD_val = "%.2f %s" % (self.btc_rate * Decimal(str(tx_info['value']))/100000000 , "USD")
+                elif cur_exchange == "Winkdex":
+                    tx_time_str = int(tx_time) - (int(tx_time) % (60 * 60 * 24))
+                    try:
+                        tx_rate = resp_hist[[x['x'] for x in resp_hist].index(tx_time_str)]['y']
+                        tx_USD_val = "%.2f %s" % (Decimal(tx_info['value']) / 100000000 * Decimal(tx_rate), "USD")
+                    except ValueError:
+                        tx_USD_val = "%.2f %s" % (self.btc_rate * Decimal(tx_info['value'])/100000000 , "USD")
+                elif cur_exchange == "BitcoinVenezuela":
+                    tx_time_str = datetime.datetime.fromtimestamp(tx_time).strftime('%Y-%m-%d')
+                    try:
+                        num = resp_hist[tx_time_str].replace(',','')
+                        tx_BTCVEN_val = "%.2f %s" % (Decimal(str(tx_info['value'])) / 100000000 * Decimal(num), cur_currency)
+                    except KeyError:
+                        tx_BTCVEN_val = _("No data")
+
+                if cur_exchange == "CoinDesk" or cur_exchange == "Winkdex":
+                    item.setText(5, tx_USD_val)
+                elif cur_exchange == "BitcoinVenezuela":
+                    item.setText(5, tx_BTCVEN_val)
+                if Decimal(str(tx_info['value'])) < 0:
                     item.setForeground(5, QBrush(QColor("#BC1E1E")))
 
             for i, width in enumerate(self.gui.main_window.column_widths['history']):
@@ -357,13 +492,14 @@ class Plugin(BasePlugin):
             self.gui.main_window.history_list.setColumnWidth(4, 140)
             self.gui.main_window.history_list.setColumnWidth(5, 120)
             self.gui.main_window.is_edit = False
-       
+
 
     def settings_widget(self, window):
         return EnterButton(_('Settings'), self.settings_dialog)
 
     def settings_dialog(self):
         d = QDialog()
+        d.setWindowTitle("Settings")
         layout = QGridLayout(d)
         layout.addWidget(QLabel(_('Exchange rate API: ')), 0, 0)
         layout.addWidget(QLabel(_('Currency: ')), 1, 0)
@@ -379,15 +515,29 @@ class Plugin(BasePlugin):
         ok_button = QPushButton(_("OK"))
 
         def on_change(x):
-            cur_request = str(self.currencies[x])
+            try:
+                cur_request = str(self.currencies[x])
+            except Exception:
+                return
             if cur_request != self.config.get('currency', "EUR"):
                 self.config.set_key('currency', cur_request, True)
-                if cur_request == "USD" and self.config.get('use_exchange', "Blockchain") == "CoinDesk":
+                cur_exchange = self.config.get('use_exchange', "Blockchain")
+                if cur_request == "USD" and (cur_exchange == "CoinDesk" or cur_exchange == "Winkdex"):
+                    hist_checkbox.setEnabled(True)
+                elif cur_request == "VEF" and (cur_exchange == "BitcoinVenezuela"):
+                    hist_checkbox.setEnabled(True)
+                elif cur_request == "ARS" and (cur_exchange == "BitcoinVenezuela"):
                     hist_checkbox.setEnabled(True)
                 else:
                     hist_checkbox.setChecked(False)
                     hist_checkbox.setEnabled(False)
                 self.win.update_status()
+                try:
+                    self.fiat_button
+                except:
+                    pass
+                else:
+                    self.fiat_button.setText(cur_request)
 
         def disable_check():
             hist_checkbox.setChecked(False)
@@ -397,9 +547,17 @@ class Plugin(BasePlugin):
             cur_request = str(self.exchanges[x])
             if cur_request != self.config.get('use_exchange', "Blockchain"):
                 self.config.set_key('use_exchange', cur_request, True)
-                self.exchanger.update_rate()
-                if cur_request == "CoinDesk":
-                    if self.config.get('currency', "EUR") == "USD":
+                self.currencies = []
+                combo.clear()
+                self.exchanger.query_rates.set()
+                cur_currency = self.config.get('currency', "EUR")
+                if cur_request == "CoinDesk" or cur_request == "Winkdex":
+                    if cur_currency == "USD":
+                        hist_checkbox.setEnabled(True)
+                    else:
+                        disable_check()
+                elif cur_request == "BitcoinVenezuela":
+                    if cur_currency == "VEF" or cur_currency == "ARS":
                         hist_checkbox.setEnabled(True)
                     else:
                         disable_check()
@@ -420,11 +578,14 @@ class Plugin(BasePlugin):
                     self.gui.main_window.history_list.setColumnWidth(i, width)
 
         def set_hist_check(hist_checkbox):
-            if self.config.get('use_exchange', "Blockchain") == "CoinDesk":
+            cur_exchange = self.config.get('use_exchange', "Blockchain")
+            if cur_exchange == "CoinDesk" or cur_exchange == "Winkdex":
+                hist_checkbox.setEnabled(True)
+            elif cur_exchange == "BitcoinVenezuela":
                 hist_checkbox.setEnabled(True)
             else:
-                hist_checkbox.setEnabled(False) 
-        
+                hist_checkbox.setEnabled(False)
+
         def set_currencies(combo):
             current_currency = self.config.get('currency', "EUR")
             try:
@@ -459,18 +620,73 @@ class Plugin(BasePlugin):
         combo.currentIndexChanged.connect(on_change)
         combo_ex.currentIndexChanged.connect(on_change_ex)
         hist_checkbox.stateChanged.connect(on_change_hist)
-        combo.connect(d, SIGNAL('refresh_currencies_combo()'), lambda: set_currencies(combo))
+        combo.connect(self.win, SIGNAL('refresh_currencies_combo()'), lambda: set_currencies(combo))
         combo_ex.connect(d, SIGNAL('refresh_exchanges_combo()'), lambda: set_exchanges(combo_ex))
         ok_button.clicked.connect(lambda: ok_clicked())
         layout.addWidget(combo,1,1)
         layout.addWidget(combo_ex,0,1)
         layout.addWidget(hist_checkbox,2,1)
         layout.addWidget(ok_button,3,1)
-        
+
         if d.exec_():
             return True
         else:
             return False
 
+    def fiat_unit(self):
+        quote_currency = self.config.get("currency", "???")
+        return quote_currency
+
+    def fiat_dialog(self):
+        if not self.config.get('use_exchange_rate'):
+          self.gui.main_window.show_message(_("To use this feature, first enable the exchange rate plugin."))
+          return
+
+        if not self.gui.main_window.network.is_connected():
+          self.gui.main_window.show_message(_("To use this feature, you must have a network connection."))
+          return
+
+        quote_currency = self.fiat_unit()
+
+        d = QDialog(self.gui.main_window)
+        d.setWindowTitle("Fiat")
+        vbox = QVBoxLayout(d)
+        text = "Amount to Send in " + quote_currency
+        vbox.addWidget(QLabel(_(text)+':'))
+
+        grid = QGridLayout()
+        fiat_e = AmountEdit(self.fiat_unit)
+        grid.addWidget(fiat_e, 1, 0)
+
+        r = {}
+        self.get_fiat_price_text(r)
+        quote = r.get(0)
+        if quote:
+          text = "1 BTC~%s"%quote
+          grid.addWidget(QLabel(_(text)), 4, 0, 3, 0)
+        else:
+            self.gui.main_window.show_message(_("Exchange rate not available.  Please check your network connection."))
+            return
+
+        vbox.addLayout(grid)
+        vbox.addLayout(ok_cancel_buttons(d))
+
+        if not d.exec_():
+            return
+
+        fiat = str(fiat_e.text())
+
+        if str(fiat) == "" or str(fiat) == ".":
+            fiat = "0"
 
-        
+        quote = quote[:-4]
+        btcamount = Decimal(fiat) / Decimal(quote)
+        if str(self.gui.main_window.base_unit()) == "mBTC":
+            btcamount = btcamount * 1000
+        quote = "%.8f"%btcamount
+        self.gui.main_window.amount_e.setText( quote )
+
+    def exchange_rate_button(self, grid):
+        quote_currency = self.config.get("currency", "EUR")
+        self.fiat_button = EnterButton(_(quote_currency), self.fiat_dialog)
+        grid.addWidget(self.fiat_button, 4, 3, Qt.AlignHCenter)