def testGetOrCreateKeyword(self):
# interactive
tui.addInputAnswers("y")
dbutils.getOrCreateKeyword("k1")
self._assertOneObject(Keyword.selectBy(name="k1"))
# !interactive
dbutils.getOrCreateKeyword("k2", interactive=False)
self._assertOneObject(Keyword.selectBy(name="k2"))
def getOrCreateKeyword(keywordName, interactive=True):
"""Get a keyword by its name. Create it if needed
@param keywordName: keyword name as a string
@param interactive: Ask user before creating keyword (this is the default)
@type interactive: Bool
@return: Keyword instance or None if user cancel creation"""
result = Keyword.selectBy(name=keywordName)
result = list(result)
if len(result):
return result[0]
if interactive and not tui.confirm("Keyword '%s' does not exist, create it" % keywordName):
return None
keyword = Keyword(name=keywordName)
print "Added keyword '%s'" % keywordName
return keyword
