on-add-shift-all-recurrence

   1 #!/usr/bin/python
   2
   3 import sys
   4 import os
   5 from tasklib.task import Task, 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 task = Task.from_input()
  21 tw = TaskWarrior(data_location=os.path.dirname(os.path.dirname(sys.argv[0])))
  22 tw.config.update(dict(recurrence="no"))
  23
  24 if is_new_local_recurrence_child_task(task):
  25     parent = tw.tasks.get(uuid=task['parent'])
  26     parent_due_shift = task['due'] - parent['due']
  27     for attr in time_attributes:
  28         if parent[attr]:
  29             task[attr] = parent[attr] + parent_due_shift
  30
  31 print task.export_data()