gui/gui_text.py

   1 import curses, datetime
   2 from decimal import Decimal
   3 _ = lambda x:x
   4 #from i18n import _
   5 from electrum.util import format_satoshis, set_verbosity
   6 from electrum.bitcoin import is_valid
   7
   8 from electrum import Wallet, WalletVerifier, WalletSynchronizer
   9
  10 import tty, sys
  11
  12
  13 class ElectrumGui:
  14
  15     def __init__(self, config, interface):
  16
  17         self.config = config
  18         found = self.config.wallet_file_exists
  19         if not found:
  20             print "Wallet not found. try 'electrum create'"
  21             exit()
  22
  23         wallet = Wallet(self.config)
  24         wallet.interface = interface
  25         self.wallet = wallet
  26
  27         verifier = WalletVerifier(interface, config)
  28         verifier.start()
  29         wallet.set_verifier(verifier)
  30         synchronizer = WalletSynchronizer(wallet, config)
  31         synchronizer.start()
  32
  33         self.stdscr = curses.initscr()
  34         curses.noecho()
  35         curses.cbreak()
  36         curses.start_color()
  37         curses.use_default_colors()
  38         curses.init_pair(1, curses.COLOR_WHITE, curses.COLOR_BLUE)
  39         curses.init_pair(2, curses.COLOR_WHITE, curses.COLOR_CYAN)
  40         self.stdscr.keypad(1)
  41         self.stdscr.border(0)
  42         self.maxy, self.maxx = self.stdscr.getmaxyx()
  43         self.set_cursor(0)
  44         self.w = curses.newwin(10, 50, 5, 5)
  45
  46         set_verbosity(False)
  47         self.tab = 0
  48         self.pos = 0
  49         self.popup_pos = 0
  50
  51         self.str_recipient = ""
  52         self.str_description = ""
  53         self.str_amount = ""
  54         self.str_fee = ""
  55
  56         self.wallet.interface.register_callback('updated', self.refresh)
  57         self.wallet.interface.register_callback('connected', self.refresh)
  58         self.wallet.interface.register_callback('disconnected', self.refresh)
  59         self.wallet.interface.register_callback('disconnecting', self.refresh)
  60         self.tab_names = [_("History"), _("Send"), _("Receive"), _("Contacts"), _("Wall")]
  61         self.num_tabs = len(self.tab_names)
  62
  63
  64     def set_cursor(self, x):
  65         try:
  66             curses.curs_set(x)
  67         except:
  68             pass
  69
  70     def restore_or_create(self):
  71         pass
  72
  73     def verify_seed(self):
  74         pass
  75
  76     def get_string(self, y, x):
  77         self.set_cursor(1)
  78         curses.echo()
  79         self.stdscr.addstr( y, x, " "*20, curses.A_REVERSE)
  80         s = self.stdscr.getstr(y,x)
  81         curses.noecho()
  82         self.set_cursor(0)
  83         return s
  84
  85
  86     def print_history(self):
  87         width = [20, 40, 14, 14]
  88         delta = (self.maxx - sum(width) - 4)/3
  89         format_str = "%"+"%d"%width[0]+"s"+"%"+"%d"%(width[1]+delta)+"s"+"%"+"%d"%(width[2]+delta)+"s"+"%"+"%d"%(width[3]+delta)+"s"
  90
  91         b = 0
  92         messages = []
  93
  94         for item in self.wallet.get_tx_history():
  95             tx_hash, conf, is_mine, value, fee, balance, timestamp = item
  96             if conf:
  97                 try:
  98                     time_str = datetime.datetime.fromtimestamp( timestamp).isoformat(' ')[:-3]
  99                 except:
 100                     time_str = "------"
 101             else:
 102                 time_str = 'pending'
 103
 104             label, is_default_label = self.wallet.get_label(tx_hash)
 105             messages.append( format_str%( time_str, label, format_satoshis(value, whitespaces=True), format_satoshis(balance, whitespaces=True) ) )
 106
 107         self.print_list(messages[::-1], format_str%( _("Date"), _("Description"), _("Amount"), _("Balance")))
 108
 109
 110     def print_balance(self):
 111         if self.wallet.interface and self.wallet.interface.is_connected:
 112             if not self.wallet.up_to_date:
 113                 msg = _( "Synchronizing..." )
 114             else:
 115                 c, u =  self.wallet.get_balance()
 116                 msg = _("Balance")+": %f  "%(Decimal( c ) / 100000000)
 117                 if u: msg += "  [%f unconfirmed]"%(Decimal( u ) / 100000000)
 118         else:
 119                 msg = _( "Not connected" )
 120
 121         self.stdscr.addstr( self.maxy -1, 3, msg)
 122
 123         for i in range(self.num_tabs):
 124             self.stdscr.addstr( 0, 2 + 2*i + len(''.join(self.tab_names[0:i])), ' '+self.tab_names[i]+' ', curses.A_BOLD if self.tab == i else 0)
 125
 126         self.stdscr.addstr( self.maxy -1, self.maxx-30, ' '.join([_("Settings"), _("Network"), _("Quit")]))
 127
 128
 129     def print_contacts(self):
 130         messages = map(lambda addr: "%30s    %30s       "%(addr, self.wallet.labels.get(addr,"")), self.wallet.addressbook)
 131         self.print_list(messages, "%19s  %25s "%("Address", "Label"))
 132
 133     def print_receive(self):
 134         messages = map(lambda addr: "%30s    %30s       "%(addr, self.wallet.labels.get(addr,"")), self.wallet.addresses())
 135         self.print_list(messages, "%19s  %25s "%("Address", "Label"))
 136
 137     def print_edit_line(self, y, label, text, index, size):
 138         text += " "*(size - len(text) )
 139         self.stdscr.addstr( y, 2, label)
 140         self.stdscr.addstr( y, 15, text, curses.A_REVERSE if self.pos%6==index else curses.color_pair(1))
 141
 142     def print_send_tab(self):
 143         self.stdscr.clear()
 144         self.print_edit_line(3, _("Pay to"), self.str_recipient, 0, 40)
 145         self.print_edit_line(5, _("Description"), self.str_description, 1, 40)
 146         self.print_edit_line(7, _("Amount"), self.str_amount, 2, 15)
 147         self.print_edit_line(9, _("Fee"), self.str_fee, 3, 15)
 148         self.stdscr.addstr( 12, 15, _("[Send]"), curses.A_REVERSE if self.pos%6==4 else curses.color_pair(2))
 149         self.stdscr.addstr( 12, 25, _("[Clear]"), curses.A_REVERSE if self.pos%6==5 else curses.color_pair(2))
 150
 151     def print_banner(self):
 152         for i, x in enumerate( self.wallet.interface.banner.split('\n') ):
 153             self.stdscr.addstr( 1+i, 1, x )
 154
 155     def print_list(self, list, firstline):
 156         self.maxpos = len(list)
 157         if not self.maxpos: return
 158         firstline += " "*(self.maxx -2 - len(firstline))
 159         self.stdscr.addstr( 1, 1, firstline )
 160         for i in range(self.maxy-4):
 161             msg = list[i] if i < len(list) else ""
 162             msg += " "*(self.maxx - 2 - len(msg))
 163             self.stdscr.addstr( i+2, 1, msg[0:self.maxx - 2], curses.A_REVERSE if i == (self.pos % self.maxpos) else 0)
 164
 165     def refresh(self):
 166         if self.tab == -1: return
 167         self.stdscr.border(0)
 168         self.print_balance()
 169         self.stdscr.refresh()
 170
 171     def main_command(self):
 172         c = self.stdscr.getch()
 173         print c
 174         if   c == curses.KEY_RIGHT: self.tab = (self.tab + 1)%self.num_tabs
 175         elif c == curses.KEY_LEFT: self.tab = (self.tab - 1)%self.num_tabs
 176         elif c == curses.KEY_DOWN: self.pos +=1
 177         elif c == curses.KEY_UP: self.pos -= 1
 178         elif c == 9: self.pos +=1 # tab
 179         elif curses.unctrl(c) in ['^W', '^C', '^X', '^Q']: self.tab = -1
 180         elif curses.unctrl(c) in ['^N']: self.network_dialog()
 181         elif curses.unctrl(c) == '^S': self.settings_dialog()
 182         else: return c
 183         if self.pos<0: self.pos=0
 184         if self.pos>=self.maxpos: self.pos=self.maxpos - 1
 185
 186     def run_tab(self, i, print_func, exec_func):
 187         while self.tab == i:
 188             self.stdscr.clear()
 189             print_func()
 190             self.refresh()
 191             c = self.main_command()
 192             if c: exec_func(c)
 193
 194
 195     def run_history_tab(self, c):
 196         if c == 10:
 197             out = self.run_popup('',["blah","foo"])
 198
 199
 200     def edit_str(self, target, c, is_num=False):
 201         # detect backspace
 202         if c in [8, 127, 263] and target:
 203             target = target[:-1]
 204         elif not is_num or curses.unctrl(c) in '0123456789.':
 205             target += curses.unctrl(c)
 206         return target
 207
 208
 209     def run_send_tab(self, c):
 210         if self.pos%6 == 0:
 211             self.str_recipient = self.edit_str(self.str_recipient, c)
 212         if self.pos%6 == 1:
 213             self.str_description = self.edit_str(self.str_description, c)
 214         if self.pos%6 == 2:
 215             self.str_amount = self.edit_str(self.str_amount, c, True)
 216         elif self.pos%6 == 3:
 217             self.str_fee = self.edit_str(self.str_fee, c, True)
 218         elif self.pos%6==4:
 219             if c == 10: self.do_send()
 220         elif self.pos%6==5:
 221             if c == 10: self.do_clear()
 222
 223
 224     def run_receive_tab(self, c):
 225         if c == 10:
 226             out = self.run_popup('Address', ["Edit label", "Freeze", "Prioritize"])
 227
 228     def run_contacts_tab(self, c):
 229         if c == 10 and self.wallet.addressbook:
 230             out = self.run_popup('Adress', ["Copy", "Pay to", "Edit label", "Delete"]).get('button')
 231             address = self.wallet.addressbook[self.pos%len(self.wallet.addressbook)]
 232             if out == "Pay to":
 233                 self.tab = 1
 234                 self.str_recipient = address
 235                 self.pos = 2
 236             elif out == "Edit label":
 237                 s = self.get_string(6 + self.pos, 18)
 238                 if s:
 239                     self.wallet.labels[address] = s
 240
 241     def run_banner_tab(self, c):
 242         self.show_message(repr(c))
 243         pass
 244
 245     def main(self,url):
 246
 247         tty.setraw(sys.stdin)
 248         while self.tab != -1:
 249             self.run_tab(0, self.print_history, self.run_history_tab)
 250             self.run_tab(1, self.print_send_tab, self.run_send_tab)
 251             self.run_tab(2, self.print_receive, self.run_receive_tab)
 252             self.run_tab(3, self.print_contacts, self.run_contacts_tab)
 253             self.run_tab(4, self.print_banner, self.run_banner_tab)
 254
 255         tty.setcbreak(sys.stdin)
 256         curses.nocbreak()
 257         self.stdscr.keypad(0)
 258         curses.echo()
 259         curses.endwin()
 260
 261
 262     def do_clear(self):
 263         self.str_amount = ''
 264         self.str_recipient = ''
 265         self.str_fee = ''
 266         self.str_description = ''
 267
 268     def do_send(self):
 269         if not is_valid(self.str_recipient):
 270             self.show_message(_('Invalid Bitcoin address'))
 271             return
 272         try:
 273             amount = int( Decimal( self.str_amount) * 100000000 )
 274         except:
 275             self.show_message(_('Invalid Amount'))
 276             return
 277         try:
 278             fee = int( Decimal( self.str_fee) * 100000000 )
 279         except:
 280             self.show_message(_('Invalid Fee'))
 281             return
 282
 283         if self.wallet.use_encryption:
 284             password = self.password_dialog()
 285             if not password:
 286                 return
 287         else:
 288             password = None
 289
 290         try:
 291             tx = self.wallet.mktx( [(self.str_recipient, amount)], password, fee)
 292         except BaseException, e:
 293             self.show_message(str(e))
 294             return
 295
 296         if self.str_description:
 297             self.wallet.labels[tx.hash()] = self.str_description
 298
 299         h = self.wallet.send_tx(tx)
 300         self.show_message(_("Please wait..."), getchar=False)
 301         self.wallet.tx_event.wait()
 302         status, msg = self.wallet.receive_tx( h )
 303
 304         if status:
 305             self.show_message(_('Payment sent.'))
 306             self.do_clear()
 307             #self.update_contacts_tab()
 308         else:
 309             self.show_message(_('Error'))
 310
 311
 312     def show_message(self, message, getchar = True):
 313         w = self.w
 314         w.clear()
 315         w.border(0)
 316         w.addstr(2,2,message)
 317         w.refresh()
 318         if getchar: c = self.stdscr.getch()
 319
 320
 321     def run_popup(self, title, items):
 322         return self.run_dialog(title, map(lambda x: {'type':'button','label':x}, items), interval=1, y_pos = self.pos+3)
 323
 324
 325     def network_dialog(self):
 326         out = self.run_dialog('Network', [
 327             {'label':'server', 'type':'str', 'value':self.wallet.interface.server},
 328             {'label':'proxy', 'type':'str', 'value':self.config.get('proxy', '')},
 329             ], buttons = 1)
 330         if out:
 331             if out.get('server'):
 332                 server = out.get('server')
 333                 if out.get('proxy'):
 334                     proxy = self.parse_proxy_options(out.get('proxy'))
 335                 else:
 336                     proxy = None
 337
 338                 self.wallet.config.set_key("proxy", proxy, True)
 339                 self.wallet.config.set_key("server", server, True)
 340                 self.wallet.interface.set_server(server, proxy)
 341
 342
 343
 344     def settings_dialog(self):
 345         out = self.run_dialog('Settings', [
 346             {'label':'Default GUI', 'type':'list', 'choices':['classic','lite','gtk','text'], 'value':self.config.get('gui')},
 347             {'label':'Default fee', 'type':'satoshis', 'value': format_satoshis(self.wallet.fee).strip() }
 348             ], buttons = 1)
 349         if out:
 350             if out.get('Default GUI'):
 351                 self.config.set_key('gui', out['Default GUI'], True)
 352             if out.get('Default fee'):
 353                 fee = int ( Decimal( out['Default fee']) *10000000 )
 354                 self.config.set_key('fee_per_kb', fee, True)
 355
 356
 357     def password_dialog(self):
 358         out = self.run_dialog('Password', [
 359             {'label':'Password', 'type':'password', 'value':''}
 360             ], buttons = 1)
 361         return out.get('Password')
 362
 363
 364     def run_dialog(self, title, items, interval=2, buttons=None, y_pos=3):
 365         self.popup_pos = 0
 366
 367         self.w = curses.newwin( 5 + len(items)*interval + (2 if buttons else 0), 50, y_pos, 5)
 368         w = self.w
 369         out = {}
 370         while True:
 371             w.clear()
 372             w.border(0)
 373             w.addstr( 0, 2, title)
 374
 375             num = len(items)
 376
 377             numpos = num
 378             if buttons: numpos += 2
 379
 380             for i in range(num):
 381                 item = items[i]
 382                 label = item.get('label')
 383                 if item.get('type') == 'list':
 384                     value = item.get('value','')
 385                 elif item.get('type') == 'satoshis':
 386                     value = item.get('value','')
 387                 elif item.get('type') == 'str':
 388                     value = item.get('value','')
 389                 elif item.get('type') == 'password':
 390                     value = '*'*len(item.get('value',''))
 391                 else:
 392                     value = ''
 393
 394                 if len(value)<20: value += ' '*(20-len(value))
 395
 396                 if item.has_key('value'):
 397                     w.addstr( 2+interval*i, 2, label)
 398                     w.addstr( 2+interval*i, 15, value, curses.A_REVERSE if self.popup_pos%numpos==i else curses.color_pair(1) )
 399                 else:
 400                     w.addstr( 2+interval*i, 2, label, curses.A_REVERSE if self.popup_pos%numpos==i else 0)
 401
 402             if buttons:
 403                 w.addstr( 5+interval*i, 10, "[  ok  ]",     curses.A_REVERSE if self.popup_pos%numpos==(numpos-2) else curses.color_pair(2))
 404                 w.addstr( 5+interval*i, 25, "[cancel]", curses.A_REVERSE if self.popup_pos%numpos==(numpos-1) else curses.color_pair(2))
 405
 406             w.refresh()
 407
 408             c = self.stdscr.getch()
 409             if c in [ord('q'), 27]: break
 410             elif c in [curses.KEY_LEFT, curses.KEY_UP]: self.popup_pos -= 1
 411             elif c in [curses.KEY_RIGHT, curses.KEY_DOWN]: self.popup_pos +=1
 412             else:
 413                 i = self.popup_pos%numpos
 414                 if buttons and c==10:
 415                     if i == numpos-2:
 416                         return out
 417                     elif i == numpos -1:
 418                         return {}
 419
 420                 item = items[i]
 421                 _type = item.get('type')
 422
 423                 if _type == 'str':
 424                     item['value'] = self.edit_str(item['value'], c)
 425                     out[item.get('label')] = item.get('value')
 426
 427                 elif _type == 'password':
 428                     item['value'] = self.edit_str(item['value'], c)
 429                     out[item.get('label')] = item ['value']
 430
 431                 elif _type == 'satoshis':
 432                     item['value'] = self.edit_str(item['value'], c, True)
 433                     out[item.get('label')] = item.get('value')
 434
 435                 elif _type == 'list':
 436                     choices = item.get('choices')
 437                     try:
 438                         j = choices.index(item.get('value'))
 439                     except:
 440                         j = 0
 441                     new_choice = choices[(j + 1)% len(choices)]
 442                     item['value'] = new_choice
 443                     out[item.get('label')] = item.get('value')
 444
 445                 elif _type == 'button':
 446                     out['button'] = item.get('label')
 447                     break
 448
 449         return out
 450