73
src/window/app.py
Normal file
73
src/window/app.py
Normal file
@@ -0,0 +1,73 @@
|
||||
from tkinter import *
|
||||
from tkinter import filedialog
|
||||
from pathlib import Path
|
||||
from src.util import Util
|
||||
from functools import partial
|
||||
from src.window.image import ImageWindow
|
||||
|
||||
|
||||
class App:
|
||||
def __init__(self, root):
|
||||
super().__init__()
|
||||
self.image_window = None
|
||||
self.root = root
|
||||
self.selected_folder = ""
|
||||
self.found_images = []
|
||||
self.timer = 0
|
||||
self.title = Label(root, text="Drawing Training")
|
||||
self.title.pack()
|
||||
|
||||
self.folder_selector = Button(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.pack()
|
||||
|
||||
self.images_len = Label(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
|
||||
|
||||
self.launch_button = Button(root, text="Let's draw!", command=self.lets_draw, state="disabled")
|
||||
self.launch_button.pack(side="bottom")
|
||||
|
||||
def lets_draw(self):
|
||||
self.image_window = ImageWindow(self)
|
||||
self.image_window.lets_draw(self.found_images.copy(), self.timer)
|
||||
|
||||
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.check_lets_draw()
|
||||
|
||||
def check_lets_draw(self):
|
||||
if self.selected_folder != "" and len(self.found_images) > 0 and self.timer != 0:
|
||||
self.launch_button.config(state="normal")
|
||||
else:
|
||||
self.launch_button.config(state="disabled")
|
||||
|
||||
def set_timer_in_seconds(self, user_data):
|
||||
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")
|
||||
self.check_lets_draw()
|
50
src/window/image.py
Normal file
50
src/window/image.py
Normal file
@@ -0,0 +1,50 @@
|
||||
import random
|
||||
from tkinter import *
|
||||
from pygame import mixer
|
||||
from src.element.image import ImagePlaceholder
|
||||
from src.element.toolbar import Toolbar
|
||||
|
||||
|
||||
class ImageWindow:
|
||||
def __init__(self, app):
|
||||
self.current_original_image = None
|
||||
self.app = app
|
||||
|
||||
self.window = Toplevel(app.root)
|
||||
self.window.title("Image")
|
||||
self.window.geometry("1280x1024")
|
||||
self.option = {
|
||||
"bw": False,
|
||||
"mirror": False,
|
||||
"always_on_top": False,
|
||||
}
|
||||
|
||||
mixer.init()
|
||||
self.countdown_sound = mixer.Sound('assets/sounds/countdown.mp3')
|
||||
|
||||
self.images = []
|
||||
self.current_image = None
|
||||
self.image_label = None
|
||||
toolbar = Frame(self.window, bd=1, relief=RAISED)
|
||||
toolbar.pack(side=BOTTOM, fill=X)
|
||||
self.timer = 0
|
||||
|
||||
self.toolbar = Toolbar(self, self.timer)
|
||||
self.image = ImagePlaceholder(self, self.images)
|
||||
|
||||
def next_image(self):
|
||||
if self.toolbar.timer_check is not None:
|
||||
self.window.after_cancel(self.toolbar.timer_check)
|
||||
self.image.display_new_image()
|
||||
self.toolbar.display_new_timer()
|
||||
|
||||
def lets_draw(self, images, timer):
|
||||
random.shuffle(images)
|
||||
# self.images = images
|
||||
self.image.images = images
|
||||
self.toolbar.timer = timer
|
||||
# self.timer = timer
|
||||
self.next_image()
|
||||
|
||||
def play_countdown(self):
|
||||
self.countdown_sound.play()
|
Reference in New Issue
Block a user