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 TestView(unittest.TestCase):
def setUp(self):
self.log = Log("testlog.txt")
self.log.add = MagicMock()
self.dummy1 = MagicMock()
self.actions = {"fileSelectedAction": self.dummy1}
self.dbPath = "arxads"
self.view = View(log=self.log,
dbPath=self.dbPath,
actions=self.actions)
self.view.menubar.enableButtonClose = MagicMock()
self.view.menubar.enableButtonDelete = MagicMock()
self.view.tabs.addEntry = MagicMock()
self.view.tabs.grid = MagicMock()
self.view.tabs.setSearch = MagicMock()
self.view.tabs.removeEntry = MagicMock()
self.view.tabs.hasTabs = MagicMock()
self.view.menubar.disableButtonClose = MagicMock()
self.view.menubar.disableButtonDelete = MagicMock()
self.view.tabs.removeSearch = MagicMock()
self.view.dbPath.changePath = MagicMock()
def tearDown(self):
pass
def testDrawEntry(self):
e = "test"
self.view.drawEntry(e)
self.view.menubar.enableButtonClose.assert_called_once()
self.view.menubar.enableButtonDelete.assert_called_once()
self.view.tabs.addEntry.assert_called_once_with(e)
self.view.tabs.grid.assert_called()
def testDrawSearch(self):
e = "test"
self.view.drawSearch(e)
self.view.menubar.enableButtonClose.assert_called_once()
self.view.tabs.setSearch(e)
self.view.tabs.grid.assert_called()
def testRemoveEntry(self):
e = "test"
self.view.tabs.hasTabs.return_value = True
self.view.removeEntry(e)
self.view.tabs.removeEntry.assert_called_with(e)
self.view.tabs.hasTabs.return_value = False
self.view.removeEntry(e)
self.view.tabs.removeEntry.assert_called_with(e)
self.view.menubar.disableButtonClose.assert_called_once()
self.view.menubar.disableButtonDelete.assert_called_once()
def testRemoveSearch(self):
self.view.tabs.hasTabs.return_value = True
self.view.removeSearch()
self.view.tabs.removeSearch.assert_called()
self.view.tabs.hasTabs.return_value = False
self.view.removeSearch()
self.view.menubar.disableButtonClose.assert_called_once()
def testSetDeleteButton(self):
self.view.setDeleteButton(True)
self.view.menubar.enableButtonDelete.assert_called_once()
self.view.menubar.disableButtonDelete.assert_not_called()
self.view.menubar.enableButtonDelete.reset_mock()
self.view.setDeleteButton(False)
self.view.menubar.enableButtonDelete.assert_not_called()
self.view.menubar.disableButtonDelete.assert_called_once()
def testChangeDbPath(self):
newPath = "dsaf"
self.view.changeDbPath(newPath)
self.view.dbPath.changePath.assert_called_once_with(newPath)
def testShowFileDialog(self):
#filename = "filenameblabla"
#self.view.showFileDialog()
#self.dummy1.assert_called_once_with(filename)
#todo
pass
def testGetActiveTab(self):
self.view.tabs.getNameOfActiveTab = MagicMock()
self.view.getActiveTab()
self.view.tabs.getNameOfActiveTab.assert_called_once()