From: Rob Golding Date: Mon, 15 Apr 2013 19:23:38 +0000 (+0100) Subject: Add serialize/deserialize for fields and implement for due date X-Git-Url: https://git.madduck.net/etc/taskwarrior.git/commitdiff_plain/82fdf26527b2c656974fdc9a13e1b8231a27d78a?ds=sidebyside;hp=-c Add serialize/deserialize for fields and implement for due date --- 82fdf26527b2c656974fdc9a13e1b8231a27d78a diff --git a/tasklib/task.py b/tasklib/task.py index cd4b92f..b78ed32 100644 --- a/tasklib/task.py +++ b/tasklib/task.py @@ -1,4 +1,5 @@ import copy +import datetime import json import os import subprocess @@ -6,7 +7,10 @@ import tempfile import uuid +DATE_FORMAT = '%Y%m%dT%H%M%SZ' + REPR_OUTPUT_SIZE = 10 + PENDING = 'pending' @@ -24,7 +28,7 @@ class Task(object): 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 @@ -32,6 +36,22 @@ class Task(object): 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())