* Pythonic work
[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 #import hildon
22
23 from gcbackend import GCDialer
24
25
26 @contextlib.contextmanager
27 def gtk_critical_section():
28         #The API changed and I hope these are the right calls
29         gtk.gdk.threads_enter()
30         yield
31         gtk.gdk.threads_leave()
32
33
34 def makeugly(prettynumber):
35         """
36         function to take a phone number and strip out all non-numeric
37         characters
38
39         >>> makeugly("+012-(345)-678-90")
40         '01234567890'
41         """
42         uglynumber = re.sub('\D', '', prettynumber)
43         return uglynumber
44
45
46 def makepretty(phonenumber):
47         """
48         Function to take a phone number and return the pretty version
49          pretty numbers:
50                 if phonenumber begins with 0:
51                         ...-(...)-...-....
52                 if phonenumber begins with 1: ( for gizmo callback numbers )
53                         1 (...)-...-....
54                 if phonenumber is 13 digits:
55                         (...)-...-....
56                 if phonenumber is 10 digits:
57                         ...-....
58         >>> makepretty("12")
59         '12'
60         >>> makepretty("1234567")
61         '123-4567'
62         >>> makepretty("1234567890")
63         '(123)-456-7890'
64         >>> makepretty("01234567890")
65         '+012-(345)-678-90'
66         """
67         if phonenumber is None:
68                 return ""
69
70         if len(phonenumber) < 3:
71                 return phonenumber
72
73         if phonenumber[0] == "0":
74                 prettynumber = ""
75                 prettynumber += "+%s" % phonenumber[0:3]
76                 if 3 < len(phonenumber):
77                         prettynumber += "-(%s)" % phonenumber[3:6]
78                         if 6 < len(phonenumber):
79                                 prettynumber += "-%s" % phonenumber[6:9]
80                                 if 9 < len(phonenumber):
81                                         prettynumber += "-%s" % phonenumber[9:]
82                 return prettynumber
83         elif len(phonenumber) <= 7:
84                 prettynumber = "%s-%s" % (phonenumber[0:3], phonenumber[3:])
85         elif len(phonenumber) > 8 and phonenumber[0] == 1:
86                 prettynumber = "1 (%s)-%s-%s" %(phonenumber[1:4], phonenumber[4:7], phonenumber[7:]) 
87         elif len(phonenumber) > 7:
88                 prettynumber = "(%s)-%s-%s" % (phonenumber[0:3], phonenumber[3:6], phonenumber[6:])
89         return prettynumber
90
91
92 class Dialpad(object):
93
94         def __init__(self):
95                 self.phonenumber = ""
96                 self.prettynumber = ""
97                 self.areacode = "518"
98                 self.gcd = GCDialer()
99                 self.wTree = gtk.Builder()
100
101                 for path in [ './gc_dialer.xml',
102                                 '../lib/gc_dialer.xml',
103                                 '/usr/local/lib/gc_dialer.xml' ]:
104                         if os.path.isfile(path):
105                                 self.wTree.add_from_file(path)
106                                 break
107                 else:
108                         self.ErrPopUp("Cannot find gc_dialer.xml")
109                         sys.exit(1)#@todo Is this the best way to force a quit on error?
110
111                 self.window = self.wTree.get_object("Dialpad")
112                 #Get the buffer associated with the number display
113                 self.numberdisplay = self.wTree.get_object("numberdisplay")
114                 self.setNumber("")
115
116                 self.recentview = self.wTree.get_object("recentview")
117                 self.recentmodel = gtk.ListStore(gobject.TYPE_STRING, gobject.TYPE_STRING)
118                 self.recentview.set_model(self.recentmodel)
119                 textrenderer = gtk.CellRendererText()
120
121                 # Add the column to the treeview
122                 column = gtk.TreeViewColumn("Calls", textrenderer, text=1)
123                 column.set_sizing(gtk.TREE_VIEW_COLUMN_FIXED)
124                 self.recentview.append_column(column)
125
126                 self.recentviewselection = self.recentview.get_selection()
127                 self.recentviewselection.set_mode(gtk.SELECTION_SINGLE)
128                 self.recenttime = 0.0
129
130                 self.notebook = self.wTree.get_object("notebook")
131
132                 self.isHildon = False
133                 #if True:
134                 try:
135                         self.app = hildon.Program()
136                         self.wTree.get_object("callbackentry").set_property('hildon-input-mode', 1|(1 << 4))
137                         self.isHildon = True
138                 except:
139                         print "No hildon"
140
141                 if self.window:
142                         self.window.connect("destroy", gtk.main_quit)
143                         self.window.show_all()
144
145                 callbackMapping = {
146                         # Process signals from buttons
147                         "on_digit_clicked"  : self.on_digit_clicked,
148                         "on_dial_clicked"    : self.on_dial_clicked,
149                         "on_loginbutton_clicked" : self.on_loginbutton_clicked,
150                         "on_clearcookies_clicked" : self.on_clearcookies_clicked,
151                         "on_callbackentry_changed" : self.on_callbackentry_changed,
152                         "on_notebook_switch_page" : self.on_notebook_switch_page,
153                         "on_recentview_row_activated" : self.on_recentview_row_activated,
154                         "on_back_clicked" : self.Backspace
155                 }
156                 self.wTree.connect_signals(callbackMapping)
157
158                 self.attemptLogin(3)
159                 if self.gcd.getCallbackNumber() is None:
160                         self.gcd.setSaneCallback()
161
162                 self.setAccountNumber()
163                 self.setupCallbackCombo()
164                 self.reduce_memory()
165
166         def reduce_memory(self):
167                 re.purge()
168                 num = gc.collect()
169                 #print "collect %d objects" % ( num )
170
171         def on_recentview_row_activated(self, treeview, path, view_column):
172                 model, itr = self.recentviewselection.get_selected()
173                 if not itr:
174                         return
175
176                 self.setNumber(self.recentmodel.get_value(itr, 0))
177                 self.notebook.set_current_page(0)
178                 self.recentviewselection.unselect_all()
179
180         def on_notebook_switch_page(self, notebook, page, page_num):
181                 if page_num == 1 and (time.time() - self.recenttime) > 300:
182                         self.populate_recentview()
183
184         def populate_recentview(self):
185                 print "Populating"
186                 self.recentmodel.clear()
187                 for item in self.gcd.get_recent():
188                         self.recentmodel.append(item)
189                 self.recenttime = time.time()
190
191         def on_clearcookies_clicked(self, data=None):
192                 self.gcd.reset()
193                 self.attemptLogin(3)
194
195         def setupCallbackCombo(self):
196                 combobox = self.wTree.get_object("callbackcombo")
197                 self.callbacklist = gtk.ListStore(gobject.TYPE_STRING)
198                 combobox.set_model(self.callbacklist)
199                 combobox.set_text_column(0)
200                 for number, description in self.gcd.getCallbackNumbers().iteritems():
201                         self.callbacklist.append([makepretty(number)] )
202
203                 self.wTree.get_object("callbackentry").set_text(makepretty(self.gcd.getCallbackNumber()))
204
205         def on_callbackentry_changed(self, data=None):
206                 text = makeugly(self.wTree.get_object("callbackentry").get_text())
207                 if self.gcd.validate(text) and text != self.gcd.getCallbackNumber():
208                         self.gcd.setCallbackNumber(text)
209                         self.wTree.get_object("callbackentry").set_text(self.wTree.get_object("callbackentry").get_text())
210                 self.reduce_memory()
211
212         def attemptLogin(self, times = 1):
213                 if self.isHildon:
214                         dialog = hildon.LoginDialog(self.window)
215                         dialog.set_message("Grandcentral Login")
216                 else:
217                         dialog = self.wTree.get_object("login_dialog")
218
219                 while (0 < times) and not self.gcd.isAuthed():
220                         if dialog.run() != gtk.RESPONSE_OK:
221                                 times = 0
222                                 continue
223
224                         if self.isHildon:
225                                 username = dialog.get_username()
226                                 password = dialog.get_password()
227                         else:
228                                 username = self.wTree.get_object("usernameentry").get_text()
229                                 password = self.wTree.get_object("passwordentry").get_text()
230                                 self.wTree.get_object("passwordentry").set_text("")
231                         print "Attempting login"
232                         self.gcd.login(username, password)
233                         dialog.hide()
234                         times -= 1
235
236                 if self.isHildon:
237                         dialog.destroy()
238
239         def ErrPopUp(self, msg):
240                 error_dialog = gtk.MessageDialog(None, 0, gtk.MESSAGE_ERROR, gtk.BUTTONS_CLOSE, msg)
241
242                 def close(dialog, response, editor):
243                         editor.about_dialog = None
244                         dialog.destroy()
245                 error_dialog.connect("response", close, self)
246                 self.error_dialog = error_dialog
247                 error_dialog.run()
248
249         def on_loginbutton_clicked(self, data=None):
250                 self.wTree.get_object("login_dialog").response(gtk.RESPONSE_OK)
251
252         def on_dial_clicked(self, widget):
253                 self.attemptLogin(3)
254
255                 if not self.gcd.isAuthed() or self.gcd.getCallbackNumber() == "":
256                         self.ErrPopUp("Backend link with grandcentral is not working, please try again")
257                         return
258
259                 #if len(self.phonenumber) == 7:
260                 #       #add default area code
261                 #       self.phonenumber = self.areacode + self.phonenumber
262
263                 if self.gcd.dial(self.phonenumber) is False:
264                         self.ErrPopUp(self.gcd._msg)
265                 else:
266                         self.setNumber("")
267
268                 self.recentmodel.clear()
269                 self.recenttime = 0.0
270                 self.reduce_memory()
271
272         def setNumber(self, number):
273                 self.phonenumber = makeugly(number)
274                 self.prettynumber = makepretty(self.phonenumber)
275                 self.numberdisplay.set_label("<span size='30000' weight='bold'>%s</span>" % ( self.prettynumber ) )
276
277         def setAccountNumber(self):
278                 accountnumber = self.gcd.getAccountNumber()
279                 self.wTree.get_object("gcnumberlabel").set_label("<span size='23000' weight='bold'>%s</span>" % (accountnumber))
280
281         def Backspace(self, widget):
282                 self.setNumber(self.phonenumber[:-1])
283
284         def on_digit_clicked(self, widget):
285                 self.setNumber(self.phonenumber + re.sub('\D', '', widget.get_label()))
286
287
288 def run_doctest():
289         import doctest
290         failureCount, testCount = doctest.testmod()
291         if not failureCount:
292                 print "Tests Successful"
293                 sys.exit(0)
294         else:
295                 sys.exit(1)
296
297
298 def run_dialpad():
299         gc.set_threshold(50, 3, 3)
300         gtk.gdk.threads_init()
301         title = 'Dialpad'
302         handle = Dialpad()
303         gtk.main()
304         sys.exit(1)
305
306
307 if __name__ == "__main__":
308         from optparse import OptionParser
309
310         parser = OptionParser()
311         parser.add_option("-t", "--test", action="store_true", dest="test", help="Run tests")
312         (options, args) = parser.parse_args()
313
314         if options.test:
315                 run_doctest()
316         else:
317                 run_dialpad()