twfzf.py

#!/usr/bin/env python3
#
# timew-fzf
#
# A timewarrior extension to list and restart
# recently tracked intervals with the help of fzf

import os
import sys

from pyfzf.pyfzf import FzfPrompt
from plumbum.commands.processes import ProcessExecutionError
from timewreport.parser import TimeWarriorParser


def tags2key(tags):
    """A hashable string from a list of tags"""
    return "|".join(sorted(tags))


def get_recent_uniquely_tagged_intervals(intervals):
    # Filter out open intervals
    intervals = [i for i in intervals if not i.is_open()]

    tags2interval = dict()
    for interval in intervals:
        key = tags2key(interval.get_tags())
        tags2interval[key] = interval

    intervals = list(tags2interval.values())
    intervals.sort(key=lambda i: i.get_start(), reverse=True)

    return intervals


def get_lines_for_fzf(intervals):
    """
    Get lines to feed to fzf. We used | to delimit fields in the output
    which means you can not use | within your timewarrior tags if you want this to work
    """
    lines = []

    for interval in get_recent_uniquely_tagged_intervals(intervals):
        start = interval.get_start()
        end = interval.get_end()
        minutes = int((end - start).seconds / 60)
        tags = interval.get_tags()

        # Build the tags string
        tags_str = ""

        for tag in tags:
            # tags with spaces needs to be in quotation marks
            if " " in tag:
                tags_str += f'"{tag}" '
            else:
                tags_str += f"{tag} "
        lines.append(f"{start} |{minutes:4}| {tags_str}")
    return lines


try:

    # Get the parser, config and intervals
    parser = TimeWarriorParser(sys.stdin)
    tw_config = parser.get_config()
    intervals = parser.get_intervals()

    config_printonly = tw_config.get_boolean("printonly", False)

    # Launch fzf and get a selection
    selection = FzfPrompt().prompt(get_lines_for_fzf(intervals), "--no-sort --exact")[0]
    tags_selection = selection.split("|")[-1].strip()

    # Put the cmd together
    cmd = f"timew start {tags_selection}"

    if config_printonly:
        # Print it
        print(cmd)
    else:
        # Run it
        os.system(cmd)

except ProcessExecutionError:
    # We get here if the user exited fzf without making a selection
    # in which case we will do nothing
    pass