b6329332e1cc4820ef772ffbde43caf882dd95fc
[gc-dialer] / gc_dialer / gc_dialer.py
1 #!/usr/bin/python2.5
2
3
4 """
5 Grandcentral Dialer
6 Python front-end to a wget script to use grandcentral.com to place outbound VOIP calls.
7 (C) 2008 Mark Bergman
8 bergman@merctech.com
9 """
10
11
12 import sys
13 import os
14 import re
15 import time
16 import threading
17 import contextlib
18 import gobject
19 import gtk
20 import gc
21 try:
22         import hildon
23 except ImportError:
24         hildon = None
25
26 try:
27         import osso
28         try:
29                 import abook
30                 import evolution.ebook as evobook
31         except ImportError:
32                 abook = None
33                 evobook = None
34 except ImportError:
35         osso = None
36
37 try:
38         import doctest
39         import optparse
40 except ImportError:
41         doctest = None
42         optparse = None
43
44 from gcbackend import GCDialer
45
46 import socket
47 socket.setdefaulttimeout(5)
48
49 @contextlib.contextmanager
50 def gtk_critical_section():
51         #The API changed and I hope these are the right calls
52         gtk.gdk.threads_enter()
53         yield
54         gtk.gdk.threads_leave()
55
56
57 def makeugly(prettynumber):
58         """
59         function to take a phone number and strip out all non-numeric
60         characters
61
62         >>> makeugly("+012-(345)-678-90")
63         '01234567890'
64         """
65         uglynumber = re.sub('\D', '', prettynumber)
66         return uglynumber
67
68
69 def makepretty(phonenumber):
70         """
71         Function to take a phone number and return the pretty version
72          pretty numbers:
73                 if phonenumber begins with 0:
74                         ...-(...)-...-....
75                 if phonenumber begins with 1: ( for gizmo callback numbers )
76                         1 (...)-...-....
77                 if phonenumber is 13 digits:
78                         (...)-...-....
79                 if phonenumber is 10 digits:
80                         ...-....
81         >>> makepretty("12")
82         '12'
83         >>> makepretty("1234567")
84         '123-4567'
85         >>> makepretty("1234567890")
86         '(123)-456-7890'
87         >>> makepretty("01234567890")
88         '+012-(345)-678-90'
89         """
90         if phonenumber is None:
91                 return ""
92
93         if len(phonenumber) < 3:
94                 return phonenumber
95
96         if phonenumber[0] == "0":
97                 prettynumber = ""
98                 prettynumber += "+%s" % phonenumber[0:3]
99                 if 3 < len(phonenumber):
100                         prettynumber += "-(%s)" % phonenumber[3:6]
101                         if 6 < len(phonenumber):
102                                 prettynumber += "-%s" % phonenumber[6:9]
103                                 if 9 < len(phonenumber):
104                                         prettynumber += "-%s" % phonenumber[9:]
105                 return prettynumber
106         elif len(phonenumber) <= 7:
107                 prettynumber = "%s-%s" % (phonenumber[0:3], phonenumber[3:])
108         elif len(phonenumber) > 8 and phonenumber[0] == "1":
109                 prettynumber = "1 (%s)-%s-%s" %(phonenumber[1:4], phonenumber[4:7], phonenumber[7:]) 
110         elif len(phonenumber) > 7:
111                 prettynumber = "(%s)-%s-%s" % (phonenumber[0:3], phonenumber[3:6], phonenumber[6:])
112         return prettynumber
113
114
115 class Dialpad(object):
116
117         def __init__(self):
118                 self.phonenumber = ""
119                 self.prettynumber = ""
120                 self.areacode = "518"
121                 self.clipboard = gtk.clipboard_get()
122                 self.wTree = gtk.Builder()
123                 self.recentmodel = gtk.ListStore(gobject.TYPE_STRING, gobject.TYPE_STRING)
124                 self.recentviewselection = None
125                 self.callbackNeedsSetup = True
126                 self.recenttime = 0.0
127
128                 for path in [ './gc_dialer.xml',
129                                 '../lib/gc_dialer.xml',
130                                 '/usr/local/lib/gc_dialer.xml' ]:
131                         if os.path.isfile(path):
132                                 self.wTree.add_from_file(path)
133                                 break
134                 else:
135                         self.ErrPopUp("Cannot find gc_dialer.xml")
136                         gtk.main_quit()
137                         return
138
139
140                 #Get the buffer associated with the number display
141                 self.numberdisplay = self.wTree.get_object("numberdisplay")
142                 self.setNumber("")
143                 self.notebook = self.wTree.get_object("notebook")
144
145                 self.window = self.wTree.get_object("Dialpad")
146                 if hildon is not None:
147                         self.app = hildon.Program()
148                         self.window.set_title("Keypad")
149                         self.app.add_window(self.window)
150                         self.wTree.get_object("callbackentry").set_property('hildon-input-mode', (1 << 4))
151                         self.wTree.get_object("usernameentry").set_property('hildon-input-mode', 7)
152                         self.wTree.get_object("passwordentry").set_property('hildon-input-mode', 7)
153                 else:
154                         print "No Hildon"
155
156                 if osso is not None:
157                         self.osso = osso.Context("gc_dialer", "0.6.0", False)
158                         device = osso.DeviceState(self.osso)
159                         device.set_device_state_callback(self.on_device_state_change, None)
160                         if abook is not None and evobook is not None:
161                                 abook.init_with_name("gc_dialer", self.osso)
162                                 self.ebook = evo.open_addressbook("default")
163                         else:
164                                 print "No abook and No evolution address book support"
165                 else:
166                         print "No OSSO"
167
168                 if self.window:
169                         self.window.connect("destroy", gtk.main_quit)
170                         self.window.show_all()
171
172                 callbackMapping = {
173                         # Process signals from buttons
174                         "on_digit_clicked"  : self.on_digit_clicked,
175                         "on_dial_clicked"    : self.on_dial_clicked,
176                         "on_loginbutton_clicked" : self.on_loginbutton_clicked,
177                         "on_clearcookies_clicked" : self.on_clearcookies_clicked,
178                         "on_callbackentry_changed" : self.on_callbackentry_changed,
179                         "on_notebook_switch_page" : self.on_notebook_switch_page,
180                         "on_recentview_row_activated" : self.on_recentview_row_activated,
181                         "on_back_clicked" : self.Backspace
182                 }
183                 self.wTree.connect_signals(callbackMapping)
184
185                 # Defer initalization of recent view
186                 self.gcd = GCDialer()
187
188                 self.attemptLogin(2)
189                 gobject.idle_add(self.init_grandcentral)
190                 #self.init_grandcentral()
191                 gobject.idle_add(self.init_recentview)
192
193                 #self.reduce_memory()
194
195         def init_grandcentral(self):
196                 """ deferred initalization of the grandcentral info """
197                 
198                 try:
199                         #self.attemptLogin(2)
200                         if self.gcd.isAuthed():
201                                 if self.gcd.getCallbackNumber() is None:
202                                         self.gcd.setSaneCallback()
203                 except:
204                         pass
205                 
206                 self.setAccountNumber()
207                 print "exit init_gc"
208                 return False
209
210         def init_recentview(self):
211                 """ deferred initalization of the recent view treeview """
212
213                 recentview = self.wTree.get_object("recentview")
214                 recentview.set_model(self.recentmodel)
215                 textrenderer = gtk.CellRendererText()
216
217                 # Add the column to the treeview
218                 column = gtk.TreeViewColumn("Calls", textrenderer, text=1)
219                 column.set_sizing(gtk.TREE_VIEW_COLUMN_FIXED)
220
221                 recentview.append_column(column)
222
223                 self.recentviewselection = recentview.get_selection()
224                 self.recentviewselection.set_mode(gtk.SELECTION_SINGLE)
225
226                 return False
227
228         def on_recentview_row_activated(self, treeview, path, view_column):
229                 model, itr = self.recentviewselection.get_selected()
230                 if not itr:
231                         return
232
233                 self.setNumber(self.recentmodel.get_value(itr, 0))
234                 self.notebook.set_current_page(0)
235                 self.recentviewselection.unselect_all()
236
237         def on_notebook_switch_page(self, notebook, page, page_num):
238                 if page_num == 1 and (time.time() - self.recenttime) > 300:
239                         gobject.idle_add(self.populate_recentview)
240                 elif page_num ==2 and self.callbackNeedsSetup:
241                         gobject.idle_add(self.setupCallbackCombo)
242                 if hildon:
243                         try:
244                                 self.window.set_title(self.notebook.get_tab_label(self.notebook.get_nth_page(page_num)).get_text())
245                         except:
246                                 self.window.set_title("")
247
248         def populate_recentview(self):
249                 print "Populating"
250                 self.recentmodel.clear()
251                 for item in self.gcd.get_recent():
252                         self.recentmodel.append(item)
253                 self.recenttime = time.time()
254
255                 return False
256
257         def on_clearcookies_clicked(self, data=None):
258                 self.gcd.reset()
259                 self.callbackNeedsSetup = True
260                 self.recenttime = 0.0
261                 self.recentmodel.clear()
262                 self.wTree.get_object("callbackentry").set_text("")
263         
264                 # re-run the inital grandcentral setup
265                 self.attemptLogin(2)
266                 gobject.idle_add(self.init_grandcentral)
267
268         def setupCallbackCombo(self):
269                 combobox = self.wTree.get_object("callbackcombo")
270                 self.callbacklist = gtk.ListStore(gobject.TYPE_STRING)
271                 combobox.set_model(self.callbacklist)
272                 combobox.set_text_column(0)
273                 for number, description in self.gcd.getCallbackNumbers().iteritems():
274                         self.callbacklist.append([makepretty(number)] )
275
276                 self.wTree.get_object("callbackentry").set_text(makepretty(self.gcd.getCallbackNumber()))
277                 self.callbackNeedsSetup = False
278
279         def on_callbackentry_changed(self, data=None):
280                 text = makeugly(self.wTree.get_object("callbackentry").get_text())
281                 if self.gcd.validate(text) and text != self.gcd.getCallbackNumber():
282                         self.gcd.setCallbackNumber(text)
283                         self.wTree.get_object("callbackentry").set_text(self.wTree.get_object("callbackentry").get_text())
284                 #self.reduce_memory()
285
286         def attemptLogin(self, times = 1):
287                 #if self.isHildon:
288                 #       dialog = hildon.LoginDialog(self.window)
289                 #       dialog.set_message("Grandcentral Login")
290                 #else:
291                 dialog = self.wTree.get_object("login_dialog")
292
293                 while (0 < times) and not self.gcd.isAuthed():
294                         if dialog.run() != gtk.RESPONSE_OK:
295                                 times = 0
296                                 continue
297
298                         #if self.isHildon:
299                         #       username = dialog.get_username()
300                         #       password = dialog.get_password()
301                         #else:
302                         username = self.wTree.get_object("usernameentry").get_text()
303                         password = self.wTree.get_object("passwordentry").get_text()
304                         self.wTree.get_object("passwordentry").set_text("")
305                         print "Attempting login"
306                         self.gcd.login(username, password)
307                         print "hiding dialog"
308                         dialog.hide()
309                         times = times - 1
310
311                 #if self.isHildon:
312                 #       print "destroy dialog"
313                 #       dialog.destroy()
314
315                 return False
316
317         def ErrPopUp(self, msg):
318                 error_dialog = gtk.MessageDialog(None, 0, gtk.MESSAGE_ERROR, gtk.BUTTONS_CLOSE, msg)
319
320                 def close(dialog, response, editor):
321                         editor.about_dialog = None
322                         dialog.destroy()
323                 error_dialog.connect("response", close, self)
324                 self.error_dialog = error_dialog
325                 error_dialog.run()
326
327         def on_paste(self, data=None):
328                 contents = self.clipboard.wait_for_text()
329                 phoneNumber = re.sub('\D', '', contents)
330                 self.setNumber(phoneNumber)
331         
332         def on_loginbutton_clicked(self, data=None):
333                 self.wTree.get_object("login_dialog").response(gtk.RESPONSE_OK)
334
335         def on_dial_clicked(self, widget):
336                 self.attemptLogin(3)
337
338                 if not self.gcd.isAuthed() or self.gcd.getCallbackNumber() == "":
339                         self.ErrPopUp("Backend link with grandcentral is not working, please try again")
340                         return
341
342                 #if len(self.phonenumber) == 7:
343                 #       #add default area code
344                 #       self.phonenumber = self.areacode + self.phonenumber
345
346                 if self.gcd.dial(self.phonenumber) is False:
347                         self.ErrPopUp(self.gcd._msg)
348                 else:
349                         self.setNumber("")
350
351                 self.recentmodel.clear()
352                 self.recenttime = 0.0
353                 #self.reduce_memory()
354         
355         #def on_device_state_change(self, shutdown, save_unsaved_data, memory_low, system_inactivity, message, userData):
356         #       """
357         #       @todo Might be useful to do something when going in offline mode or low memory
358         #       @note Hildon specific
359         #       """
360         #       pass
361
362         def setNumber(self, number):
363                 self.phonenumber = makeugly(number)
364                 self.prettynumber = makepretty(self.phonenumber)
365                 self.numberdisplay.set_label("<span size='30000' weight='bold'>%s</span>" % ( self.prettynumber ) )
366
367         def setAccountNumber(self):
368                 accountnumber = self.gcd.getAccountNumber()
369                 self.wTree.get_object("gcnumberlabel").set_label("<span size='23000' weight='bold'>%s</span>" % (accountnumber))
370
371         def Backspace(self, widget):
372                 self.setNumber(self.phonenumber[:-1])
373
374         def on_digit_clicked(self, widget):
375                 self.setNumber(self.phonenumber + widget.get_name()[5])
376
377
378 def run_doctest():
379         failureCount, testCount = doctest.testmod()
380         if not failureCount:
381                 print "Tests Successful"
382                 sys.exit(0)
383         else:
384                 sys.exit(1)
385
386
387 def run_dialpad():
388         gc.set_threshold(50, 3, 3)
389         gtk.gdk.threads_init()
390         title = 'Dialpad'
391         handle = Dialpad()
392         gtk.main()
393         sys.exit(1)
394
395
396 class DummyOptions(object):
397         def __init__(self):
398                 self.test = False
399
400
401 if __name__ == "__main__":
402         if hildon:
403                 gtk.set_application_name("Dialer")
404
405         try:
406                 parser = optparse.OptionParser()
407                 parser.add_option("-t", "--test", action="store_true", dest="test", help="Run tests")
408                 (options, args) = parser.parse_args()
409         except:
410                 args = []
411                 options = DummyOptions()
412
413         if options.test:
414                 run_doctest()
415         else:
416                 run_dialpad()