gui/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 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.interface import DEFAULT_SERVERS, DEFAULT_PORTS
  27
  28 from qt_util import *
  29
  30 protocol_names = ['TCP', 'HTTP', 'SSL', 'HTTPS']
  31 protocol_letters = 'thsg'
  32
  33 class NetworkDialog(QDialog):
  34     def __init__(self, interface, config, parent):
  35
  36         QDialog.__init__(self,parent)
  37         self.setModal(1)
  38         self.setWindowTitle(_('Server'))
  39         self.setMinimumSize(375, 20)
  40
  41         self.interface = interface
  42         self.config = config
  43         self.protocol = None
  44
  45         if parent:
  46             if interface.is_connected:
  47                 status = _("Connected to")+" %s"%(interface.host) + "\n%d "%(parent.wallet.verifier.height)+_("blocks")
  48             else:
  49                 status = _("Not connected")
  50             server = interface.server
  51         else:
  52             import random
  53             status = _("Please choose a server.") + "\n" + _("Select 'Cancel' if you are offline.")
  54             server = interface.server
  55
  56         self.servers = interface.get_servers()
  57
  58         vbox = QVBoxLayout()
  59         vbox.setSpacing(30)
  60
  61         hbox = QHBoxLayout()
  62         l = QLabel()
  63         l.setPixmap(QPixmap(":icons/network.png"))
  64         hbox.addStretch(10)
  65         hbox.addWidget(l)
  66         hbox.addWidget(QLabel(status))
  67         hbox.addStretch(50)
  68         vbox.addLayout(hbox)
  69
  70         # grid layout
  71         grid = QGridLayout()
  72         grid.setSpacing(8)
  73         vbox.addLayout(grid)
  74
  75         # server
  76         self.server_protocol = QComboBox()
  77         self.server_host = QLineEdit()
  78         self.server_host.setFixedWidth(200)
  79         self.server_port = QLineEdit()
  80         self.server_port.setFixedWidth(60)
  81
  82         self.server_protocol.addItems(protocol_names)
  83
  84         grid.addWidget(QLabel(_('Server') + ':'), 0, 0)
  85         grid.addWidget(self.server_protocol, 0, 1)
  86         grid.addWidget(self.server_host, 0, 2)
  87         grid.addWidget(self.server_port, 0, 3)
  88
  89         self.server_protocol.connect(self.server_protocol, SIGNAL('currentIndexChanged(int)'), self.change_protocol)
  90
  91         label = _('Active Servers') if interface.servers else _('Default Servers')
  92         self.servers_list_widget = QTreeWidget(parent)
  93         self.servers_list_widget.setHeaderLabels( [ label, _('Limit') ] )
  94         self.servers_list_widget.setMaximumHeight(150)
  95         self.servers_list_widget.setColumnWidth(0, 240)
  96
  97         if server:
  98             host, port, protocol = server.split(':')
  99             self.set_protocol(protocol)
 100             self.change_server(host, protocol)
 101         else:
 102             self.set_protocol('s')
 103
 104         self.servers_list_widget.connect(self.servers_list_widget,
 105                                          SIGNAL('currentItemChanged(QTreeWidgetItem*,QTreeWidgetItem*)'),
 106                                          lambda x,y: self.server_changed(x))
 107         grid.addWidget(self.servers_list_widget, 1, 1, 1, 3)
 108
 109         if not config.is_modifiable('server'):
 110             for w in [self.server_host, self.server_port, self.server_protocol, self.servers_list_widget]: w.setEnabled(False)
 111
 112         # auto cycle
 113         self.autocycle_cb = QCheckBox(_('Try random servers if disconnected'))
 114         self.autocycle_cb.setChecked(self.config.get('auto_cycle', True))
 115         grid.addWidget(self.autocycle_cb, 3, 1, 3, 2)
 116         if not self.config.is_modifiable('auto_cycle'): self.autocycle_cb.setEnabled(False)
 117
 118         # proxy setting
 119         self.proxy_mode = QComboBox()
 120         self.proxy_host = QLineEdit()
 121         self.proxy_host.setFixedWidth(200)
 122         self.proxy_port = QLineEdit()
 123         self.proxy_port.setFixedWidth(60)
 124         self.proxy_mode.addItems(['NONE', 'SOCKS4', 'SOCKS5', 'HTTP'])
 125
 126         def check_for_disable(index = False):
 127             if self.proxy_mode.currentText() != 'NONE':
 128                 self.proxy_host.setEnabled(True)
 129                 self.proxy_port.setEnabled(True)
 130             else:
 131                 self.proxy_host.setEnabled(False)
 132                 self.proxy_port.setEnabled(False)
 133
 134         check_for_disable()
 135         self.proxy_mode.connect(self.proxy_mode, SIGNAL('currentIndexChanged(int)'), check_for_disable)
 136
 137         if not self.config.is_modifiable('proxy'):
 138             for w in [self.proxy_host, self.proxy_port, self.proxy_mode]: w.setEnabled(False)
 139
 140         proxy_config = interface.proxy if interface.proxy else { "mode":"none", "host":"localhost", "port":"8080"}
 141         self.proxy_mode.setCurrentIndex(self.proxy_mode.findText(str(proxy_config.get("mode").upper())))
 142         self.proxy_host.setText(proxy_config.get("host"))
 143         self.proxy_port.setText(proxy_config.get("port"))
 144
 145         grid.addWidget(QLabel(_('Proxy') + ':'), 2, 0)
 146         grid.addWidget(self.proxy_mode, 2, 1)
 147         grid.addWidget(self.proxy_host, 2, 2)
 148         grid.addWidget(self.proxy_port, 2, 3)
 149
 150         # buttons
 151         vbox.addLayout(ok_cancel_buttons(self))
 152         self.setLayout(vbox)
 153
 154
 155     def init_servers_list(self):
 156         self.servers_list_widget.clear()
 157         for _host, d in self.servers.items():
 158             if d.get(self.protocol):
 159                 pruning_level = d.get('pruning','')
 160                 self.servers_list_widget.addTopLevelItem(QTreeWidgetItem( [ _host, pruning_level ] ))
 161
 162
 163     def set_protocol(self, protocol):
 164         if protocol != self.protocol:
 165             self.protocol = protocol
 166             self.init_servers_list()
 167
 168     def change_protocol(self, index):
 169         p = protocol_letters[index]
 170         host = unicode(self.server_host.text())
 171         pp = self.servers.get(host)
 172         if p not in pp.keys():
 173             p = pp.keys()[0]
 174         port = pp[p]
 175         self.server_host.setText( host )
 176         self.server_port.setText( port )
 177         self.set_protocol(p)
 178
 179     def server_changed(self, x):
 180         if x:
 181             self.change_server(str(x.text(0)), self.protocol)
 182
 183     def change_server(self, host, protocol):
 184
 185         pp = self.servers.get(host, DEFAULT_PORTS)
 186         if protocol:
 187             port = pp.get(protocol)
 188             if not port: protocol = None
 189
 190         if not protocol:
 191             if 's' in pp.keys():
 192                 protocol = 's'
 193                 port = pp.get(protocol)
 194             else:
 195                 protocol = pp.keys()[0]
 196                 port = pp.get(protocol)
 197
 198         self.server_host.setText( host )
 199         self.server_port.setText( port )
 200         self.server_protocol.setCurrentIndex(protocol_letters.index(protocol))
 201
 202         if not self.servers: return
 203         for p in protocol_letters:
 204             i = protocol_letters.index(p)
 205             j = self.server_protocol.model().index(i,0)
 206             #if p not in pp.keys(): # and self.interface.is_connected:
 207             #    self.server_protocol.model().setData(j, QVariant(0), Qt.UserRole-1)
 208             #else:
 209             #    self.server_protocol.model().setData(j, QVariant(33), Qt.UserRole-1)
 210
 211
 212     def do_exec(self):
 213
 214         if not self.exec_():
 215             return
 216
 217         server = ':'.join([str( self.server_host.text() ),
 218                            str( self.server_port.text() ),
 219                            (protocol_letters[self.server_protocol.currentIndex()]) ])
 220
 221         if self.proxy_mode.currentText() != 'NONE':
 222             proxy = { 'mode':str(self.proxy_mode.currentText()).lower(),
 223                       'host':str(self.proxy_host.text()),
 224                       'port':str(self.proxy_port.text()) }
 225         else:
 226             proxy = None
 227
 228         self.config.set_key("proxy", proxy, True)
 229         self.config.set_key("server", server, True)
 230         self.interface.set_server(server, proxy)
 231         self.config.set_key('auto_cycle', self.autocycle_cb.isChecked(), True)
 232         return True