pirate_add_shift_recurrence.py

   1 #!/usr/bin/python
   2
   3 import sys
   4 import os
   5 from tasklib.task import TaskWarrior
   6
   7 time_attributes = ('wait', 'until', 'scheduled')
   8
   9 def is_new_local_recurrence_child_task(task):
  10     # Do not affect tasks not spun by recurrence
  11     if not task['parent']:
  12         return False
  13
  14     # Newly created recurrence tasks actually have
  15     # modified field copied from the parent, thus
  16     # older than entry field (until their ID is generated)
  17     if (task['modified'] - task['entry']).total_seconds() < 0:
  18         return True
  19
  20 tw = TaskWarrior(data_location=os.path.dirname(os.path.dirname(sys.argv[0])))
  21
  22 def hook_shift_recurrence(task):
  23     if is_new_local_recurrence_child_task(task):
  24         parent = tw.tasks.get(uuid=task['parent'])
  25         parent_due_shift = task['due'] - parent['due']
  26         for attr in time_attributes:
  27             if parent[attr]:
  28                 task[attr] = parent[attr] + parent_due_shift