def storage():
ds = DataStorage()
ds.create_item("John", "Smith", "+1111111111", "05.11.1955")
ds.create_item("John", "Doe", "+122222222", "14.11.1970")
ds.create_item("Maria", "Hanasaku", "+1333333333", "11.11.1911")
ds.create_item("David", "Iroha", "+1444444444", "08.11.1855")
return ds
def test_delete_and_save_data(saved_storage):
saved_storage.delete_item(1)
saved_storage.save()
loaded_storage = DataStorage(saved_storage.filename)
assert len(loaded_storage.items) == len(saved_storage.items)
for i in range(len(loaded_storage.items)):
assert saved_storage.items[i] == loaded_storage.items[i]
def test_add_few_items():
ds = DataStorage()
ds.create_item("John", "First", "+1111111111", "05.11.2016")
ds.create_item("John", "Second", "+1111111111", "05.11.2016")
ds.create_item("John", "Third", "+1111111111", "05.11.2016")
items = ds.items
assert len(items) == 3
assert items[0].last_name == "First" and items[
1].last_name == "Second" and items[2].last_name == "Third"
def __menu_load_pressed(self):
file_opts = {
"defaultextension": ".not",
"filetypes": [('all files', '.*'), ('data files', '.not')],
"parent": self.__main_window
}
filename = tkFileDialog.askopenfilename(**file_opts)
self.__datasource = DataStorage(filename=filename)
self.__update_table()
birthdays = self.__datasource.todays_birthdays
if len(birthdays) != 0:
names = [
"%s %s" % (item.first_name, item.last_name)
for item in birthdays
]
message = "Today is birthday of %s" % ", ".join(names)
tkMessageBox.showinfo("Birthday notification", message)
def __init__(self, datasource=None):
self.__main_window = Tk()
self.__main_window.minsize(800, 600)
self.__main_window.iconbitmap("resources/quill.ico")
self.__main_window.wm_title("Notebook Demo")
self.__add_menu()
cols = ("Last Name", "First Name", "Phone", "Birthday")
tv = Treeview(self.__main_window, columns=cols, show="headings")
self.__table = tv
for col in cols:
tv.heading(col, text=col)
tv.grid(row=0, sticky=Tkinter.N + Tkinter.S + Tkinter.E + Tkinter.W)
Tkinter.Grid.columnconfigure(self.__main_window, 0, weight=1)
Tkinter.Grid.rowconfigure(self.__main_window, 0, weight=1)
Tkinter.Grid.columnconfigure(self.__main_window, 1, minsize=100)
self.__add_buttons()
# To be able to use this UI from another app and as autonomous app
# Also, I just hate an idea of creating storage in UI
if datasource is None:
self.__datasource = DataStorage()
else:
self.__datasource = datasource
def test_load_stored_data(saved_storage):
loaded_storage = DataStorage(saved_storage.filename)
assert len(loaded_storage.items) == len(saved_storage.items)
for i in range(len(loaded_storage.items)):
assert saved_storage.items[i] == loaded_storage.items[i]
def test_add_one_item():
ds = DataStorage()
ds.create_item("John", "Smith", "+1111111111", "05.11.2016")
items = ds.items
assert len(items) == 1
assert items[0].first_name == "John" and items[0].last_name == "Smith"
def saved_storage():
ds = DataStorage()
ds.create_item("John", "Smith", "+1111111111", "05.11.1955")
ds.create_item("John", "Doe", "+122222222", "14.11.1970")
ds.create_item("Maria", "Hanasaku", "+1333333333", "11.11.1911")
ds.create_item("David", "Iroha", "+1444444444", "08.11.1855")
ds.save("tmp.db")
yield ds
os.remove("tmp.db")
def __menu_new_pressed(self):
self.__datasource = DataStorage()
self.__update_table()
class MainWindow(object):
def __init__(self, datasource=None):
self.__main_window = Tk()
self.__main_window.minsize(800, 600)
self.__main_window.iconbitmap("resources/quill.ico")
self.__main_window.wm_title("Notebook Demo")
self.__add_menu()
cols = ("Last Name", "First Name", "Phone", "Birthday")
tv = Treeview(self.__main_window, columns=cols, show="headings")
self.__table = tv
for col in cols:
tv.heading(col, text=col)
tv.grid(row=0, sticky=Tkinter.N + Tkinter.S + Tkinter.E + Tkinter.W)
Tkinter.Grid.columnconfigure(self.__main_window, 0, weight=1)
Tkinter.Grid.rowconfigure(self.__main_window, 0, weight=1)
Tkinter.Grid.columnconfigure(self.__main_window, 1, minsize=100)
self.__add_buttons()
# To be able to use this UI from another app and as autonomous app
# Also, I just hate an idea of creating storage in UI
if datasource is None:
self.__datasource = DataStorage()
else:
self.__datasource = datasource
def __add_buttons(self):
frame = Tkinter.Frame(self.__main_window)
frame.grid(row=0, column=1, sticky=Tkinter.N + Tkinter.E + Tkinter.W)
button_add = Button(frame,
text="Add",
command=self.__add_button_pressed)
button_add.pack(fill=Tkinter.X)
button_edit = Button(frame,
text="Edit",
command=self.__edit_button_pressed)
button_edit.pack(fill=Tkinter.X)
button_remove = Button(frame,
text="Remove",
command=self.__remove_button_pressed)
button_remove.pack(fill=Tkinter.X)
def run(self):
self.__main_window.mainloop()
def __add_menu(self):
menubar = Menu(self.__main_window)
file_menu = Menu(menubar, tearoff=0)
file_menu.add_command(label="New", command=self.__menu_new_pressed)
file_menu.add_command(label="Save", command=self.__menu_save_pressed)
file_menu.add_command(label="Save As..",
command=self.__menu_save_as_pressed)
file_menu.add_command(label="Load", command=self.__menu_load_pressed)
file_menu.add_command(label="Exit")
menubar.add_cascade(label="File", menu=file_menu)
self.__main_window.config(menu=menubar)
def __update_table(self):
self.__table.delete(*self.__table.get_children())
items = self.__datasource.items
for item_id in range(len(items)):
item = items[item_id]
self.__table.insert("",
"end",
values=(item.last_name, item.first_name,
item.phone_number, item.birthday,
item_id))
def __add_button_pressed(self):
edit_dialog = EditDialog(self.__main_window,
datasource=self.__datasource)
edit_dialog.show()
if edit_dialog.ok:
self.__update_table()
def __edit_button_pressed(self):
selection_id = self.__table.selection()[0]
cur_item = self.__table.item(selection_id)
item_id = cur_item["values"][4]
item = self.__datasource.items[item_id]
edit_dialog = EditDialog(self.__main_window, item=item)
edit_dialog.show()
if edit_dialog.ok:
self.__update_table()
def __remove_button_pressed(self):
selection_id = self.__table.selection()[0]
cur_item = self.__table.item(selection_id)
item_id = cur_item["values"][4]
self.__datasource.delete_item(item_id)
self.__update_table()
def __menu_new_pressed(self):
self.__datasource = DataStorage()
self.__update_table()
def __menu_save_pressed(self):
if self.__datasource.filename is not None:
self.__datasource.save()
else:
self.__menu_save_as_pressed()
def __menu_save_as_pressed(self):
file_opts = {
"defaultextension": ".not",
"filetypes": [('all files', '.*'), ('data files', '.not')],
"parent": self.__main_window
}
filename = tkFileDialog.asksaveasfilename(**file_opts)
self.__datasource.save(filename=filename)
def __menu_load_pressed(self):
file_opts = {
"defaultextension": ".not",
"filetypes": [('all files', '.*'), ('data files', '.not')],
"parent": self.__main_window
}
filename = tkFileDialog.askopenfilename(**file_opts)
self.__datasource = DataStorage(filename=filename)
self.__update_table()
birthdays = self.__datasource.todays_birthdays
if len(birthdays) != 0:
names = [
"%s %s" % (item.first_name, item.last_name)
for item in birthdays
]
message = "Today is birthday of %s" % ", ".join(names)
tkMessageBox.showinfo("Birthday notification", message)