taskwarrior_hooks/due-eod.py

40 lines
1.0 KiB
Python
Executable File

#!/usr/bin/env python3
# TaskWarrior's hook that sets due date to the end of date.
# By default TaskWarrior sets it to 00:00:00.
import datetime
import json
import sys
def get_task_data() -> str:
"""
Gets task data from stdin.
It might be first line on task addition and second line on task modification.
:return: str
"""
input_data = sys.stdin.readlines()
with open("/Users/pztrn/TEMP/abracadabra", "w") as f:
f.write(input_data[-1])
return input_data[-1]
task_data_raw = get_task_data()
task_data = json.loads(task_data_raw)
if "due" in task_data:
new_due_ts = datetime.datetime.strptime(task_data["due"], "%Y%m%dT%H%M%S%z").astimezone()
if new_due_ts.hour == 00 and new_due_ts.minute == 00 and new_due_ts.second == 00:
new_due_ts = new_due_ts.replace(hour = 23, minute = 59, second = 59)
task_data["due"] = new_due_ts.astimezone(tz = datetime.timezone.utc).strftime("%Y%m%dT%H%M%SZ")
print(json.dumps(task_data, ensure_ascii = False))
sys.exit(0)