Logo Search packages:      
Sourcecode: yokadi version File versions  Download package

completers.py

# -*- coding: UTF-8 -*-
"""
Implementation of completers for various Yokadi objects.

@author: Aurélien Gâteau <aurelien.gateau@free.fr>
@license: GPLv3
"""
from sqlobject import LIKE

import parseutils
from dateutil import rrule
from db import Config, Keyword, Project, Task, FREQUENCY


def computeCompleteParameterPosition(text, line, begidx, endidx):
    before = parseutils.simplifySpaces(line[:begidx].strip())
    return before.count(" ") + 1


def getItemPropertiesStartingWith(item, field, text):
    """Return a list of item.field starting with text
    @param item: the object item, example : Task, Project, Keyword...
    @param field: the item's field lookup : Project.q.name, Task.q.title, Keyword.q.name. Don't forget the magic q
    @param text: The begining of the text as a str
    @return: list of matching strings"""
    return [x.name for x in item.select(LIKE(field, text + "%"))]


class ProjectCompleter(object):
    def __init__(self, position):
        self.position = position


    def __call__(self, text, line, begidx, endidx):
        if computeCompleteParameterPosition(text, line, begidx, endidx) == self.position:
            return ["%s " % x for x in getItemPropertiesStartingWith(Project, Project.q.name, text)]
        else:
            return []


class KeywordCompleter(object):
    def __init__(self, position):
        self.position = position


    def __call__(self, text, line, begidx, endidx):
        if computeCompleteParameterPosition(text, line, begidx, endidx) == self.position:
            return getItemPropertiesStartingWith(Keyword, Keyword.q.name, text)
        else:
            return []

def projectAndKeywordCompleter(cmd, text, line, begidx, endidx):
    position=computeCompleteParameterPosition(text, line, begidx, endidx)
    position-=len(parseutils.parseParameters(line)[0]) # remove arguments from position count
    if   position == 1: # Projects
        return ["%s" % x for x in getItemPropertiesStartingWith(Project, Project.q.name, text)]
    elif position >= 2 and line[-1] != " " and line.split()[-1][0] == "@": # Keywords (we ensure that it starts with @
        return ["%s" % x for x in getItemPropertiesStartingWith(Keyword, Keyword.q.name, text)]

def confCompleter(cmd, text, line, begidx, endidx):
    return getItemPropertiesStartingWith(Config, Config.q.name, text)

def taskIdCompleter(cmd, text, line, begidx, endidx):
    #TODO: filter on parameter position
    #TODO: potential performance issue with lots of tasks, find a better way to do it
    tasks=[x for x in Task.select(Task.q.status!='done') if str(x.id).startswith(text)]
    print
    for task in tasks:
        # Move that in a renderer class ?
        print "%s: %s / %s" % (task.id, task.project.name, task.title)
    return [str(x.id) for x in tasks]

def recurrenceCompleter(cmd, text, line, begidx, endidx):
    position=computeCompleteParameterPosition(text, line, begidx, endidx)
    if position == 1: # Task id
        return taskIdCompleter(cmd, text, line, begidx, endidx)
    elif position == 2: # frequency
        return [x for x in FREQUENCY.values()+["None"] if x.lower().startswith(text.lower())]
    elif position == 3 and "weekly" in line.lower():
        return [str(x) for x in rrule.weekdays if str(x).lower().startswith(text.lower())]
# vi: ts=4 sw=4 et

Generated by  Doxygen 1.6.0   Back to index