3 # Electrum - lightweight Bitcoin client
4 # Copyright (C) 2012 thomasv@gitorious
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.
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.
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/>.
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
24 from PyQt4.QtGui import *
25 from PyQt4.QtCore import *
26 from electrum import DEFAULT_SERVERS, DEFAULT_PORTS
30 protocol_names = ['TCP', 'HTTP', 'SSL', 'HTTPS']
31 protocol_letters = 'thsg'
33 class NetworkDialog(QDialog):
34 def __init__(self, network, config, parent):
36 QDialog.__init__(self,parent)
38 self.setWindowTitle(_('Server'))
39 self.setMinimumSize(375, 20)
41 self.network = network
42 self.interface = interface = network.interface
47 n = len(network.interfaces)
49 status = _("Connected to %d servers")%n + ", %d "%(network.blockchain.height) + _("blocks")
51 status = _("Not connected")
53 if interface.is_connected:
54 status += "\n" + _("Main server:") + " %s"%(interface.host)
56 status += "\n" + _("Disconnected from main server")
59 server = interface.server
62 status = _("Please choose a server.") + "\n" + _("Select 'Cancel' if you are offline.")
63 server = interface.server
65 self.servers = network.get_servers()
73 l.setPixmap(QPixmap(":icons/network.png"))
76 hbox.addWidget(QLabel(status))
78 hbox.addWidget(HelpButton(_("As of version 1.9, Electrum connects to several servers in order to download block headers and find out the longest blockchain. However, your wallet addresses are sent to a single server, in order to receive your transaction history.")))
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)
93 self.server_protocol.addItems(protocol_names)
95 grid.addWidget(QLabel(_('Server') + ':'), 0, 0)
96 grid.addWidget(self.server_protocol, 0, 1)
97 grid.addWidget(self.server_host, 0, 2)
98 grid.addWidget(self.server_port, 0, 3)
100 self.server_protocol.connect(self.server_protocol, SIGNAL('currentIndexChanged(int)'), self.change_protocol)
102 label = _('Active Servers') if network.irc_servers else _('Default Servers')
103 self.servers_list_widget = QTreeWidget(parent)
104 self.servers_list_widget.setHeaderLabels( [ label, _('Limit') ] )
105 self.servers_list_widget.setMaximumHeight(150)
106 self.servers_list_widget.setColumnWidth(0, 240)
109 host, port, protocol = server.split(':')
110 self.set_protocol(protocol)
111 self.change_server(host, protocol)
113 self.set_protocol('s')
115 self.servers_list_widget.connect(self.servers_list_widget,
116 SIGNAL('currentItemChanged(QTreeWidgetItem*,QTreeWidgetItem*)'),
117 lambda x,y: self.server_changed(x))
118 grid.addWidget(self.servers_list_widget, 1, 1, 1, 3)
120 if not config.is_modifiable('server'):
121 for w in [self.server_host, self.server_port, self.server_protocol, self.servers_list_widget]: w.setEnabled(False)
124 self.autocycle_cb = QCheckBox(_('Try random servers if disconnected'))
125 self.autocycle_cb.setChecked(self.config.get('auto_cycle', True))
126 grid.addWidget(self.autocycle_cb, 3, 1, 3, 2)
127 if not self.config.is_modifiable('auto_cycle'): self.autocycle_cb.setEnabled(False)
130 self.proxy_mode = QComboBox()
131 self.proxy_host = QLineEdit()
132 self.proxy_host.setFixedWidth(200)
133 self.proxy_port = QLineEdit()
134 self.proxy_port.setFixedWidth(60)
135 self.proxy_mode.addItems(['NONE', 'SOCKS4', 'SOCKS5', 'HTTP'])
137 def check_for_disable(index = False):
138 if self.proxy_mode.currentText() != 'NONE':
139 self.proxy_host.setEnabled(True)
140 self.proxy_port.setEnabled(True)
142 self.proxy_host.setEnabled(False)
143 self.proxy_port.setEnabled(False)
146 self.proxy_mode.connect(self.proxy_mode, SIGNAL('currentIndexChanged(int)'), check_for_disable)
148 if not self.config.is_modifiable('proxy'):
149 for w in [self.proxy_host, self.proxy_port, self.proxy_mode]: w.setEnabled(False)
151 proxy_config = interface.proxy if interface.proxy else { "mode":"none", "host":"localhost", "port":"8080"}
152 self.proxy_mode.setCurrentIndex(self.proxy_mode.findText(str(proxy_config.get("mode").upper())))
153 self.proxy_host.setText(proxy_config.get("host"))
154 self.proxy_port.setText(proxy_config.get("port"))
156 grid.addWidget(QLabel(_('Proxy') + ':'), 2, 0)
157 grid.addWidget(self.proxy_mode, 2, 1)
158 grid.addWidget(self.proxy_host, 2, 2)
159 grid.addWidget(self.proxy_port, 2, 3)
162 vbox.addLayout(ok_cancel_buttons(self))
166 def init_servers_list(self):
167 self.servers_list_widget.clear()
168 for _host, d in self.servers.items():
169 if d.get(self.protocol):
170 pruning_level = d.get('pruning','')
171 self.servers_list_widget.addTopLevelItem(QTreeWidgetItem( [ _host, pruning_level ] ))
174 def set_protocol(self, protocol):
175 if protocol != self.protocol:
176 self.protocol = protocol
177 self.init_servers_list()
179 def change_protocol(self, index):
180 p = protocol_letters[index]
181 host = unicode(self.server_host.text())
182 pp = self.servers.get(host)
183 if p not in pp.keys():
186 self.server_host.setText( host )
187 self.server_port.setText( port )
190 def server_changed(self, x):
192 self.change_server(str(x.text(0)), self.protocol)
194 def change_server(self, host, protocol):
196 pp = self.servers.get(host, DEFAULT_PORTS)
198 port = pp.get(protocol)
199 if not port: protocol = None
204 port = pp.get(protocol)
206 protocol = pp.keys()[0]
207 port = pp.get(protocol)
209 self.server_host.setText( host )
210 self.server_port.setText( port )
211 self.server_protocol.setCurrentIndex(protocol_letters.index(protocol))
213 if not self.servers: return
214 for p in protocol_letters:
215 i = protocol_letters.index(p)
216 j = self.server_protocol.model().index(i,0)
217 #if p not in pp.keys(): # and self.interface.is_connected:
218 # self.server_protocol.model().setData(j, QVariant(0), Qt.UserRole-1)
220 # self.server_protocol.model().setData(j, QVariant(33), Qt.UserRole-1)
228 server = ':'.join([str( self.server_host.text() ),
229 str( self.server_port.text() ),
230 (protocol_letters[self.server_protocol.currentIndex()]) ])
232 if self.proxy_mode.currentText() != 'NONE':
233 proxy = { 'mode':str(self.proxy_mode.currentText()).lower(),
234 'host':str(self.proxy_host.text()),
235 'port':str(self.proxy_port.text()) }
239 self.config.set_key("proxy", proxy, True)
240 self.config.set_key("server", server, True)
241 self.network.set_proxy(proxy)
242 self.network.set_server(server)
244 self.config.set_key('auto_cycle', self.autocycle_cb.isChecked(), True)