def testGetOrCreateProject(self):
# interactive
tui.addInputAnswers("y")
dbutils.getOrCreateProject("p1")
self._assertOneObject(Project.selectBy(name="p1"))
# !interactive
dbutils.getOrCreateProject("p2", interactive=False)
self._assertOneObject(Project.selectBy(name="p2"))
def testTitleFormater(self):
dbutils.getOrCreateProject("x", interactive=False)
dbutils.getOrCreateKeyword("k1", interactive=False)
dbutils.getOrCreateKeyword("k2", interactive=False)
t1 = dbutils.addTask("x", "t1", {})
t2 = dbutils.addTask("x", "t2", {"k1":None, "k2":12})
longerTask = dbutils.addTask("x", "A longer task name", {})
longerTask.description = "And it has a description"
out = StringIO()
renderer = TextListRenderer(out, termWidth=80)
renderer.addTaskList("Foo", [t1])
self.assertEquals(renderer.maxTitleWidth, 5)
renderer.end()
expected=(u"\n" \
+ "%(CYAN)s Foo %(RESET)s\n" \
+ "%(BOLD)sID|Title|U |S|Age |Due date%(RESET)s\n" \
+ "--------------------------------\n" \
+ "1 |t1 |0 |N|0m | \n" \
)% dict(CYAN=C.CYAN, RESET=C.RESET, BOLD=C.BOLD)
testutils.multiLinesAssertEqual(self, out.getvalue(), expected)
out = StringIO()
renderer = TextListRenderer(out, termWidth=80)
renderer.addTaskList("Foo", [t1, t2])
self.assertEquals(renderer.maxTitleWidth, 11)
renderer.end()
expected=(u"\n" \
+ "%(CYAN)s Foo %(RESET)s\n" \
+ "%(BOLD)sID|Title |U |S|Age |Due date%(RESET)s\n" \
+ "--------------------------------------\n" \
+ "1 |t1 |0 |N|0m | \n" \
+ "2 |t2 (%(BOLD)sk1, k2%(RESET)s)|0 |N|0m | \n" \
)% dict(CYAN=C.CYAN, RESET=C.RESET, BOLD=C.BOLD)
testutils.multiLinesAssertEqual(self, out.getvalue(), expected)
out = StringIO()
renderer = TextListRenderer(out, termWidth=80)
renderer.addTaskList("Foo", [t2, longerTask])
self.assertEquals(renderer.maxTitleWidth, len(longerTask.title) + 1)
renderer.end()
expected=(u"\n" \
+ "%(CYAN)s Foo %(RESET)s\n" \
+ "%(BOLD)sID|Title |U |S|Age |Due date%(RESET)s\n" \
+ "----------------------------------------------\n" \
+ "2 |t2 (%(BOLD)sk1, k2%(RESET)s) |0 |N|0m | \n" \
+ "3 |A longer task name*|0 |N|0m | \n" \
)% dict(CYAN=C.CYAN, RESET=C.RESET, BOLD=C.BOLD)
testutils.multiLinesAssertEqual(self, out.getvalue(), expected)
def do_t_project(self, line):
"""Set task's project.
t_project <id> <project>"""
tokens = line.split(" ")
if len(tokens)!=2:
raise YokadiException("You should give two arguments: <task id> <project>")
task=dbutils.getTaskFromId(tokens[0])
projectName = tokens[1]
task.project = dbutils.getOrCreateProject(projectName)
if task.project:
print "Moved task '%s' to project '%s'" % (task.title, projectName)
def do_p_edit(self, line):
"""Edit a project.
p_edit <project name>"""
project = dbutils.getOrCreateProject(line, createIfNeeded=False)
if not project:
raise YokadiException("Project does not exist.")
# Create project line
projectLine = parseutils.createLine(project.name, "", project.getKeywordDict())
# Edit
line = tui.editLine(projectLine)
# Update project
projectName, garbage, keywordDict = parseutils.parseLine(line)
if garbage:
raise YokadiException("Cannot parse line, got garbage (%s)" % garbage)
if not dbutils.createMissingKeywords(keywordDict.keys()):
return
project.name = projectName
project.setKeywordDict(keywordDict)
