vcs.maemo.org Git - doneit/blob - src/gtk_rtmilk.py

   1 import webbrowser
   2 import datetime
   3 import urlparse
   4 import base64
   5
   6 import gobject
   7 import gtk
   8
   9 import toolbox
  10 import gtk_toolbox
  11 import rtmilk
  12 import rtmapi
  13
  14
  15 def abbreviate(text, expectedLen):
  16         singleLine = " ".join(text.split("\n"))
  17         lineLen = len(singleLine)
  18         if lineLen <= expectedLen:
  19                 return singleLine
  20
  21         abbrev = "..."
  22
  23         leftLen = expectedLen // 2 - 1
  24         rightLen = max(expectedLen - leftLen - len(abbrev) + 1, 1)
  25
  26         abbrevText =  singleLine[0:leftLen] + abbrev + singleLine[-rightLen:-1]
  27         assert len(abbrevText) <= expectedLen, "Too long: '%s'" % abbrevText
  28         return abbrevText
  29
  30
  31 def abbreviate_url(url, domainLength, pathLength):
  32         urlParts = urlparse.urlparse(url)
  33
  34         netloc = urlParts.netloc
  35         path = urlParts.path
  36
  37         pathLength += max(domainLength - len(netloc), 0)
  38         domainLength += max(pathLength - len(path), 0)
  39
  40         netloc = abbreviate(netloc, domainLength)
  41         path = abbreviate(path, pathLength)
  42         return netloc + path
  43
  44
  45 def get_token(username, apiKey, secret):
  46         token = None
  47         rtm = rtmapi.RTMapi(username, apiKey, secret, token)
  48
  49         authURL = rtm.getAuthURL()
  50         webbrowser.open(authURL)
  51         # mb = gtk_toolbox.MessageBox2("You need to authorize DoneIt with\nRemember The Milk.\nClick OK after you authorize.")
  52         # mb.run()
  53
  54         token = rtm.getToken()
  55         return token
  56
  57
  58 def get_credentials(credentialsDialog):
  59         # @todo Figure out storage of credentials
  60         username, password = credentialsDialog.request_credentials()
  61         token = get_token(username, rtmilk.RtMilkManager.API_KEY, rtmilk.RtMilkManager.SECRET)
  62         return username, password, token
  63
  64
  65 class GtkRtMilk(object):
  66
  67         ID_IDX = 0
  68         COMPLETION_IDX = 1
  69         NAME_IDX = 2
  70         PRIORITY_IDX = 3
  71         DUE_IDX = 4
  72         FUZZY_IDX = 5
  73         LINK_IDX = 6
  74         NOTES_IDX = 7
  75
  76         def __init__(self, widgetTree):
  77                 """
  78                 @note Thread agnostic
  79                 """
  80                 self._clipboard = gtk.clipboard_get()
  81
  82                 self._showCompleted = False
  83                 self._showIncomplete = True
  84
  85                 self._editDialog = gtk_toolbox.EditTaskDialog(widgetTree)
  86
  87                 self._projectsList = gtk.ListStore(gobject.TYPE_STRING)
  88                 self._projectsCombo = widgetTree.get_widget("projectsCombo")
  89                 self._onListActivateId = 0
  90
  91                 self._itemList = gtk.ListStore(
  92                         gobject.TYPE_STRING, # id
  93                         gobject.TYPE_BOOLEAN, # is complete
  94                         gobject.TYPE_STRING, # name
  95                         gobject.TYPE_STRING, # priority
  96                         gobject.TYPE_STRING, # due
  97                         gobject.TYPE_STRING, # fuzzy due
  98                         gobject.TYPE_STRING, # Link
  99                         gobject.TYPE_STRING, # Notes
 100                 )
 101                 self._completionColumn = gtk.TreeViewColumn('') # Complete?
 102                 self._completionCell = gtk.CellRendererToggle()
 103                 self._completionCell.set_property("activatable", True)
 104                 self._completionCell.connect("toggled", self._on_completion_change)
 105                 self._completionColumn.pack_start(self._completionCell, False)
 106                 self._completionColumn.set_attributes(self._completionCell, active=self.COMPLETION_IDX)
 107                 self._priorityColumn = gtk.TreeViewColumn('') # Priority
 108                 self._priorityCell = gtk.CellRendererText()
 109                 self._priorityColumn.pack_start(self._priorityCell, False)
 110                 self._priorityColumn.set_attributes(self._priorityCell, text=self.PRIORITY_IDX)
 111                 self._nameColumn = gtk.TreeViewColumn('Name')
 112                 self._nameCell = gtk.CellRendererText()
 113                 self._nameColumn.pack_start(self._nameCell, True)
 114                 self._nameColumn.set_attributes(self._nameCell, text=self.NAME_IDX)
 115                 self._dueColumn = gtk.TreeViewColumn('Due')
 116                 self._dueCell = gtk.CellRendererText()
 117                 self._dueColumn.pack_start(self._nameCell, False)
 118                 self._dueColumn.set_attributes(self._nameCell, text=self.FUZZY_IDX)
 119                 self._linkColumn = gtk.TreeViewColumn('') # Link
 120                 self._linkCell = gtk.CellRendererText()
 121                 self._linkColumn.pack_start(self._nameCell, False)
 122                 self._linkColumn.set_attributes(self._nameCell, text=self.LINK_IDX)
 123                 self._notesColumn = gtk.TreeViewColumn('') # Notes
 124                 self._notesCell = gtk.CellRendererText()
 125                 self._notesColumn.pack_start(self._nameCell, False)
 126                 self._notesColumn.set_attributes(self._nameCell, text=self.NOTES_IDX)
 127
 128                 self._todoItemTree = widgetTree.get_widget("todoItemTree")
 129                 self._onItemSelectId = 0
 130
 131                 self._taskNameEntry = widgetTree.get_widget("add-taskNameEntry")
 132                 self._addTaskButton = widgetTree.get_widget("add-addTaskButton")
 133                 self._pasteTaskNameButton = widgetTree.get_widget("add-pasteTaskNameButton")
 134                 self._clearTaskNameButton = widgetTree.get_widget("add-clearTaskNameButton")
 135                 self._onAddId = None
 136                 self._onAddClickedId = None
 137                 self._onAddReleasedId = None
 138                 self._addToEditTimerId = None
 139                 self._onClearId = None
 140                 self._onPasteId = None
 141
 142                 self._credentialsDialog = gtk_toolbox.LoginWindow(widgetTree)
 143                 self._credentials = "", "", ""
 144                 self._manager = None
 145
 146         @staticmethod
 147         def name():
 148                 return "Remember The Milk"
 149
 150         def load_settings(self, config):
 151                 """
 152                 @note Thread Agnostic
 153                 """
 154                 username = config.get(self.name(), "username")
 155                 password = None
 156                 blobbedToken = config.get(self.name(), "bin_blob")
 157                 token = base64.b64decode(blobbedToken)
 158                 self._credentials = username, password, token
 159
 160         def save_settings(self, config):
 161                 """
 162                 @note Thread Agnostic
 163                 """
 164                 config.add_section(self.name())
 165                 config.set(self.name(), "username", self._credentials[0])
 166                 blobbedToken = base64.b64encode(self._credentials[2])
 167                 config.set(self.name(), "bin_blob", blobbedToken)
 168
 169         def login(self):
 170                 """
 171                 @note UI Thread
 172                 """
 173                 if self._manager is not None:
 174                         return
 175
 176                 credentials = self._credentials
 177                 while True:
 178                         try:
 179                                 self._manager = rtmilk.RtMilkManager(*credentials)
 180                                 self._credentials = credentials
 181                                 return # Login succeeded
 182                         except rtmapi.AuthStateMachine.NoData:
 183                                 # Login failed, grab new credentials
 184                                 credentials = get_credentials(self._credentialsDialog)
 185
 186         def logout(self):
 187                 """
 188                 @note Thread Agnostic
 189                 """
 190                 self._credentials = "", "", ""
 191                 self._manager = None
 192
 193         def enable(self):
 194                 """
 195                 @note UI Thread
 196                 """
 197                 self._projectsList.clear()
 198                 self._populate_projects()
 199
 200                 self._itemList.clear()
 201                 self._todoItemTree.append_column(self._completionColumn)
 202                 self._todoItemTree.append_column(self._priorityColumn)
 203                 self._todoItemTree.append_column(self._nameColumn)
 204                 self._todoItemTree.append_column(self._dueColumn)
 205                 self._todoItemTree.append_column(self._linkColumn)
 206                 self._todoItemTree.append_column(self._notesColumn)
 207                 self._reset_task_list()
 208
 209                 self._todoItemTree.set_headers_visible(False)
 210                 self._nameColumn.set_expand(True)
 211
 212                 self._editDialog.enable(self._manager)
 213
 214                 self._onListActivateId = self._projectsCombo.connect("changed", self._on_list_activate)
 215                 self._onItemSelectId = self._todoItemTree.connect("row-activated", self._on_item_select)
 216                 self._onAddId = self._addTaskButton.connect("clicked", self._on_add)
 217                 self._onAddClickedId = self._addTaskButton.connect("pressed", self._on_add_pressed)
 218                 self._onAddReleasedId = self._addTaskButton.connect("released", self._on_add_released)
 219                 self._onPasteId = self._pasteTaskNameButton.connect("clicked", self._on_paste)
 220                 self._onClearId = self._clearTaskNameButton.connect("clicked", self._on_clear)
 221
 222         def disable(self):
 223                 """
 224                 @note UI Thread
 225                 """
 226                 self._projectsCombo.disconnect(self._onListActivateId)
 227                 self._todoItemTree.disconnect(self._onItemSelectId)
 228                 self._addTaskButton.disconnect(self._onAddId)
 229                 self._addTaskButton.disconnect(self._onAddClickedId)
 230                 self._addTaskButton.disconnect(self._onAddReleasedId)
 231                 self._pasteTaskNameButton.disconnect(self._onPasteId)
 232                 self._clearTaskNameButton.disconnect(self._onClearId)
 233
 234                 self._editDialog.disable()
 235
 236                 self._projectsList.clear()
 237                 self._projectsCombo.set_model(None)
 238                 self._projectsCombo.disconnect("changed", self._on_list_activate)
 239
 240                 self._todoItemTree.remove_column(self._completionColumn)
 241                 self._todoItemTree.remove_column(self._priorityColumn)
 242                 self._todoItemTree.remove_column(self._nameColumn)
 243                 self._todoItemTree.remove_column(self._dueColumn)
 244                 self._todoItemTree.remove_column(self._linkColumn)
 245                 self._todoItemTree.remove_column(self._notesColumn)
 246                 self._itemList.clear()
 247                 self._itemList.set_model(None)
 248
 249                 self._manager = None
 250
 251         def _populate_projects(self):
 252                 for project in self._manager.get_projects():
 253                         projectName = project["name"]
 254                         isVisible = project["isVisible"]
 255                         row = (projectName, )
 256                         if isVisible:
 257                                 self._projectsList.append(row)
 258                 self._projectsCombo.set_model(self._projectsList)
 259                 cell = gtk.CellRendererText()
 260                 self._projectsCombo.pack_start(cell, True)
 261                 self._projectsCombo.add_attribute(cell, 'text', 0)
 262                 self._projectsCombo.set_active(0)
 263
 264         def _reset_task_list(self):
 265                 currentProject = self._get_project()
 266                 projId = self._manager.lookup_project(currentProject)["id"]
 267                 isMeta = self._manager.get_project(projId)["isMeta"]
 268                 # @todo RTM handles this by defaulting to a specific list
 269                 self._addTaskButton.set_sensitive(not isMeta)
 270
 271                 self._itemList.clear()
 272                 self._populate_items()
 273
 274         def _get_project(self):
 275                 currentProjectName = self._projectsCombo.get_active_text()
 276                 return currentProjectName
 277
 278         def _populate_items(self):
 279                 # @todo Look using a button for notes and links, and labels for all else
 280                 currentProject = self._get_project()
 281                 projId = self._manager.lookup_project(currentProject)["id"]
 282                 for taskDetails in self._manager.get_tasks_with_details(projId):
 283                         show = self._showCompleted if taskDetails["isCompleted"] else self._showIncomplete
 284                         if not show:
 285                                 continue
 286                         id = taskDetails["id"]
 287                         isCompleted = taskDetails["isCompleted"]
 288                         name = abbreviate(taskDetails["name"], 100)
 289                         priority = taskDetails["priority"]
 290                         dueDescription = taskDetails["dueDate"]
 291                         if dueDescription:
 292                                 dueDate = datetime.datetime.strptime(dueDescription, "%Y-%m-%dT%H:%M:%SZ")
 293                                 fuzzyDue = toolbox.to_fuzzy_date(dueDate)
 294                         else:
 295                                 fuzzyDue = ""
 296
 297                         linkDisplay = taskDetails["url"]
 298                         linkDisplay = abbreviate_url(linkDisplay, 20, 10)
 299
 300                         notes = taskDetails["notes"]
 301                         notesDisplay = "%d Notes" % len(notes) if notes else ""
 302
 303                         row = (id, isCompleted, name, priority, dueDescription, fuzzyDue, linkDisplay, notesDisplay)
 304                         self._itemList.append(row)
 305                 self._todoItemTree.set_model(self._itemList)
 306
 307         def _on_list_activate(self, *args):
 308                 self._reset_task_list()
 309
 310         def _on_item_select(self, treeView, path, viewColumn):
 311                 # @todo See if there is a way to get a right click / tap'n'hold for more task goodness
 312                 taskId = self._itemList[path[0]][self.ID_IDX]
 313
 314                 if viewColumn is self._priorityColumn:
 315                         pass
 316                 elif viewColumn is self._nameColumn:
 317                         self._editDialog.request_task(self._manager, taskId)
 318                         self._reset_task_list()
 319                 elif viewColumn is self._dueColumn:
 320                         self._editDialog.request_task(self._manager, taskId)
 321                         self._reset_task_list()
 322                 elif viewColumn is self._linkColumn:
 323                         webbrowser.open(self._manager.get_task_details(taskId)["url"])
 324                 elif viewColumn is self._notesColumn:
 325                         pass
 326
 327         def _on_add(self, *args):
 328                 name = self._taskNameEntry.get_text()
 329
 330                 currentProject = self._get_project()
 331                 projId = self._manager.lookup_project(currentProject)["id"]
 332                 taskId = self._manager.add_task(projId, name)
 333
 334                 self._taskNameEntry.set_text("")
 335                 self._reset_task_list()
 336
 337         def _on_add_edit(self, *args):
 338                 name = self._taskNameEntry.get_text()
 339
 340                 currentProject = self._get_project()
 341                 projId = self._manager.lookup_project(currentProject)["id"]
 342                 taskId = self._manager.add_task(projId, name)
 343
 344                 try:
 345                         self._editDialog.request_task(self._manager, taskId)
 346                 finally:
 347                         self._taskNameEntry.set_text("")
 348                         self._reset_task_list()
 349
 350         def _on_add_pressed(self, widget):
 351                 self._addToEditTimerId = gobject.timeout_add(1000, self._on_add_edit)
 352
 353         def _on_add_released(self, widget):
 354                 if self._addToEditTimerId is not None:
 355                         gobject.source_remove(self._addToEditTimerId)
 356                 self._addToEditTimerId = None
 357
 358         def _on_paste(self, *args):
 359                 entry = self._taskNameEntry.get_text()
 360                 entry += self._clipboard.wait_for_text()
 361                 self._taskNameEntry.set_text(entry)
 362
 363         def _on_clear(self, *args):
 364                 self._taskNameEntry.set_text("")
 365
 366         def _on_completion_change(self, cell, path):
 367                 taskId = self._itemList[path[0]][self.ID_IDX]
 368                 self._manager.complete_task(taskId)
 369                 self._reset_task_list()