gui/qt/network_dialog.py

   1 #!/usr/bin/env python
   2 #
   3 # Electrum - lightweight Bitcoin client
   4 # Copyright (C) 2012 thomasv@gitorious
   5 #
   6 # This program is free software: you can redistribute it and/or modify
   7 # it under the terms of the GNU General Public License as published by
   8 # the Free Software Foundation, either version 3 of the License, or
   9 # (at your option) any later version.
  10 #
  11 # This program is distributed in the hope that it will be useful,
  12 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  13 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  14 # GNU General Public License for more details.
  15 #
  16 # You should have received a copy of the GNU General Public License
  17 # along with this program. If not, see <http://www.gnu.org/licenses/>.
  18
  19 import sys, time, datetime, re, threading
  20 from electrum.i18n import _
  21 from electrum.util import print_error, print_msg
  22 import os.path, json, ast, traceback
  23
  24 from PyQt4.QtGui import *
  25 from PyQt4.QtCore import *
  26 from electrum import DEFAULT_SERVERS, DEFAULT_PORTS
  27
  28 from util import *
  29
  30 protocol_names = ['TCP', 'HTTP', 'SSL', 'HTTPS']
  31 protocol_letters = 'thsg'
  32
  33 class NetworkDialog(QDialog):
  34     def __init__(self, network, config, parent):
  35
  36         QDialog.__init__(self,parent)
  37         self.setModal(1)
  38         self.setWindowTitle(_('Network'))
  39         self.setMinimumSize(375, 20)
  40
  41         self.network = network
  42         self.config = config
  43         self.protocol = None
  44
  45         if parent:
  46             n = len(network.interfaces)
  47             if n:
  48                 status = _("Blockchain") + ": " + "%d "%(network.blockchain.height()) + _("blocks") +  ".\n" + _("Getting block headers from %d nodes.")%n
  49             else:
  50                 status = _("Not connected")
  51
  52             if network.is_connected():
  53                 status += "\n" + _("Server") + ": %s"%(network.interface.host)
  54             else:
  55                 status += "\n" + _("Disconnected from server")
  56
  57         else:
  58             import random
  59             status = _("Please choose a server.") + "\n" + _("Select 'Cancel' if you are offline.")
  60
  61         server = network.default_server
  62         self.servers = network.get_servers()
  63
  64
  65         vbox = QVBoxLayout()
  66         vbox.setSpacing(30)
  67
  68         hbox = QHBoxLayout()
  69         l = QLabel()
  70         l.setPixmap(QPixmap(":icons/network.png"))
  71         hbox.addStretch(10)
  72         hbox.addWidget(l)
  73         hbox.addWidget(QLabel(status))
  74         hbox.addStretch(50)
  75         msg = _("Electrum sends your wallet addresses to a single server, in order to receive your transaction history.") + "\n\n" \
  76             + _("In addition, Electrum connects to several nodes in order to download block headers and find out the longest blockchain.") + " " \
  77             + _("This blockchain is used to verify the transactions sent by the address server.")
  78         hbox.addWidget(HelpButton(msg))
  79         vbox.addLayout(hbox)
  80
  81         # grid layout
  82         grid = QGridLayout()
  83         grid.setSpacing(8)
  84         vbox.addLayout(grid)
  85
  86         # protocol
  87         self.server_protocol = QComboBox()
  88         self.server_host = QLineEdit()
  89         self.server_host.setFixedWidth(200)
  90         self.server_port = QLineEdit()
  91         self.server_port.setFixedWidth(60)
  92         self.server_protocol.addItems(protocol_names)
  93         self.server_protocol.connect(self.server_protocol, SIGNAL('currentIndexChanged(int)'), self.change_protocol)
  94
  95         grid.addWidget(QLabel(_('Protocol') + ':'), 3, 0)
  96         grid.addWidget(self.server_protocol, 3, 1)
  97
  98
  99         # server
 100         grid.addWidget(QLabel(_('Server') + ':'), 0, 0)
 101
 102         # auto connect
 103         self.autocycle_cb = QCheckBox(_('Auto-connect'))
 104         self.autocycle_cb.setChecked(self.config.get('auto_cycle', True))
 105         grid.addWidget(self.autocycle_cb, 0, 1)
 106         if not self.config.is_modifiable('auto_cycle'): self.autocycle_cb.setEnabled(False)
 107         msg = _("If auto-connect is enabled, Electrum will always use a server that is on the longest blockchain.") + " " \
 108             + _("If it is disabled, Electrum will warn you if your server is lagging.")
 109         grid.addWidget(HelpButton(msg), 0, 4)
 110
 111         grid.addWidget(self.server_host, 0, 2, 1, 2)
 112         grid.addWidget(self.server_port, 0, 3)
 113
 114
 115         label = _('Active Servers') if network.irc_servers else _('Default Servers')
 116         self.servers_list_widget = QTreeWidget(parent)
 117         self.servers_list_widget.setHeaderLabels( [ label, _('Limit') ] )
 118         self.servers_list_widget.setMaximumHeight(150)
 119         self.servers_list_widget.setColumnWidth(0, 240)
 120
 121         if server:
 122             host, port, protocol = server.split(':')
 123             self.change_server(host, protocol)
 124
 125         self.set_protocol(self.network.protocol)
 126
 127         self.servers_list_widget.connect(self.servers_list_widget,
 128                                          SIGNAL('currentItemChanged(QTreeWidgetItem*,QTreeWidgetItem*)'),
 129                                          lambda x,y: self.server_changed(x))
 130         grid.addWidget(self.servers_list_widget, 1, 1, 1, 3)
 131
 132         if not config.is_modifiable('server'):
 133             for w in [self.server_host, self.server_port, self.server_protocol, self.servers_list_widget]: w.setEnabled(False)
 134
 135
 136
 137         def enable_set_server():
 138             enabled = not self.autocycle_cb.isChecked()
 139             self.server_host.setEnabled(enabled)
 140             self.server_port.setEnabled(enabled)
 141             self.servers_list_widget.setEnabled(enabled)
 142
 143         self.autocycle_cb.clicked.connect(enable_set_server)
 144         enable_set_server()
 145
 146         # proxy setting
 147         self.proxy_mode = QComboBox()
 148         self.proxy_host = QLineEdit()
 149         self.proxy_host.setFixedWidth(200)
 150         self.proxy_port = QLineEdit()
 151         self.proxy_port.setFixedWidth(60)
 152         self.proxy_mode.addItems(['NONE', 'SOCKS4', 'SOCKS5', 'HTTP'])
 153
 154         def check_for_disable(index = False):
 155             if self.proxy_mode.currentText() != 'NONE':
 156                 self.proxy_host.setEnabled(True)
 157                 self.proxy_port.setEnabled(True)
 158             else:
 159                 self.proxy_host.setEnabled(False)
 160                 self.proxy_port.setEnabled(False)
 161
 162         check_for_disable()
 163         self.proxy_mode.connect(self.proxy_mode, SIGNAL('currentIndexChanged(int)'), check_for_disable)
 164
 165         if not self.config.is_modifiable('proxy'):
 166             for w in [self.proxy_host, self.proxy_port, self.proxy_mode]: w.setEnabled(False)
 167
 168         proxy_config = network.proxy if network.proxy else { "mode":"none", "host":"localhost", "port":"8080"}
 169         self.proxy_mode.setCurrentIndex(self.proxy_mode.findText(str(proxy_config.get("mode").upper())))
 170         self.proxy_host.setText(proxy_config.get("host"))
 171         self.proxy_port.setText(proxy_config.get("port"))
 172
 173         grid.addWidget(QLabel(_('Proxy') + ':'), 4, 0)
 174         grid.addWidget(self.proxy_mode, 4, 1)
 175         grid.addWidget(self.proxy_host, 4, 2)
 176         grid.addWidget(self.proxy_port, 4, 3)
 177
 178         # buttons
 179         vbox.addLayout(ok_cancel_buttons(self))
 180         self.setLayout(vbox)
 181
 182
 183     def init_servers_list(self):
 184         self.servers_list_widget.clear()
 185         for _host, d in self.servers.items():
 186             if d.get(self.protocol):
 187                 pruning_level = d.get('pruning','')
 188                 self.servers_list_widget.addTopLevelItem(QTreeWidgetItem( [ _host, pruning_level ] ))
 189
 190
 191     def set_protocol(self, protocol):
 192         if protocol != self.protocol:
 193             self.protocol = protocol
 194             self.init_servers_list()
 195
 196     def change_protocol(self, index):
 197         p = protocol_letters[index]
 198         host = unicode(self.server_host.text())
 199         pp = self.servers.get(host, DEFAULT_PORTS)
 200         if p not in pp.keys():
 201             p = pp.keys()[0]
 202         port = pp[p]
 203         self.server_host.setText( host )
 204         self.server_port.setText( port )
 205         self.set_protocol(p)
 206
 207     def server_changed(self, x):
 208         if x:
 209             self.change_server(str(x.text(0)), self.protocol)
 210
 211     def change_server(self, host, protocol):
 212
 213         pp = self.servers.get(host, DEFAULT_PORTS)
 214         if protocol:
 215             port = pp.get(protocol)
 216             if not port: protocol = None
 217
 218         if not protocol:
 219             if 's' in pp.keys():
 220                 protocol = 's'
 221                 port = pp.get(protocol)
 222             else:
 223                 protocol = pp.keys()[0]
 224                 port = pp.get(protocol)
 225
 226         self.server_host.setText( host )
 227         self.server_port.setText( port )
 228         self.server_protocol.setCurrentIndex(protocol_letters.index(protocol))
 229
 230         if not self.servers: return
 231         for p in protocol_letters:
 232             i = protocol_letters.index(p)
 233             j = self.server_protocol.model().index(i,0)
 234             #if p not in pp.keys(): # and self.interface.is_connected:
 235             #    self.server_protocol.model().setData(j, QVariant(0), Qt.UserRole-1)
 236             #else:
 237             #    self.server_protocol.model().setData(j, QVariant(33), Qt.UserRole-1)
 238
 239
 240     def do_exec(self):
 241
 242         if not self.exec_():
 243             return
 244
 245         host = str( self.server_host.text() )
 246         port = str( self.server_port.text() )
 247         protocol = protocol_letters[self.server_protocol.currentIndex()]
 248
 249         if self.proxy_mode.currentText() != 'NONE':
 250             proxy = { 'mode':str(self.proxy_mode.currentText()).lower(),
 251                       'host':str(self.proxy_host.text()),
 252                       'port':str(self.proxy_port.text()) }
 253         else:
 254             proxy = None
 255
 256         auto_connect = self.autocycle_cb.isChecked()
 257
 258         self.network.set_parameters(host, port, protocol, proxy, auto_connect)
 259         return True