Shikiryu/DrawingTraining
1
0
Fork 0
Code Issues 2 Pull Requests 1 Packages Projects 1 Releases 2 Wiki Activity

enhance/ui (#28)

Browse Source 
Fix #24

Reviewed-on: #28
This commit is contained in:
Shikiryu
2023-06-07 00:59:45 +02:00
parent a149ab71bb
commit 693c596224
10 changed files with 158 additions and 99 deletions
Download Patch File Download Diff File Expand all files Collapse all files

86
src/window/app.py
View File

@@ -1,10 +1,9 @@
from pathlib import Path
from tkinter import *
from tkinter import filedialog
from pathlib import Path
import customtkinter
from src.entity.config import Config
from src.util import Util
from functools import partial
from src.window.image import ImageWindow
from src.window.session import SessionWindow
@@ -22,41 +21,49 @@ class App:
self.selected_folder = ""
self.found_images = []
self.timer = 0
self.title = Label(root, text="Drawing Training")
self.title = customtkinter.CTkLabel(root, text="Drawing Training")
self.title.pack()
self.folder_selector = Button(root, text="Select a folder", command=self.select_folder)
self.folder_selector = customtkinter.CTkButton(root, text="Select a folder", command=self.select_folder)
self.folder_selector.pack()
self.folder_name = Label(root, text="Folder : " + self.selected_folder)
self.folder_name = customtkinter.CTkLabel(root, text="Folder : " + self.selected_folder)
self.folder_name.pack()
self.images_len = Label(root, text="Found : " + str(len(self.found_images)))
self.images_len = customtkinter.CTkLabel(root, text="Found : " + str(len(self.found_images)))
self.images_len.pack()
self.button_frame = Frame(root)
self.button_frame.pack(fill=X)
timers = [30, 45, 60, 120, 300, 600]
self.buttons = []
i = 0
for i in timers:
t = i
new_button = Button(self.button_frame, text=Util.format_seconds(t),
command=partial(self.set_timer_in_seconds, t))
self.buttons.append(new_button)
self.button_frame.columnconfigure(i, weight=1)
new_button.grid(row=0, column=i, sticky=W + E)
i += 1
timers = [
Util.format_seconds(30),
Util.format_seconds(45),
Util.format_seconds(60),
Util.format_seconds(120),
Util.format_seconds(300),
Util.format_seconds(600),
"Custom"
]
self.buttons = customtkinter.CTkSegmentedButton(
root, values=timers,
command=self.set_timer_in_seconds
)
self.buttons.pack(fill=X)
new_button = Button(self.button_frame, text="Custom", command=self.custom_session)
self.buttons.append(new_button)
self.button_frame.columnconfigure(i, weight=1)
new_button.grid(row=0, column=i, sticky=W + E)
self.launch_button = Button(root, text="Let's draw!", command=self.lets_draw, state="disabled")
self.launch_button = customtkinter.CTkButton(root, text="Let's draw!", command=self.lets_draw, state="disabled")
self.launch_button.pack(side="bottom")
self.read_config()
def read_config(self):
try:
self.selected_folder = Config.get_required_config_var("default_folder")
self.folder_name.configure(text="Folder : " + self.selected_folder)
self.find_images_in_folder()
except Exception:
pass
def lets_draw(self):
self.image_window = ImageWindow(self)
self.image_window.lets_draw(self.found_images.copy(), self.timer)
@@ -67,32 +74,33 @@ class App:
def select_folder(self):
self.selected_folder = filedialog.askdirectory()
self.found_images = list(
p.resolve() for p in Path(self.selected_folder).glob("**/*") if p.suffix in {".jpg", ".gif", ".png"})
self.folder_name.config(text="Folder : " + self.selected_folder)
self.images_len.config(text="Found : " + str(len(self.found_images)))
self.folder_name.configure(text="Folder : " + self.selected_folder)
Config.set_config_var("default_folder", self.selected_folder).save()
self.find_images_in_folder()
self.check_lets_draw()
def find_images_in_folder(self):
if Path(self.selected_folder).exists():
self.found_images = list(
p.resolve() for p in Path(self.selected_folder).glob("**/*") if p.suffix in {".jpg", ".gif", ".png"}
)
self.images_len.configure(text="Found : " + str(len(self.found_images)))
def check_lets_draw(self):
if self.selected_folder != "" and len(self.found_images) > 0 and (self.timer != 0 or self.custom):
self.launch_button.config(state="normal")
self.launch_button.configure(state="normal")
else:
self.launch_button.config(state="disabled")
self.launch_button.configure(state="disabled")
def set_timer_in_seconds(self, user_data):
self.custom = False
self.timer = user_data
for button in self.buttons:
if button['text'] == Util.format_seconds(self.timer):
button.config(bg="blue")
else:
button.config(bg="gray85")
if user_data == "Custom":
return self.custom_session()
self.timer = Util.format_time_to_seconds(user_data)
self.check_lets_draw()
def set_custom(self, list_in_session):
self.custom = True
self.list_in_session = list_in_session
self.check_lets_draw()
for button in self.buttons:
button.config(bg="gray85")
self.buttons[-1].config(bg="blue")

4
src/window/image.py
View File

@@ -1,5 +1,7 @@
import random
from tkinter import *
from customtkinter import CTkToplevel
from pygame import mixer
from src.element.image import ImagePlaceholder
from src.element.toolbar import Toolbar
@@ -10,7 +12,7 @@ class ImageWindow:
self.current_original_image = None
self.app = app
self.window = Toplevel(app.root)
self.window = CTkToplevel(app.root)
self.window.title("Image")
self.window.geometry("1280x1024")
self.window.protocol("WM_DELETE_WINDOW", self.on_closing)

73
src/window/session.py
View File

@@ -1,6 +1,5 @@
from functools import partial
from tkinter import *
from customtkinter import *
from src.entity.config import Config
from src.entity.session.drawing import DrawingElement
from src.util import Util
@@ -18,16 +17,16 @@ class SessionWindow:
self.add_button = None
self.app = app
self.window = Toplevel(app.root)
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 = Frame(self.window, width=300, height=600)
self.right_column = Frame(self.window, width=300, height=600)
self.left_column = CTkFrame(self.window, width=300, height=60)
self.right_column = CTkFrame(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.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 = []
@@ -40,30 +39,32 @@ class SessionWindow:
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 = 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())
)
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)
buttons = CTkSegmentedButton(
self.left_column, values=["+", "-", "^", "v", "save"],
command=self.button_command
)
buttons.grid()
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 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()
else:
print("unknown command")
def reset_element(self):
for widget in self.right_column.winfo_children():
@@ -82,34 +83,34 @@ class SessionWindow:
else:
element = DrawingElement()
number_of_drawings_label = Label(self.right_column, text="Number of drawings")
number_of_drawings_label = CTkLabel(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 = CTkEntry(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 = CTkLabel(self.right_column, text="Time per drawing in seconds")
timer_label.pack()
self.timer_input = Entry(self.right_column)
self.timer_input = CTkEntry(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 = CTkLabel(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 = CTkLabel(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 = 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.config(
self.total_row_time_value_label.configure(
text=Util.format_seconds(int(self.timer_input.get()) * int(self.number_of_drawings_input.get()))
)