mirror of
https://github.com/tbabej/taskwiki.git
synced 2025-08-19 06:43:06 +02:00
Move VimwikiTask into separate module
This commit is contained in:
Tomas Babej
parent
fb90391538
commit
7268a659b5
2 changed files with 180 additions and 173 deletions
178
taskwiki/task.py
Normal file
178
taskwiki/task.py
Normal file
|
|
@ -0,0 +1,178 @@
|
|||
import vim
|
||||
|
||||
from regexp import *
|
||||
from tasklib.task import Task
|
||||
|
||||
|
||||
def convert_priority_from_tw_format(priority):
|
||||
return {None: 0, 'L': 1, 'M': 2, 'H': 3}[priority]
|
||||
|
||||
|
||||
def convert_priority_to_tw_format(priority):
|
||||
return {0: None, 1: 'L', 2: 'M', 3: 'H'}[priority]
|
||||
|
||||
|
||||
class VimwikiTask(object):
|
||||
def __init__(self, line, position, tw, cache):
|
||||
"""
|
||||
Constructs a Vimwiki task from line at given position at the buffer
|
||||
"""
|
||||
self.tw = tw
|
||||
self.cache = cache
|
||||
|
||||
match = re.search(GENERIC_TASK, line)
|
||||
self.indent = match.group('space')
|
||||
self.text = match.group('text')
|
||||
self.uuid = match.group('uuid') # can be None for new tasks
|
||||
self.due = match.group('due') # TODO: convert to proper timestamp
|
||||
self.completed_mark = match.group('completed')
|
||||
self.completed = self.completed_mark is 'X'
|
||||
self.line_number = position
|
||||
self.priority = len(match.group('priority') or []) # This is either 0,1,2 or 3
|
||||
|
||||
# We need to track depedency set in a extra attribute, since
|
||||
# this may be a new task, and hence it need not to be saved yet.
|
||||
# We circumvent this problem by iteration order in the TaskCache
|
||||
self.add_dependencies = set()
|
||||
|
||||
# First set the task attribute to None, then try to load it, if possible
|
||||
self.task = None
|
||||
|
||||
if self.uuid:
|
||||
try:
|
||||
self.task = self.tw.tasks.get(uuid=self.uuid)
|
||||
except Task.DoesNotExist:
|
||||
self.task = Task(self.tw)
|
||||
# If task cannot be loaded, we need to remove the UUID
|
||||
vim.command('echom "UUID not found: %s,'
|
||||
'will be replaced if saved"' % self.uuid)
|
||||
self.uuid = None
|
||||
else:
|
||||
self.task = Task(self.tw)
|
||||
|
||||
# To get local time aware timestamp, we need to convert to from local datetime
|
||||
# to UTC time, since that is what tasklib (correctly) uses
|
||||
if self.due:
|
||||
# With strptime, we get a native datetime object
|
||||
try:
|
||||
due_native_datetime = datetime.strptime(self.due, DATETIME_FORMAT)
|
||||
except ValueError:
|
||||
try:
|
||||
due_native_datetime = datetime.strptime(self.due, DATE_FORMAT)
|
||||
except ValueError:
|
||||
vim.command('echom "Taskwiki: Invalid timestamp on line %s, '
|
||||
'ignored."' % self.line_number)
|
||||
|
||||
# We need to interpret it as timezone aware object in user's timezone
|
||||
# This properly handles DST, timezone offset and everything
|
||||
due_local_datetime = local_timezone.localize(due_native_datetime)
|
||||
# Convert to UTC
|
||||
due_utc_datetime = due_local_datetime.astimezone(pytz.utc)
|
||||
self.due = due_utc_datetime
|
||||
|
||||
self.parent = self.find_parent_task()
|
||||
|
||||
# Make parent task dependant on this task
|
||||
if self.parent:
|
||||
self.parent.add_dependencies |= set([self])
|
||||
|
||||
self.project = self.find_project()
|
||||
|
||||
@property
|
||||
def priority_from_tw_format(self):
|
||||
return convert_priority_from_tw_format(self.task['priority'])
|
||||
|
||||
@property
|
||||
def priority_to_tw_format(self):
|
||||
return convert_priority_to_tw_format(self.priority)
|
||||
|
||||
@property
|
||||
def due_local_tz_string(self):
|
||||
if not self.due:
|
||||
return ''
|
||||
|
||||
due_local_datetime = self.due.astimezone(local_timezone)
|
||||
return due_local_datetime.strftime(DATETIME_FORMAT)
|
||||
|
||||
@property
|
||||
def tainted(self):
|
||||
return any([
|
||||
self.task['description'] != self.text,
|
||||
self.task['priority'] != self.priority_to_tw_format,
|
||||
self.task['due'] != self.due,
|
||||
self.task['project'] != self.project,
|
||||
])
|
||||
|
||||
def save_to_tw(self):
|
||||
|
||||
# Push the values to the Task only if the Vimwiki representation
|
||||
# somehow differs
|
||||
# TODO: Check more than description
|
||||
if self.tainted:
|
||||
self.task['description'] = self.text
|
||||
self.task['priority'] = self.priority_to_tw_format
|
||||
self.task['due'] = self.due
|
||||
# TODO: this does not solve the issue of changed or removed deps (moved task)
|
||||
self.task['depends'] |= set(s.task for s in self.add_dependencies
|
||||
if not s.task.completed)
|
||||
# Since project is not updated in vimwiki on change per task, push to TW only
|
||||
# if defined
|
||||
if self.project:
|
||||
self.task['project'] = self.project
|
||||
self.task.save()
|
||||
|
||||
# If we saved the task, we need to update. Hooks may have chaned data.
|
||||
self.update_from_tw(refresh=False)
|
||||
|
||||
# Load the UUID
|
||||
if not self.uuid:
|
||||
self.uuid = self.task['uuid']
|
||||
|
||||
# Mark task as done. This works fine with already completed tasks.
|
||||
if self.completed and (self.task.pending or self.task.waiting):
|
||||
self.task.done()
|
||||
|
||||
|
||||
def update_from_tw(self, refresh=False):
|
||||
if not self.task.saved:
|
||||
return
|
||||
|
||||
# We refresh only if specified, since sometimes we
|
||||
# know that the TW task is up to date (e.g. because we just
|
||||
# loaded the object)
|
||||
if refresh:
|
||||
self.task.refresh()
|
||||
|
||||
self.text = self.task['description']
|
||||
self.priority = self.priority_from_tw_format
|
||||
self.completed = (self.task['status'] == u'completed')
|
||||
self.due = self.task['due']
|
||||
self.project = self.task['project']
|
||||
|
||||
def update_in_buffer(self):
|
||||
vim.current.buffer[self.line_number] = str(self)
|
||||
|
||||
def __str__(self):
|
||||
return ''.join([
|
||||
self.indent,
|
||||
'* [',
|
||||
'X' if self.completed else self.completed_mark,
|
||||
'] ',
|
||||
self.text if self.text else 'TEXT MISSING?',
|
||||
' ' + '!' * self.priority if self.priority else '',
|
||||
' ' + self.due_local_tz_string if self.due else '',
|
||||
' #' + self.uuid if self.uuid else '',
|
||||
])
|
||||
|
||||
def find_parent_task(self):
|
||||
for i in reversed(range(0, self.line_number)):
|
||||
if re.search(GENERIC_TASK, vim.current.buffer[i]):
|
||||
task = self.cache[i]
|
||||
if len(task.indent) < len(self.indent):
|
||||
return task
|
||||
|
||||
def find_project(self):
|
||||
for i in reversed(range(0, self.line_number)):
|
||||
match = re.search(PROJECT_DEFINITION, vim.current.buffer[i])
|
||||
if match:
|
||||
return match.group('project')
|
||||
|
|
@ -11,6 +11,7 @@ vim.command("echom '%s'" % vim.eval("s:plugin_path"))
|
|||
sys.path.insert(0, vim.eval("s:plugin_path") + "/taskwiki")
|
||||
|
||||
from regexp import *
|
||||
from task import VimwikiTask
|
||||
|
||||
"""
|
||||
How this plugin works:
|
||||
|
|
@ -38,7 +39,7 @@ class TaskCache(object):
|
|||
task = self.cache.get(key)
|
||||
|
||||
if task is None:
|
||||
task = VimwikiTask(vim.current.buffer[key], key)
|
||||
task = VimwikiTask(vim.current.buffer[key], key, tw, self)
|
||||
self.cache[key] = task
|
||||
|
||||
return task
|
||||
|
|
@ -62,178 +63,6 @@ class TaskCache(object):
|
|||
cache = TaskCache()
|
||||
|
||||
|
||||
def convert_priority_from_tw_format(priority):
|
||||
return {None: 0, 'L': 1, 'M': 2, 'H': 3}[priority]
|
||||
|
||||
|
||||
def convert_priority_to_tw_format(priority):
|
||||
return {0: None, 1: 'L', 2: 'M', 3: 'H'}[priority]
|
||||
|
||||
|
||||
class VimwikiTask(object):
|
||||
def __init__(self, line, position):
|
||||
"""
|
||||
Constructs a Vimwiki task from line at given position at the buffer
|
||||
"""
|
||||
|
||||
match = re.search(GENERIC_TASK, line)
|
||||
self.indent = match.group('space')
|
||||
self.text = match.group('text')
|
||||
self.uuid = match.group('uuid') # can be None for new tasks
|
||||
self.due = match.group('due') # TODO: convert to proper timestamp
|
||||
self.completed_mark = match.group('completed')
|
||||
self.completed = self.completed_mark is 'X'
|
||||
self.line_number = position
|
||||
self.priority = len(match.group('priority') or []) # This is either 0,1,2 or 3
|
||||
|
||||
# We need to track depedency set in a extra attribute, since
|
||||
# this may be a new task, and hence it need not to be saved yet.
|
||||
# We circumvent this problem by iteration order in the TaskCache
|
||||
self.add_dependencies = set()
|
||||
|
||||
# First set the task attribute to None, then try to load it, if possible
|
||||
self.task = None
|
||||
|
||||
if self.uuid:
|
||||
try:
|
||||
self.task = tw.tasks.get(uuid=self.uuid)
|
||||
except Task.DoesNotExist:
|
||||
self.task = Task(tw)
|
||||
# If task cannot be loaded, we need to remove the UUID
|
||||
vim.command('echom "UUID not found: %s,'
|
||||
'will be replaced if saved"' % self.uuid)
|
||||
self.uuid = None
|
||||
else:
|
||||
self.task = Task(tw)
|
||||
|
||||
# To get local time aware timestamp, we need to convert to from local datetime
|
||||
# to UTC time, since that is what tasklib (correctly) uses
|
||||
if self.due:
|
||||
# With strptime, we get a native datetime object
|
||||
try:
|
||||
due_native_datetime = datetime.strptime(self.due, DATETIME_FORMAT)
|
||||
except ValueError:
|
||||
try:
|
||||
due_native_datetime = datetime.strptime(self.due, DATE_FORMAT)
|
||||
except ValueError:
|
||||
vim.command('echom "Taskwiki: Invalid timestamp on line %s, '
|
||||
'ignored."' % self.line_number)
|
||||
|
||||
# We need to interpret it as timezone aware object in user's timezone
|
||||
# This properly handles DST, timezone offset and everything
|
||||
due_local_datetime = local_timezone.localize(due_native_datetime)
|
||||
# Convert to UTC
|
||||
due_utc_datetime = due_local_datetime.astimezone(pytz.utc)
|
||||
self.due = due_utc_datetime
|
||||
|
||||
self.parent = self.find_parent_task()
|
||||
|
||||
# Make parent task dependant on this task
|
||||
if self.parent:
|
||||
self.parent.add_dependencies |= set([self])
|
||||
|
||||
self.project = self.find_project()
|
||||
|
||||
@property
|
||||
def priority_from_tw_format(self):
|
||||
return convert_priority_from_tw_format(self.task['priority'])
|
||||
|
||||
@property
|
||||
def priority_to_tw_format(self):
|
||||
return convert_priority_to_tw_format(self.priority)
|
||||
|
||||
@property
|
||||
def due_local_tz_string(self):
|
||||
if not self.due:
|
||||
return ''
|
||||
|
||||
due_local_datetime = self.due.astimezone(local_timezone)
|
||||
return due_local_datetime.strftime(DATETIME_FORMAT)
|
||||
|
||||
@property
|
||||
def tainted(self):
|
||||
return any([
|
||||
self.task['description'] != self.text,
|
||||
self.task['priority'] != self.priority_to_tw_format,
|
||||
self.task['due'] != self.due,
|
||||
self.task['project'] != self.project,
|
||||
])
|
||||
|
||||
def save_to_tw(self):
|
||||
|
||||
# Push the values to the Task only if the Vimwiki representation
|
||||
# somehow differs
|
||||
# TODO: Check more than description
|
||||
if self.tainted:
|
||||
self.task['description'] = self.text
|
||||
self.task['priority'] = self.priority_to_tw_format
|
||||
self.task['due'] = self.due
|
||||
# TODO: this does not solve the issue of changed or removed deps (moved task)
|
||||
self.task['depends'] |= set(s.task for s in self.add_dependencies
|
||||
if not s.task.completed)
|
||||
# Since project is not updated in vimwiki on change per task, push to TW only
|
||||
# if defined
|
||||
if self.project:
|
||||
self.task['project'] = self.project
|
||||
self.task.save()
|
||||
|
||||
# If we saved the task, we need to update. Hooks may have chaned data.
|
||||
self.update_from_tw(refresh=False)
|
||||
|
||||
# Load the UUID
|
||||
if not self.uuid:
|
||||
self.uuid = self.task['uuid']
|
||||
|
||||
# Mark task as done. This works fine with already completed tasks.
|
||||
if self.completed and (self.task.pending or self.task.waiting):
|
||||
self.task.done()
|
||||
|
||||
|
||||
def update_from_tw(self, refresh=False):
|
||||
if not self.task.saved:
|
||||
return
|
||||
|
||||
# We refresh only if specified, since sometimes we
|
||||
# know that the TW task is up to date (e.g. because we just
|
||||
# loaded the object)
|
||||
if refresh:
|
||||
self.task.refresh()
|
||||
|
||||
self.text = self.task['description']
|
||||
self.priority = self.priority_from_tw_format
|
||||
self.completed = (self.task['status'] == u'completed')
|
||||
self.due = self.task['due']
|
||||
self.project = self.task['project']
|
||||
|
||||
def update_in_buffer(self):
|
||||
vim.current.buffer[self.line_number] = str(self)
|
||||
|
||||
def __str__(self):
|
||||
return ''.join([
|
||||
self.indent,
|
||||
'* [',
|
||||
'X' if self.completed else self.completed_mark,
|
||||
'] ',
|
||||
self.text if self.text else 'TEXT MISSING?',
|
||||
' ' + '!' * self.priority if self.priority else '',
|
||||
' ' + self.due_local_tz_string if self.due else '',
|
||||
' #' + self.uuid if self.uuid else '',
|
||||
])
|
||||
|
||||
def find_parent_task(self):
|
||||
for i in reversed(range(0, self.line_number)):
|
||||
if re.search(GENERIC_TASK, vim.current.buffer[i]):
|
||||
task = cache[i]
|
||||
if len(task.indent) < len(self.indent):
|
||||
return task
|
||||
|
||||
def find_project(self):
|
||||
for i in reversed(range(0, self.line_number)):
|
||||
match = re.search(PROJECT_DEFINITION, vim.current.buffer[i])
|
||||
if match:
|
||||
return match.group('project')
|
||||
|
||||
|
||||
def update_from_tw():
|
||||
"""
|
||||
Updates all the incomplete tasks in the vimwiki file if the info from TW is different.
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue