class Controller():
configDataBase = "databasepath"
tempFilePath = "tempfilepath"
def __init__(self, log, config):
'''Constructor'''
self.actions = {
"searchAction": self.searchAction,
"entryChangeAction": self.entryChangeAction,
"newAction": self.newEntryAction,
"showEntryAction": self.entryClickedInVSearch,
"closedAction": self.closeTabAction,
"tabChangeAction": self.tabChangeAction,
"deleteAction": self.deleteEntryAction,
"pathChangeAction": self.changePathAction,
"newImageAction": self.newImageAction,
"imageSelectedAction": self.newFileOrImageSelectedAction,
"addTagAction": self.newTagAction,
"deleteTagAction": self.deleteTagAction,
"deleteImageAction": self.deleteImageAction,
"deleteFileAction": self.deleteFileAction,
"newFileAction": self.newFileAction,
"fileSelectedAction": self.newFileOrImageSelectedAction,
"openFileAction": self.openFileAction,
"openEntryOverviewAction": self.openEntryOverviewAction
}
if log != None:
self.log = log
else:
self.log = Log("log.txt")
self.config = ConfigFile(self.log, config)
self.dbPath = self.config.getValue(self.configDataBase)
self.tempFilePath = self.config.getValue(self.tempFilePath)
self.view = View(self.log, self.dbPath, self.actions)
self.model = Model(self.log, self.dbPath)
self.log.add(self.log.Info, __file__, "init")
def run(self):
entries = self.model.getAllEntriesSorted()
self.view.drawOverview(entries)
self.view.run()
def searchAction(self, tag):
self.log.add(self.log.Info, __file__, "search for : " + tag)
results = self.model.getEntries(tag)
self.view.drawSearch(results)
def entryChangeAction(self, newName, newDescription):
'''Simply calls update name from model with current entry'''
self.view.removeEntry(self.model.currentEntry)
self.model.updateNameOfEntry(self.model.currentEntry, newName)
self.model.currentEntry.description = newDescription
self.model.updateContentOfEntry(self.model.currentEntry)
self.model.currentEntry.name = newName
self.view.drawEntry(self.model.currentEntry)
def newEntryAction(self):
'''Adds a new entry'''
newNameText = "enter name"
self.model.currentEntry = ModelEntry(self.log, newNameText)
i = 0
while self.model.hasEntry(self.model.currentEntry):
i += 1
newName = newNameText + str(i)
self.model.currentEntry.name = newName
self.model.openedEntries.append(self.model.currentEntry)
self.model.addEntry(self.model.currentEntry)
self.view.drawEntry(self.model.currentEntry)
def entryClickedInVSearch(self, entryName):
'''Shows the clicked entry'''
foundEntry = self.model.getFoundEntry(entryName)
if foundEntry != None:
self.model.currentEntry = foundEntry
self.model.openedEntries.append(foundEntry)
self.view.drawEntry(foundEntry)
def closeTabAction(self):
'''Closes the currently active tab'''
activeTab = self.view.getActiveTab()
if activeTab == "Search":
self.view.removeSearch()
self.model.currentEntry = None
else:
self.model.openedEntries.remove(self.model.currentEntry)
self.view.removeEntry(self.model.currentEntry)
def tabChangeAction(self, activeTabName):
'''Is called when tab focus changes'''
# only do something when has a valid name
if activeTabName != None:
if activeTabName == "Overview":
entries = self.model.getAllEntriesSorted()
self.view.setDeleteButton(False)
self.view.drawOverview(entries)
if activeTabName == "Search":
self.view.drawSearch(self.model.foundEntries)
for e in self.model.openedEntries:
if activeTabName == e.name:
self.model.currentEntry = e
self.view.setDeleteButton(True)
def deleteEntryAction(self):
'''Deletes the currently active entry'''
self.model.removeEntry(self.model.currentEntry)
self.view.removeEntry(self.model.currentEntry)
def changePathAction(self, newPath):
'''Changes the database path'''
self.dbPath = newPath
self.config.setValue(self.configDataBase, self.dbPath)
self.model = Model(self.log, self.dbPath)
self.view.changeDbPath(self.dbPath)
def newImageAction(self):
'''Is called when user wants to add a new image
by button click'''
self.view.showNewImageSelectDialog()
def deleteImageAction(self, imageToDelete):
'''Deletes image number imageToDelete of current entry'''
del self.model.currentEntry.images[imageToDelete]
self.model.updateContentOfEntry(self.model.currentEntry)
self.view.removeEntry(self.model.currentEntry)
self.view.drawEntry(self.model.currentEntry)
def newTagAction(self, newTag):
'''This action is called when user entered a new tag'''
self.model.currentEntry.tags.append(newTag)
self.model.updateContentOfEntry(self.model.currentEntry)
self.view.removeEntry(self.model.currentEntry)
self.view.drawEntry(self.model.currentEntry)
def deleteTagAction(self, tagToDelete):
'''Is called when user deletes a tag'''
self.model.currentEntry.tags.remove(tagToDelete)
self.model.updateContentOfEntry(self.model.currentEntry)
self.view.removeEntry(self.model.currentEntry)
self.view.drawEntry(self.model.currentEntry)
def newFileAction(self):
'''Is called when user wants to add a new file
by button click'''
self.view.showNewFileSelectDialog()
def newFileOrImageSelectedAction(self, filename):
'''Is called when user has selected a new file/image. Method
adds the file/image to the model and shows it in view'''
self.log.add(self.log.Info, __file__, filename + " selected")
if os.path.exists(filename):
if self.model.currentEntry.isSupportedImageFile(filename):
self.addImage(filename)
else:
self.addFile(filename)
self.view.removeEntry(self.model.currentEntry)
self.view.drawEntry(self.model.currentEntry)
def deleteFileAction(self, fileToDelete):
'''Deletes file in current entry'''
del self.model.currentEntry.files[fileToDelete]
self.model.updateContentOfEntry(self.model.currentEntry)
self.view.removeEntry(self.model.currentEntry)
self.view.drawEntry(self.model.currentEntry)
def addFile(self, filename):
'''Adds a file to currentEntry and updates the view'''
f = open(filename, "rb")
content = f.read()
f.close()
name = os.path.basename(filename)
self.model.currentEntry.files[name] = content
self.model.updateContentOfEntry(self.model.currentEntry)
self.view.removeEntry(self.model.currentEntry)
self.view.drawEntry(self.model.currentEntry)
def addImage(self, filename):
'''Adds an image to current entry and updates the view'''
f = open(filename, "rb")
content = f.read()
f.close()
name = os.path.basename(filename)
self.model.currentEntry.images[name] = content
self.model.updateContentOfEntry(self.model.currentEntry)
self.view.removeEntry(self.model.currentEntry)
self.view.drawEntry(self.model.currentEntry)
def openFileAction(self, filename):
'''Opens a file of current entry'''
temppath = os.path.abspath(self.tempFilePath)
if not os.path.exists(temppath):
os.makedirs(temppath)
path = temppath + "\\" + filename
if filename in self.model.currentEntry.files:
content = self.model.currentEntry.files[filename]
elif filename in self.model.currentEntry.images:
content = self.model.currentEntry.images[filename]
else:
self.log.add(self.log.Warning, __file__, filename + " not in db")
return
f = open(path, "wb")
f.write(content)
f.close()
os.startfile(path)
def openEntryOverviewAction(self, entryName):
'''Opens the clicked entry, which is currently showed in overview'''
entry = self.model.getEntryByName(entryName)
if entry != None:
self.model.currentEntry = entry
self.model.openedEntries.append(entry)
self.view.drawEntry(entry)
class TestLog(unittest.TestCase):
logfile = os.path.dirname(os.path.abspath(__file__)) + "\log.txt"
def setUp(self):
self.capturedOutput = StringIO.StringIO()
sys.stdout = self.capturedOutput
self.log = Log(self.logfile)
def tearDown(self):
sys.stdout = sys.__stdout__
if os.path.exists(self.logfile):
os.remove(self.logfile)
def testIfWritesNotBeforeFirstAdd(self):
'''Tests whether log is not printing to console and file before
add has been called'''
expected = ""
actual = self.capturedOutput.getvalue()
self.assertEqual(expected, actual, "Log has already printed out:-(")
# Logfile should be created at first call of add. So it should not
# exist here
self.assertFalse(os.path.exists(self.logfile),
"Logfile already exists")
def testConsole(self):
'''Tests console output and header by adding two log entries'''
self.log.add(Log.Error, "File1", "This is the text")
self.log.add(Log.Error, "File2", "This is the text")
expected = "Severity\tFile Text\n"
expected += "Error\t\tFile1 This is the text \n"
expected += "Error\t\tFile2 This is the text \n"
actual = self.capturedOutput.getvalue()
self.assertEqual(
expected, actual,
"\n\nExpected:\n" + expected + "\n\nActual:\n" + actual)
def testLogfile(self):
'''Tests logfile output and header by adding two log entries'''
time = datetime.datetime.now()
self.log.now = time
self.log.add(Log.Warning, "File1", "This is the text")
self.log.add(Log.Info, "File2", "This is the text")
# Overwrite current time for this test.
# Otherwise this test would not run stable
user = getpass.getuser()
s = "LOGFILE\n"
s += "Date: " + repr(time.day) + \
"." + repr(time.month) + "." + repr(time.year) + "\n"
s += "Time: " + repr(time.hour) + ":" + repr(time.minute) + \
":" + repr(time.second) + "\n"
s += "User: "******"\n"
s += "-----------------------------\n"
s += "Severity\tFile Text\n"
s += "Warning\t\tFile1 This is the text \n"
s += "Info\t\tFile2 This is the text \n"
f = open(self.logfile, 'r')
actual = f.read()
f.close()
self.assertEqual(s, actual,
"\n\nExpected:\n" + s + "\n\nActual:\n" + actual)
def testLongStrings(self):
'''Tests whether log shortend name and text correctly'''
self.log.add(Log.Error, "Verylongsillyfilename",
"This is the text which is much longer than allowed")
expected = "Severity\tFile Text\n"
expected += "Error\t\tVerylongsillyfi This is the text which is much\n"
actual = self.capturedOutput.getvalue()
self.assertEqual(
expected, actual,
"\n\nExpected:\n" + expected + "\n\nActual:\n" + actual)
def testResizeLongString(self):
s = "abcdefgh"
expected = "abcd"
actual = self.log.resizeString(s, 4)
self.assertEqual(
expected, actual,
"\n\nExpected:\n" + expected + "\n\nActual:\n" + actual)
def testResizeShortString(self):
s = "abc"
expected = "abc "
actual = self.log.resizeString(s, 5)
self.assertEqual(
expected, actual,
"\n\nExpected:\n" + expected + "\n\nActual:\n" + actual)