79 lines
2.2 KiB
Python
79 lines
2.2 KiB
Python
|
from tkinter import *
|
||
|
|
from tkinter import filedialog
|
||
|
|
from pathlib import Path
|
||
|
|
from functools import partial
|
||
|
|
|
||
|
|
root = Tk()
|
||
|
|
root.title("Drawing Training!")
|
||
|
|
root.geometry("300x600")
|
||
|
|
|
||
|
|
selected_folder = ""
|
||
|
|
found_images = []
|
||
|
|
timer = 0
|
||
|
|
|
||
|
|
|
||
|
|
def set_timer_in_seconds(user_data):
|
||
|
|
global timer
|
||
|
|
timer = user_data
|
||
|
|
print(timer)
|
||
|
|
for button in buttons:
|
||
|
|
if button['text'] == str(timer) + "s":
|
||
|
|
button.config(bg="blue")
|
||
|
|
else:
|
||
|
|
button.config(bg="grey")
|
||
|
|
|
||
|
|
|
||
|
|
def select_folder():
|
||
|
|
global selected_folder
|
||
|
|
global found_images
|
||
|
|
global folder_name
|
||
|
|
global images_len
|
||
|
|
selected_folder = filedialog.askdirectory()
|
||
|
|
found_images = list(p.resolve() for p in Path(selected_folder).glob("**/*") if p.suffix in {".jpg", ".gif", ".png"})
|
||
|
|
folder_name.config(text="Folder : " + selected_folder)
|
||
|
|
images_len.config(text="Found : " + str(len(found_images)))
|
||
|
|
|
||
|
|
|
||
|
|
title = Label(root, text="Drawing Training")
|
||
|
|
title.pack()
|
||
|
|
|
||
|
|
folder_selector = Button(root, text="Select a folder", command=select_folder)
|
||
|
|
folder_selector.pack()
|
||
|
|
|
||
|
|
folder_name = Label(root, text="Folder : " + selected_folder)
|
||
|
|
folder_name.pack()
|
||
|
|
|
||
|
|
images_len = Label(root, text="Found : " + str(len(found_images)))
|
||
|
|
images_len.pack()
|
||
|
|
|
||
|
|
timers = [30, 45, 60, 120, 300, 600]
|
||
|
|
buttons = []
|
||
|
|
for i in timers:
|
||
|
|
t = i
|
||
|
|
new_button = Button(root, text=str(t) + "s", command=partial(set_timer_in_seconds, t))
|
||
|
|
new_button.pack(side="left")
|
||
|
|
buttons.append(new_button)
|
||
|
|
|
||
|
|
# timer_30 = Button(root, text="30s", command=lambda: set_timer_in_seconds(30))
|
||
|
|
# timer_30.pack(side="left")
|
||
|
|
#
|
||
|
|
# timer_45 = Button(root, text="45s", command=lambda: set_timer_in_seconds(45))
|
||
|
|
# timer_45.pack(side="left")
|
||
|
|
#
|
||
|
|
# timer_60 = Button(root, text="1m", command=lambda: set_timer_in_seconds(1 * 60))
|
||
|
|
# timer_60.pack(side="left")
|
||
|
|
#
|
||
|
|
# timer_120 = Button(root, text="2m", command=lambda: set_timer_in_seconds(2 * 60))
|
||
|
|
# timer_120.pack(side="left")
|
||
|
|
#
|
||
|
|
# timer_300 = Button(root, text="5m", command=lambda: set_timer_in_seconds(5 * 60))
|
||
|
|
# timer_300.pack(side="left")
|
||
|
|
#
|
||
|
|
# timer_600 = Button(root, text="10m", command=lambda: set_timer_in_seconds(10 * 60))
|
||
|
|
# timer_600.pack(side="left")
|
||
|
|
|
||
|
|
launch_button = Button(root, text="Let's draw!")
|
||
|
|
launch_button.pack(side="bottom")
|
||
|
|
|
||
|
|
root.mainloop()
|