167 lines
6.2 KiB
Python
167 lines
6.2 KiB
Python
from functools import partial
|
|
from tkinter import *
|
|
|
|
from src.entity.config import Config
|
|
from src.entity.session.drawing import DrawingElement
|
|
from src.util import Util
|
|
|
|
|
|
class SessionWindow:
|
|
def __init__(self, app):
|
|
self.total_row_time_value_label = None
|
|
self.down_button = None
|
|
self.up_button = None
|
|
self.timer_input = None
|
|
self.number_of_drawings_input = None
|
|
self.selected_element_index = -1
|
|
self.delete_button = None
|
|
self.add_button = None
|
|
self.app = app
|
|
|
|
self.window = Toplevel(app.root)
|
|
self.window.title("Custom session")
|
|
self.window.geometry("600x600")
|
|
self.window.protocol("WM_DELETE_WINDOW", self.save_on_closing)
|
|
|
|
self.left_column = Frame(self.window, width=300, height=600)
|
|
self.right_column = Frame(self.window, width=300, height=600)
|
|
|
|
self.left_column.grid(row=0, column=0, sticky="ns")
|
|
self.right_column.grid(row=0, column=1, sticky="ns")
|
|
|
|
self.list_in_session = []
|
|
|
|
def open(self):
|
|
list = Config._CONFIG['session']
|
|
if list:
|
|
self.list_in_session = [DrawingElement(e) for e in list]
|
|
self.update_session_list()
|
|
|
|
def update_session_list(self):
|
|
self.reset_list()
|
|
for i, session in enumerate(self.list_in_session):
|
|
bg = "blue" if i == self.selected_element_index else "grey85"
|
|
new_session = Button(self.left_column,
|
|
text=session.number_of_drawings + " drawings of " + session.timer + "s each",
|
|
command=partial(self.edit_element, i),
|
|
bg=bg
|
|
)
|
|
new_session.grid(row=i, column=0)
|
|
|
|
button_frame = Frame(self.left_column)
|
|
self.add_button = Button(button_frame, text="+", command=self.add_element)
|
|
self.add_button.pack(side=LEFT, padx=0, pady=0)
|
|
|
|
self.delete_button = Button(button_frame, text="-", command=self.remove_element)
|
|
self.delete_button.pack(side=LEFT, padx=0, pady=0)
|
|
|
|
self.up_button = Button(button_frame, text="^", command=self.up_element)
|
|
self.up_button.pack(side=LEFT, padx=0, pady=0)
|
|
|
|
self.down_button = Button(button_frame, text="v", command=self.down_element)
|
|
self.down_button.pack(side=LEFT, padx=0, pady=0)
|
|
|
|
save_session_button = Button(button_frame, text="v", command=self.save_session)
|
|
save_session_button.pack(side=LEFT, padx=0, pady=0)
|
|
button_frame.grid()
|
|
|
|
def reset_element(self):
|
|
for widget in self.right_column.winfo_children():
|
|
widget.destroy()
|
|
|
|
def reset_list(self):
|
|
for widget in self.left_column.winfo_children():
|
|
widget.destroy()
|
|
|
|
def update_element(self, index=None):
|
|
self.reset_element()
|
|
if index is not None:
|
|
self.selected_element_index = index
|
|
element = self.list_in_session[self.selected_element_index]
|
|
self.update_session_list()
|
|
else:
|
|
element = DrawingElement()
|
|
|
|
number_of_drawings_label = Label(self.right_column, text="Number of drawings")
|
|
number_of_drawings_label.pack()
|
|
|
|
self.number_of_drawings_input = Entry(self.right_column)
|
|
self.number_of_drawings_input.insert(INSERT, element.number_of_drawings)
|
|
self.number_of_drawings_input.bind("<KeyRelease>", lambda e: self.update_local_timer(e))
|
|
self.number_of_drawings_input.pack()
|
|
self.number_of_drawings_input.focus_set()
|
|
|
|
timer_label = Label(self.right_column, text="Time per drawing in seconds")
|
|
timer_label.pack()
|
|
|
|
self.timer_input = Entry(self.right_column)
|
|
self.timer_input.insert(INSERT, element.timer)
|
|
self.timer_input.bind("<KeyRelease>", lambda e: self.update_local_timer(e))
|
|
self.timer_input.pack()
|
|
|
|
total_row_time_label = Label(self.right_column, text="Total row time")
|
|
total_row_time_label.pack()
|
|
self.total_row_time_value_label = Label(self.right_column, text="")
|
|
self.total_row_time_value_label.pack()
|
|
self.update_local_timer()
|
|
|
|
save_button = Button(self.right_column, text="Save", command=self.save)
|
|
save_button.pack()
|
|
|
|
def update_local_timer(self, event=None):
|
|
self.total_row_time_value_label.config(
|
|
text=Util.format_seconds(int(self.timer_input.get()) * int(self.number_of_drawings_input.get()))
|
|
)
|
|
|
|
def remove_element(self):
|
|
self.list_in_session.pop(self.selected_element_index)
|
|
self.selected_element_index = -1
|
|
self.update_session_list()
|
|
|
|
def edit_element(self, i):
|
|
self.update_element(i)
|
|
|
|
def add_element(self):
|
|
self.selected_element_index = -1
|
|
self.update_session_list()
|
|
self.update_element()
|
|
|
|
def up_element(self):
|
|
self.list_in_session.insert(self.selected_element_index - 1,
|
|
self.list_in_session.pop(self.selected_element_index))
|
|
self.selected_element_index -= 1
|
|
self.update_session_list()
|
|
|
|
def down_element(self):
|
|
self.list_in_session.insert(self.selected_element_index + 1,
|
|
self.list_in_session.pop(self.selected_element_index))
|
|
self.selected_element_index += 1
|
|
self.update_session_list()
|
|
|
|
def save(self):
|
|
element = DrawingElement() if self.selected_element_index < 0 \
|
|
else self.list_in_session[self.selected_element_index]
|
|
element.timer = self.timer_input.get()
|
|
element.number_of_drawings = self.number_of_drawings_input.get()
|
|
|
|
if self.selected_element_index < 0:
|
|
self.list_in_session.append(element)
|
|
else:
|
|
self.list_in_session[self.selected_element_index] = element
|
|
|
|
self.update_session_list()
|
|
self.selected_element_index = -1
|
|
self.reset_element()
|
|
|
|
def save_on_closing(self):
|
|
# TODO save
|
|
self.window.destroy()
|
|
|
|
def save_session(self):
|
|
Config._CONFIG['session'] = []
|
|
for element in self.list_in_session:
|
|
Config._CONFIG['session'].append(element.__dict__)
|
|
Config.save()
|
|
self.app.set_custom(self.list_in_session)
|
|
self.window.destroy()
|