from functools import partial from customtkinter import * from src.entity.config import Config from src.entity.session.drawing import DrawingElement from src.entity.session.breakEntity import BreakElement 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 = CTkToplevel(app.root) self.window.title("Custom session") self.window.geometry("600x600") self.window.protocol("WM_DELETE_WINDOW", self.save_on_closing) self.left_column = CTkFrame(self.window, width=300, height=60) self.right_column = CTkFrame(self.window, width=300, height=600) self.left_column.pack(padx=10, pady=10, side=LEFT, fill=BOTH, expand=True) self.right_column.pack(padx=10, pady=10, side=LEFT, fill=BOTH, expand=True) self.list_in_session = [] def open(self): list = Config._CONFIG['session'] if list: self.list_in_session = [DrawingElement(e) if e["type"] == "Drawing" else BreakElement(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): if isinstance(session, DrawingElement): new_session = CTkButton(self.left_column, text=session.number_of_drawings + " drawings of " + Util.format_seconds(int(session.timer)) + " each", command=partial(self.edit_element, i), fg_color=( Util.get_default_button_color() if i == self.selected_element_index else Util.get_default_active_button_color()) ) elif isinstance(session, BreakElement): new_session = CTkButton(self.left_column, text="Break of " + Util.format_seconds(int(session.timer)) + " each", command=partial(self.edit_element, i), fg_color=( Util.get_default_button_color() if i == self.selected_element_index else Util.get_default_active_button_color()) ) else: raise Exception("unknown type.") new_session.grid(row=i, column=0) buttons = CTkSegmentedButton( self.left_column, values=["+", "-", "☕", "^", "v", "save"], command=self.button_command ) buttons.grid() def button_command(self, command): if command == "+": self.add_element() elif command == "-": self.remove_element() elif command == "^": self.up_element() elif command == "v": self.down_element() elif command == "save": self.save_session() elif command == "☕": self.add_break() else: print("unknown command") 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() if isinstance(element, DrawingElement): self.print_drawing_form(element) elif isinstance(element, BreakElement): self.print_break_form(self.selected_element_index) def print_drawing_form(self, element): number_of_drawings_label = CTkLabel(self.right_column, text="Number of drawings") number_of_drawings_label.pack() self.number_of_drawings_input = CTkEntry(self.right_column) self.number_of_drawings_input.insert(INSERT, element.number_of_drawings) self.number_of_drawings_input.bind("", lambda e: self.update_local_timer(e)) self.number_of_drawings_input.pack() self.number_of_drawings_input.focus_set() timer_label = CTkLabel(self.right_column, text="Time per drawing in seconds") timer_label.pack() self.timer_input = CTkEntry(self.right_column) self.timer_input.insert(INSERT, element.timer) self.timer_input.bind("", lambda e: self.update_local_timer(e)) self.timer_input.pack() total_row_time_label = CTkLabel(self.right_column, text="Total row time") total_row_time_label.pack() self.total_row_time_value_label = CTkLabel(self.right_column, text="") self.total_row_time_value_label.pack() self.update_local_timer() save_button = CTkButton(self.right_column, text="Save", command=self.save) save_button.pack() def update_local_timer(self, event=None): self.total_row_time_value_label.configure( 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_type="Drawing"): if element_type == "Drawing": 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() elif element_type == "Break": element = BreakElement() if self.selected_element_index < 0 \ else self.list_in_session[self.selected_element_index] element.timer = self.timer_input.get() else: raise Exception("unknown type.") 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() def add_break(self): self.selected_element_index = -1 self.update_session_list() self.print_break_form() def print_break_form(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 = BreakElement() timer_label = CTkLabel(self.right_column, text="Time for this break in seconds") timer_label.pack() self.timer_input = CTkEntry(self.right_column) self.timer_input.insert(INSERT, element.timer) self.timer_input.pack() save_button = CTkButton(self.right_column, text="Save", command=partial(self.save, "Break")) save_button.pack()