autodoist/nextaction.py

140 lines
5.6 KiB
Python
Raw Normal View History

2014-03-10 00:11:58 -04:00
#!/usr/bin/env python
import logging
2015-08-29 14:15:09 -04:00
import argparse
2014-03-10 00:11:58 -04:00
2015-12-12 09:07:48 -05:00
# noinspection PyPackageRequirements
from todoist.api import TodoistAPI
2015-12-12 09:07:48 -05:00
import time
import sys
from datetime import datetime
def get_subitems(items, parent_item=None):
2015-08-29 13:32:24 -04:00
"""Search a flat item list for child items"""
result_items = []
found = False
if parent_item:
required_indent = parent_item['indent'] + 1
else:
required_indent = 1
for item in items:
if parent_item:
if not found and item['id'] != parent_item['id']:
continue
else:
found = True
if item['indent'] == parent_item['indent'] and item['id'] != parent_item['id']:
return result_items
elif item['indent'] == required_indent:
result_items.append(item)
return result_items
2014-11-15 14:45:52 -05:00
2014-03-10 00:11:58 -04:00
def main():
2015-08-29 14:15:09 -04:00
parser = argparse.ArgumentParser()
2015-09-05 10:56:30 -04:00
parser.add_argument('-a', '--api_key', help='Todoist API Key')
parser.add_argument('-l', '--label', help='The next action label to use', default='next_action')
parser.add_argument('-d', '--delay', help='Specify the delay in seconds between syncs', default=5, type=int)
2015-08-29 14:15:09 -04:00
parser.add_argument('--debug', help='Enable debugging', action='store_true')
parser.add_argument('--inbox', help='The method the Inbox project should be processed',
2015-09-05 10:56:30 -04:00
default='parallel', choices=['parallel', 'serial'])
parser.add_argument('--parallel_suffix', default='=')
parser.add_argument('--serial_suffix', default='-')
parser.add_argument('--hide_future', help='Hide future dated next actions until the specified number of days',
2015-09-05 10:56:30 -04:00
default=7, type=int)
2015-08-29 14:15:09 -04:00
args = parser.parse_args()
# Set debug
2015-09-05 10:56:30 -04:00
if args.debug:
2014-11-15 15:31:36 -05:00
log_level = logging.DEBUG
else:
log_level = logging.INFO
logging.basicConfig(level=log_level)
2015-08-29 14:15:09 -04:00
# Check we have a API key
if not args.api_key:
2014-11-15 14:49:52 -05:00
logging.error('No API key set, exiting...')
sys.exit(1)
2015-08-29 14:15:09 -04:00
# Run the initial sync
logging.debug('Connecting to the Todoist API')
2015-08-29 14:15:09 -04:00
api = TodoistAPI(token=args.api_key)
logging.debug('Syncing the current state from the API')
api.sync(resource_types=['projects', 'labels', 'items'])
2015-08-29 14:15:09 -04:00
# Check the next action label exists
labels = api.labels.all(lambda x: x['name'] == args.label)
if len(labels) > 0:
label_id = labels[0]['id']
2015-08-29 14:15:09 -04:00
logging.debug('Label %s found as label id %d', args.label, label_id)
else:
2015-08-29 14:15:09 -04:00
logging.error("Label %s doesn't exist, please create it or change TODOIST_NEXT_ACTION_LABEL.", args.label)
sys.exit(1)
2015-08-29 14:15:09 -04:00
def get_project_type(project_object):
"""Identifies how a project should be handled"""
name = project_object['name'].strip()
if project['name'] == 'Inbox':
return args.inbox
elif name[-1] == args.parallel_suffix:
return 'parallel'
elif name[-1] == args.serial_suffix:
return 'serial'
# Main loop
2014-11-15 14:45:52 -05:00
while True:
2015-12-12 08:54:42 -05:00
try:
api.sync(resource_types=['projects', 'labels', 'items'])
except Exception as e:
logging.exception('Error trying to sync with Todoist API: %s' % str(e))
else:
for project in api.projects.all():
project_type = get_project_type(project)
if project_type:
logging.debug('Project %s being processed as %s', project['name'], project_type)
items = sorted(api.items.all(lambda x: x['project_id'] == project['id']), key=lambda x: x['item_order'])
for item in items:
labels = item['labels']
# If its too far in the future, remove the next_action tag and skip
if args.hide_future > 0 and 'due_date_utc' in item.data and item['due_date_utc'] is not None:
due_date = datetime.strptime(item['due_date_utc'], '%a %d %b %Y %H:%M:%S +0000')
future_diff = (due_date - datetime.utcnow()).total_seconds()
if future_diff >= (args.hide_future * 86400):
if label_id in labels:
labels.remove(label_id)
logging.debug('Updating %s without label as its too far in the future', item['content'])
item.update(labels=labels)
continue
# Process item
if project_type == 'serial':
if item['item_order'] == 1:
if label_id not in labels:
labels.append(label_id)
logging.debug('Updating %s with label', item['content'])
item.update(labels=labels)
else:
if label_id in labels:
labels.remove(label_id)
logging.debug('Updating %s without label', item['content'])
item.update(labels=labels)
elif project_type == 'parallel':
if label_id not in labels:
logging.debug('Updating %s with label', item['content'])
2015-12-12 08:54:42 -05:00
labels.append(label_id)
item.update(labels=labels)
2015-12-12 08:54:42 -05:00
api.commit()
2015-08-29 14:15:09 -04:00
logging.debug('Sleeping for %d seconds', args.delay)
time.sleep(args.delay)
2014-11-15 14:45:52 -05:00
2014-03-10 00:11:58 -04:00
if __name__ == '__main__':
2014-11-15 14:45:52 -05:00
main()