Editorial Workflows

import editor
import console
import os
import re
import sys
import codecs
import workflow
from StringIO import StringIO
import time

theme = editor.get_theme()
workflow.set_variable('CSS', workflow.get_variable('CSS Dark' if theme == 'Dark' else 'CSS Light'))

today_date = time.strftime('%Y-%m-%d')

workflow.set_variable("todayDate", today_date)

p = editor.get_path()

start_pattern = r'(@start|@due|@started)\((20[1-9][0-9]\-[0-1][0-9]\-[0-3][0-9])\)'

term = start_pattern
pattern = re.compile(term, flags=re.IGNORECASE)

from urllib import quote
dir = os.path.split(p)[0]
valid_extensions = set(['.txt', '.md', '.markd', '.text', '.mdown', '.taskpaper'])
html = StringIO()
match_count = 0
for w in os.walk(dir):
	dir_path = w[0]
	filenames = w[2]
	for name in filenames:
		full_path = os.path.join(dir_path, name)
		ext = os.path.splitext(full_path)[1]
		if ext.lower() not in valid_extensions:
			continue
		found_snippets = []
		i = 0
		try:
			with codecs.open(full_path, 'r', 'utf-8') as f:
				for line in f:
					for match in re.finditer(pattern, line):
						start = max(0, match.start(0) - 100)
						end = min(len(line)-1, match.end(0) + 100)
						snippet = (line[start:match.start(0)],
						           match.group(0),
						           line[match.end(0):end],
						           match.start(0) + i,
						           match.end(0) + i)
						if time.strptime(match.group(2), '%Y-%m-%d') <= time.strptime(today_date, '%Y-%m-%d'):
							found_snippets.append(snippet)
					i += len(line)
		except UnicodeDecodeError, e:
			pass
		if len(found_snippets) > 0:
			match_count += 1
			root, rel_path = editor.to_relative_path(full_path)
			ed_url = 'editorial://open/' + quote(rel_path) + '?root=' + root
			html.write('<h2><a href="' + ed_url + '">' + name + '</a></h2>')
			for snippet in found_snippets:
				start = snippet[3]
				end = snippet[4]
				select_url = 'editorial://open/' + quote(rel_path) + '?root=' + root
				select_url += '&selection=' + str(start) + '-' + str(end)
				html.write('<a class="result-box" href="' + select_url + '">' + snippet[0] + '<span class="highlight">' + snippet[1] + '</span>' + snippet[2] + '</a>')
if match_count == 0:
	html.write('<p>No matches found.</p>')

workflow.set_output(html.getvalue())