[hamster-applet] reading activities from gtg if enabled

[Toms Baugis <tbaugis src gnome org>]

commit a2cb72a59df25c5eda8bcaf7f502ebfb074ac5f4
Author: Toms Bauģis <toms baugis gmail com>
Date:   Fri May 14 18:13:34 2010 +0100

    reading activities from gtg if enabled

 src/hamster/Makefile.am              |    2 +-
 src/hamster/eds.py                   |   53 ----------------
 src/hamster/external.py              |  111 ++++++++++++++++++++++++++++++++++
 src/hamster/widgets/activityentry.py |    4 +-
 4 files changed, 115 insertions(+), 55 deletions(-)
---
diff --git a/src/hamster/Makefile.am b/src/hamster/Makefile.am
index f6c107f..c52a38a 100644
--- a/src/hamster/Makefile.am
+++ b/src/hamster/Makefile.am
@@ -9,7 +9,7 @@ hamster_PYTHON = \
 	defs.py \
 	storage.py \
 	db.py \
-	eds.py \
+	external.py \
 	stats.py \
         overview.py \
         overview_totals.py \
diff --git a/src/hamster/external.py b/src/hamster/external.py
new file mode 100644
index 0000000..05624c3
--- /dev/null
+++ b/src/hamster/external.py
@@ -0,0 +1,111 @@
+# - coding: utf-8 -
+
+# Copyright (C) 2007 Patryk Zawadzki <patrys at pld-linux.org>
+# Copyright (C) 2008, 2010 Toms Bauģis <toms.baugis at gmail.com>
+
+# This file is part of Project Hamster.
+
+# Project Hamster is free software: you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+
+# Project Hamster is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+
+# You should have received a copy of the GNU General Public License
+# along with Project Hamster.  If not, see <http://www.gnu.org/licenses/>.
+
+import logging
+from configuration import conf
+import gobject
+import dbus, dbus.mainloop.glib
+
+try:
+    import evolution
+    from evolution import ecal
+except:
+    evolution = None
+
+class ActivitiesSource(gobject.GObject):
+    def __init__(self):
+        gobject.GObject.__init__(self)
+        self.source = conf.get("activities_source")
+        self.__gtg_connection = None
+
+        if self.source == "evo" and not evolution:
+            self.source == "" # on failure pretend that there is no evolution
+        elif self.source == "gtg":
+            gobject.GObject.__init__(self)
+            dbus.mainloop.glib.DBusGMainLoop(set_as_default=True)
+
+    def get_activities(self, query = None):
+        if not self.source:
+            return []
+
+        if self.source == "evo":
+            return [activity for activity in get_eds_tasks()
+                         if query is None or activity['name'].startswith(query)]
+
+        elif self.source == "gtg":
+            conn = self.__get_gtg_connection()
+            if not conn:
+                return []
+
+            activities = []
+
+            tasks = []
+            try:
+                tasks = conn.get_tasks()
+            except dbus.exceptions.DBusException:  #TODO too lame to figure out how to connect to the disconnect signal
+                self.__gtg_connection = None
+                return self.get_activities(query) # reconnect
+
+
+            for task in tasks:
+                if query is None or task['title'].lower().startswith(query):
+                    name = task['title']
+                    if len(task['tags']):
+                        name = "%s, %s" % (name, " ".join([tag.replace("@", "#") for tag in task['tags']]))
+
+                    activities.append({"name": name,
+                                       "category": ""})
+
+            return activities
+
+    def __get_gtg_connection(self):
+        bus = dbus.SessionBus()
+        if self.__gtg_connection and bus.name_has_owner("org.GTG"):
+            return self.__gtg_connection
+
+        if bus.name_has_owner("org.GTG"):
+            self.__gtg_connection = dbus.Interface(bus.get_object('org.GTG', '/org/GTG'),
+                                                   dbus_interface='org.GTG')
+            return self.__gtg_connection
+        else:
+            return None
+
+
+
+def get_eds_tasks():
+    try:
+        sources = ecal.list_task_sources()
+        tasks = []
+        if not sources:
+            # BUG - http://bugzilla.gnome.org/show_bug.cgi?id=546825
+            sources = [('default', 'default')]
+
+        for source in sources:
+            category = source[0]
+
+            data = ecal.open_calendar_source(source[1], ecal.CAL_SOURCE_TYPE_TODO)
+            if data:
+                for task in data.get_all_objects():
+                    if task.get_status() in [ecal.ICAL_STATUS_NONE, ecal.ICAL_STATUS_INPROCESS]:
+                        tasks.append({'name': task.get_summary(), 'category' : category})
+        return tasks
+    except Exception, e:
+        logging.warn(e)
+        return []
diff --git a/src/hamster/widgets/activityentry.py b/src/hamster/widgets/activityentry.py
index 777df6d..8ebf2e8 100644
--- a/src/hamster/widgets/activityentry.py
+++ b/src/hamster/widgets/activityentry.py
@@ -21,9 +21,9 @@ import gtk, gobject
 import datetime as dt
 
 from .hamster.configuration import runtime
-
 from .hamster import stuff
 from .hamster.stuff import format_duration
+from .hamster import external
 
 class ActivityEntry(gtk.Entry):
     __gsignals__ = {
@@ -38,6 +38,7 @@ class ActivityEntry(gtk.Entry):
         self.categories = None
         self.filter = None
         self.max_results = 10 # limit popup size to 10 results
+        self.external = external.ActivitiesSource()
 
         self.popup = gtk.Window(type = gtk.WINDOW_POPUP)
 
@@ -192,6 +193,7 @@ class ActivityEntry(gtk.Entry):
 
         # do not cache as ordering and available options change over time
         self.activities = runtime.storage.get_autocomplete_activities(input_activity.activity_name)
+        self.activities.extend(self.external.get_activities(input_activity.activity_name))
         self.categories = self.categories or runtime.storage.get_categories()