fixup! fixup! Add support for categories. This fixes #10

Author: christf

todoman/cli.py

@@ -371,6 +371,8 @@ def new(ctx, summary, list, todo_properties, read_description, interactive):
     for key, value in todo_properties.items():
         if value:
             logger.debug("property: " + key + " value: " + ','.join(value) +  " (" + str(type(value)) + ")" )
+            if key == "categories":
+                value = [v for v in value]
             setattr(todo, key, value)
     todo.summary = ' '.join(summary)
 

todoman/model.py

@@ -299,7 +299,11 @@ def set_field(self, name, value):
         self.vtodo.pop(name)
         if value:
             logger.debug("Setting field %s to %s.", name, value)
-            self.vtodo.add(name, value)
+            if name == 'categories':
+                v = icalendar.prop.vInline(','.join(value))
+                self.vtodo.add(name, v)
+            else:
+                self.vtodo.add(name, value)
 
     def serialize(self, original=None):
         """Serialize a Todo into a VTODO."""
@@ -530,7 +534,6 @@ def add_file(self, list_name, path, mtime):
             raise exceptions.AlreadyExists('file', list_name) from e
 
     def add_category(self, uid, category):
-        try:
             self._conn.execute(
                '''
                INSERT INTO categories (
@@ -648,6 +651,7 @@ def todos(
         due=None,
         start=None,
         startable=False,
+        categories=None,
         status=(
             'NEEDS-ACTION',
             'IN-PROCESS',
@@ -800,7 +804,7 @@ def _todo_from_db(self, row):
         todo.status = row['status']
         todo.description = row['description']
         todo.location = row['location']
-        todo.categories = []
+        todo.categories = None
 	#row['categories']
         # TODO: category korrekt befüllen
         todo.sequence = row['sequence']
@@ -959,6 +963,9 @@ def update_cache(self):
 
         self.cache.save_to_disk()
 
+    def todos(self, **kwargs):
+        return self.cache.todos(**kwargs)
+
     def todo(self, id, **kwargs):
         return self.cache.todo(id, **kwargs)