Add serialize/deserialize for fields and implement for due date

diff --git a/tasklib/task.py b/tasklib/task.py
index cd4b92f1d3af491bc001229406ac26f3e509c5a7..b78ed32fb4aa0aee25e9c17a4f74302ff66f39e7 100644 (file)
--- a/tasklib/task.py
+++ b/tasklib/task.py
@@ -1,4 +1,5 @@
 import copy
 import copy

+import datetime

 import json
 import os
 import subprocess
 import json
 import os
 import subprocess


@@ -6,7 +7,10 @@ import tempfile
 import uuid
 
 
 import uuid
 
 

+DATE_FORMAT = '%Y%m%dT%H%M%SZ'
+

 REPR_OUTPUT_SIZE = 10
 REPR_OUTPUT_SIZE = 10

+

 PENDING = 'pending'
 
 
 PENDING = 'pending'
 
 


@@ -24,7 +28,7 @@ class Task(object):
         self._data = data
 
     def __getitem__(self, key):
         self._data = data
 
     def __getitem__(self, key):

-        return self._data.get(key)
+        return self._get_field(key)

 
     def __setitem__(self, key, val):
         self._data[key] = val
 
     def __setitem__(self, key, val):
         self._data[key] = val


@@ -32,6 +36,22 @@ class Task(object):
     def __unicode__(self):
         return self._data.get('description')
 
     def __unicode__(self):
         return self._data.get('description')
 

+    def _get_field(self, key):
+        hydrate_func = getattr(self, 'deserialize_{0}'.format(key), lambda x:x)
+        return hydrate_func(self._data.get(key))
+
+    def _set_field(self, key, value):
+        dehydrate_func = getattr(self, 'serialize_{0}'.format(key), lambda x:x)
+        self._data[key] = dehydrate_func(value)
+
+    def serialize_due(self, date):
+        return date.strftime(DATE_FORMAT)
+
+    def deserialize_due(self, date_str):
+        if not date_str:
+            return None
+        return datetime.datetime.strptime(date_str, DATE_FORMAT)
+

     def regenerate_uuid(self):
         self['uuid'] = str(uuid.uuid4())
 
     def regenerate_uuid(self):
         self['uuid'] = str(uuid.uuid4())