diff options
Diffstat (limited to 'files/.task/hooks/on-add-default-time.py')
| -rwxr-xr-x | files/.task/hooks/on-add-default-time.py | 15 |
1 files changed, 7 insertions, 8 deletions
diff --git a/files/.task/hooks/on-add-default-time.py b/files/.task/hooks/on-add-default-time.py index ec2cc41..7f7aca9 100755 --- a/files/.task/hooks/on-add-default-time.py +++ b/files/.task/hooks/on-add-default-time.py @@ -1,26 +1,25 @@ #!/usr/bin/env python3 # Adopted from https://github.com/tbabej/task.default-date-time from datetime import time -from tasklib import Task, local_zone +from tasklib import Task DEFAULT_TIME = time(22, 0, 0) def is_local_midnight(timestamp): - return timestamp.astimezone(local_zone).time() == time(0, 0, 0) + return timestamp.astimezone().time() == time(0, 0, 0) + def set_default_time(timestamp): - return timestamp.astimezone(local_zone).replace( - hour=DEFAULT_TIME.hour, - minute=DEFAULT_TIME.minute, - second=DEFAULT_TIME.second + return timestamp.astimezone().replace( + hour=DEFAULT_TIME.hour, minute=DEFAULT_TIME.minute, second=DEFAULT_TIME.second ) task = Task.from_input() -if task['due'] and is_local_midnight(task['due']): - task['due'] = set_default_time(task['due']) +if task["due"] and is_local_midnight(task["due"]): + task["due"] = set_default_time(task["due"]) print("Default due time has been set.") print(task.export_data()) |