🚧 Add breaks

For #11

src/window/session.py

@@ -2,6 +2,7 @@ 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
 
 
@@ -33,21 +34,33 @@ class SessionWindow:
     def open(self):
         list = Config._CONFIG['session']
         if list:
-            self.list_in_session = [DrawingElement(e) for e in 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):
-            new_session = CTkButton(self.left_column,
-                                    text=session.number_of_drawings + " drawings of " + session.timer + "s 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())
-                                    )
+            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"],
+            self.left_column, values=["+", "-", "☕", "^", "v", "save"],
             command=self.button_command
         )
         buttons.grid()
@@ -63,6 +76,8 @@ class SessionWindow:
             self.down_element()
         elif command == "save":
             self.save_session()
+        elif command == "☕":
+            self.add_break()
         else:
             print("unknown command")
 
@@ -83,6 +98,12 @@ class SessionWindow:
         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()
 
@@ -139,11 +160,18 @@ class SessionWindow:
         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()
+    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)
@@ -165,3 +193,27 @@ class SessionWindow:
         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()