def test_search_entries_by_title():
db_handler = DBHandlerClass()
test1_title = generate_random_strings(10)
test1_tag = [generate_random_strings(4), generate_random_strings(3)]
test1_content = "File"
db_handler.insert_to_entries_table(test1_title, test1_tag, test1_content)
result1 = db_handler.search_entries_by_title(test1_title)
result2 = db_handler.search_entries_by_title(generate_random_strings(5))
r1 = True
rr1 = False
r2 = False
if not isinstance(result1, bool):
r1 = False
if result1["Title"] == test1_title:
rr1 = True
if result2 is None:
r2 = True
assert rr1 or r1
assert r2 or r1
def test_insert_to_entries_table():
test1_title = generate_random_strings(10)
test1_tag = [generate_random_strings(4), generate_random_strings(3)]
test1_content = "File"
test2_title = generate_random_strings(10)
test2_tag = [generate_random_strings(4), generate_random_strings(3)]
test2_content = None
test3_title = generate_random_strings(10)
test3_tag = []
test3_content = "File"
test4_title = ""
test4_tag = [generate_random_strings(4), generate_random_strings(3)]
test4_content = "File" # TODO implement the new test cases
db_handler = DBHandlerClass()
# print "Test Case one with a non-Null Title "
assert db_handler.insert_to_entries_table(
generate_random_strings(10),
[generate_random_strings(4),
generate_random_strings(4)], "File")
# print "Test Case two with a Null Title "
assert not db_handler.insert_to_entries_table(
"", [generate_random_strings(4),
generate_random_strings(4)], "File")
class Main(QtGui.QMainWindow):
"""Main Control class.
This is the main class that controls GUI and DataBase.
"""
def __init__(self, parent=None):
"""Initiate GUI and Database.
Args:
db_handler (DBHandlerClass) : calls and initialize database.
ui (QMainWindow) : calls and initialize main window.
"""
self.current_selection = False
self.db_handler = DBHandlerClass()
super(Main, self).__init__(parent)
self.ui = Ui_MainWindow()
self.ui.setupUi(self)
self.ui.center_widget.entry.journalEntry.textChanged.connect(self.text_triggered)
self.ui.center_widget.journalList.currentItemChanged.connect(self.item_changed)
self.ui.actionNew_Entry.triggered.connect(self.new_entry_action)
self.ui.action_Save_Entry.triggered.connect(self.save_button_action)
self.ui.action_edit_entry.triggered.connect(self.edit_button_action)
self.ui.action_Delete_Entry.triggered.connect(self.delete_button_action)
self.ui.action_Cancel.triggered.connect(self.action_button_action)
self.ui.center_widget.entry.journalEntry.setReadOnly(True)
self.refresh_list_view()
self.refresh_list_view()
def action_button_action(self):
self.current_selection = None
self.ui.center_widget.journalList.setEnabled(True)
self.ui.center_widget.entry.journalEntry.setReadOnly(True)
self.ui.center_widget.journalList.setEnabled(True)
def text_triggered(self):
"""Update text to markdown text simultaneously.
Args:
raw (String) : Recieves the string from entry every time an input
is made
ntxt (html string) : Converted html string
"""
raw = self.ui.center_widget.entry.journalEntry.toPlainText()
md = markdown.Markdown()
# raw = raw.encode('utf-8')
ntxt = md.convert(raw)
self.ui.center_widget.view.journalView.setHtml(ntxt)
def item_changed(self):
"""Not my portion."""
item = self.ui.center_widget.journalList.currentItem()
result = self.db_handler.search_entries_by_id(item.toolTip())[0]
st = result["MarkdownFile"].encode('utf-8')
self.ui.center_widget.entry.journalEntry.setPlainText(st)
self.text_triggered()
def remove_all_list(self):
while self.ui.center_widget.journalList.count() != 0:
self.ui.center_widget.journalList.clear()
def refresh_list_view(self):
self.remove_all_list()
results = self.db_handler.get_all()
for x in results:
item = QtGui.QListWidgetItem('Title: ' + str(x['Title']))
item.setToolTip(str(x["_id"]))
self.ui.center_widget.journalList.addItem(item)
def edit_button_action(self):
item = self.ui.center_widget.journalList.currentItem()
self.ui.center_widget.journalList.setEnabled(False)
self.current_selection = item.toolTip()
self.ui.center_widget.entry.journalEntry.setReadOnly(False)
def delete_button_action(self):
item = self.ui.center_widget.journalList.currentItem()
result = self.db_handler.delete_entries(item.toolTip())
self.refresh_list_view()
def new_entry_action(self):
self.ui.center_widget.journalList.setEnabled(False)
self.current_selection = None
self.ui.center_widget.entry.journalEntry.setReadOnly(False)
self.ui.center_widget.entry.journalEntry.setPlainText("")
def save_button_action(self):
"""
Controls the menu Bar save button click event
:return: None
"""
if self.current_selection is not None:
result = self.db_handler.search_entries_by_id(self.current_selection)[0]
result["MarkdownFile"] = self.ui.center_widget.entry.journalEntry.toPlainText()
self.db_handler.update_entries(self.current_selection, result)
else:
self.db_handler.insert_to_entries_table(self.get_title(), [],
self.ui.center_widget.entry.journalEntry.toPlainText())
self.ui.center_widget.entry.journalEntry.setPlainText("")
self.current_selection = None
self.ui.center_widget.journalList.setEnabled(True)
self.ui.center_widget.entry.journalEntry.setReadOnly(True)
self.ui.center_widget.journalList.setEnabled(True)
self.refresh_list_view()
def get_title(self):
return "Title "+ str(randint(1,100))