feature/session #27
1
.gitignore
vendored
1
.gitignore
vendored
@ -6,3 +6,4 @@ theme.py
|
||||
!/dist/drawingtraining
|
||||
/*.spec
|
||||
/build/
|
||||
/config.json
|
||||
|
49
src/entity/config.py
Normal file
49
src/entity/config.py
Normal file
@ -0,0 +1,49 @@
|
||||
import json
|
||||
import os
|
||||
from pathlib import Path
|
||||
|
||||
from configparser import *
|
||||
import typing
|
||||
|
||||
|
||||
class Config(object):
|
||||
|
||||
_CONFIG_FILE: typing.Optional[str] = None
|
||||
_CONFIG: typing.Optional[dict] = None
|
||||
|
||||
def __init__(self, config_file=None):
|
||||
if config_file is None:
|
||||
config_file = "config.json"
|
||||
|
||||
# Check that specified config file exists
|
||||
if not os.path.exists(config_file):
|
||||
Path(config_file).touch()
|
||||
with open(config_file, 'w') as configfile:
|
||||
json.dump({"session": []}, configfile)
|
||||
|
||||
Config._CONFIG_FILE = config_file
|
||||
|
||||
with open(config_file, 'r') as f:
|
||||
Config._CONFIG = json.load(f)
|
||||
|
||||
@staticmethod
|
||||
def save():
|
||||
with open(Config._CONFIG_FILE, 'w') as configfile:
|
||||
json.dump(Config._CONFIG, configfile)
|
||||
|
||||
@staticmethod
|
||||
def get_config_file() -> str:
|
||||
return Config._CONFIG_FILE
|
||||
|
||||
@staticmethod
|
||||
def get_required_env_var(env_var: str) -> str:
|
||||
if env_var not in os.environ:
|
||||
raise Exception(f"Please set the {env_var} environment variable")
|
||||
return os.environ[env_var]
|
||||
|
||||
@staticmethod
|
||||
def get_required_config_var(config_var: str) -> str:
|
||||
assert Config._CONFIG
|
||||
if config_var not in Config._CONFIG:
|
||||
raise Exception(f"Please set the {config_var} variable in the config file {Config._CONFIG_FILE}")
|
||||
return Config._CONFIG[config_var]
|
@ -1,6 +1,8 @@
|
||||
from tkinter import *
|
||||
from tkinter import filedialog
|
||||
from pathlib import Path
|
||||
|
||||
from src.entity.config import Config
|
||||
from src.util import Util
|
||||
from functools import partial
|
||||
from src.window.image import ImageWindow
|
||||
@ -10,6 +12,7 @@ from src.window.session import SessionWindow
|
||||
class App:
|
||||
def __init__(self, root):
|
||||
super().__init__()
|
||||
self.config = Config()
|
||||
self.session_window = None
|
||||
self.image_window = None
|
||||
self.root = root
|
||||
|
@ -1,11 +1,16 @@
|
||||
from functools import partial
|
||||
from tkinter import *
|
||||
|
||||
from src.entity.config import Config
|
||||
from src.entity.session.drawing import DrawingElement
|
||||
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
|
||||
@ -32,17 +37,30 @@ 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)
|
||||
text=session.number_of_drawings + " drawings of " + session.timer + "s each",
|
||||
command=partial(self.edit_element, i),
|
||||
bg=bg
|
||||
)
|
||||
new_session.grid(row=i, column=0)
|
||||
|
||||
self.add_button = Button(self.left_column, text="+", command=self.add_element)
|
||||
self.add_button.grid()
|
||||
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)
|
||||
|
||||
self.delete_button = Button(self.left_column, text="-", command=self.remove_element)
|
||||
self.delete_button.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)
|
||||
|
||||
self.save_session_button = Button(button_frame, text="v", command=self.save_session)
|
||||
self.save_session_button.pack(side=LEFT, padx=0, pady=0)
|
||||
button_frame.grid()
|
||||
|
||||
def reset_element(self):
|
||||
for widget in self.right_column.winfo_children():
|
||||
@ -53,9 +71,11 @@ class SessionWindow:
|
||||
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()
|
||||
|
||||
@ -64,24 +84,32 @@ class SessionWindow:
|
||||
|
||||
self.number_of_drawings_input = Entry(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.pack()
|
||||
|
||||
self.timer_input = Entry(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()
|
||||
|
||||
row_time = IntVar()
|
||||
total_row_time_label = Label(self.right_column, text="Total row time")
|
||||
total_row_time_label.pack()
|
||||
total_row_time_value_label = Label(self.right_column, textvariable=row_time)
|
||||
total_row_time_value_label.pack()
|
||||
self.total_row_time_value_label = Label(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.pack()
|
||||
|
||||
def update_local_timer(self, event=None):
|
||||
self.total_row_time_value_label.config(
|
||||
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
|
||||
@ -91,8 +119,22 @@ class SessionWindow:
|
||||
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 = DrawingElement() if self.selected_element_index < 0 \
|
||||
else self.list_in_session[self.selected_element_index]
|
||||
@ -111,3 +153,9 @@ class SessionWindow:
|
||||
def save_on_closing(self):
|
||||
# TODO save
|
||||
self.window.destroy()
|
||||
|
||||
def save_session(self):
|
||||
for element in self.list_in_session:
|
||||
Config._CONFIG['session'].append(element)
|
||||
Config.save()
|
||||
self.window.destroy()
|
||||
|
Loading…
Reference in New Issue
Block a user