class PortsTab(Tab,
HasCommonSetup,
SubscribesToEvents,
Resetable):
"""Used to create, edit and remove components.
Attributes:
__selected_port: Currently selected port in the ports listview.
__selected_component: Currently selected component in the components taxonomy view.
"""
def __init__(self, parent_notebook):
self.__state = State()
self.__selected_port: Optional[Port] = None
self.__selected_component: Optional[Component] = None
Tab.__init__(self, parent_notebook, TAB_NAME)
HasCommonSetup.__init__(self)
SubscribesToEvents.__init__(self)
# HasCommonSetup
def _create_widgets(self) -> None:
self.__taxonomy_tree = ScrollbarListbox(self,
on_select_callback=self.__on_select_tree_item,
heading=TREEVIEW_HEADING,
extract_id=lambda x: x.id_,
extract_text=lambda x: x.name,
extract_ancestor=lambda x: '' if x.parent_id is None else x.parent_id,
extract_values=self.__extract_values,
columns=[Column('Amount')],
values=self.__state.model.taxonomy)
self.__left_frame = ttk.Frame(self)
# Ports combobox
self.__port_combobox_var = tk.StringVar(value=SELECT_PORT)
self.__port_combobox_var.trace('w', self.__on_combobox_changed)
# state='readonly' means you cannot write freely in the combobox
self.__port_combobox = ttk.Combobox(self.__left_frame, state='readonly',
textvariable=self.__port_combobox_var, font=FONT)
ports_names = self.__state.model.get_all_ports_names()
self.__port_combobox['values'] = sorted(ports_names)
# C(r)ud Buttons
self.__add_port_button = ttk.Button(self.__left_frame, text='Add', state=tk.NORMAL,
command=self.__on_add)
self.__rename_port_button = ttk.Button(self.__left_frame, text='Rename', state=tk.DISABLED,
command=self.__on_rename)
self.__remove_port_button = ttk.Button(self.__left_frame, text='Remove', state=tk.DISABLED,
command=self.__remove)
# Force connection
self.__force_connection_checkbox_var = tk.BooleanVar(value=False)
self.__force_connection_checkbox_var.trace('w', self.__on_force_connection_toggled)
self.__force_connection_checkbox_label = ttk.Label(self.__left_frame,
text='Force connection:')
self.__force_connection_checkbox = ttk.Checkbutton(self.__left_frame, state=tk.DISABLED,
variable=self.__force_connection_checkbox_var)
# Force connection checkbox
self.__compatible_with_edit_button = ttk.Button(self.__left_frame, text='Edit compatibility',
command=self.__on_edit_compatible_with, state=tk.DISABLED)
self.__compatible_with_listbox = ScrollbarListbox(self.__left_frame,
extract_text=lambda prt: prt.name,
extract_id=lambda prt: prt.id_,
columns=[Column('Compatible with', main=True, stretch=tk.YES)])
self.__cmp_label_var = tk.StringVar(value='')
self.__cmp_label = ttk.Label(self.__left_frame, textvariable=self.__cmp_label_var, style='Big.TLabel', anchor=tk.CENTER)
self.__amount_spinbox_label = ttk.Label(self.__left_frame, text='Has:')
self.__amount_spinbox_var = tk.IntVar(value='')
self.__amount_spinbox_var.trace('w', self.__on_amount_changed)
self.__amount_spinbox = ttk.Spinbox(self.__left_frame, from_=0, to=math.inf,
textvariable=self.__amount_spinbox_var)
self.__all_children_amount_spinbox_label = ttk.Label(self.__left_frame, text='Children have:')
self.__all_children_amount_spinbox_var = tk.IntVar(value='')
self.__all_children_amount_spinbox = ttk.Spinbox(self.__left_frame, from_=0, to=math.inf,
textvariable=self.__all_children_amount_spinbox_var)
self.__apply_to_all_children_button = ttk.Button(self.__left_frame, text='Apply to all children',
command=self.__apply_to_all_children)
def _setup_layout(self) -> None:
self.__taxonomy_tree.grid(row=0, column=1, sticky=tk.NSEW)
self.__left_frame.grid(row=0, column=0, sticky=tk.NSEW, pady=FRAME_PAD_Y, padx=FRAME_PAD_X)
self.__port_combobox.grid(row=0, column=0, columnspan=2, sticky=tk.NSEW, pady=CONTROL_PAD_Y)
self.__add_port_button.grid(row=1, column=0, columnspan=2, sticky=tk.NSEW, pady=CONTROL_PAD_Y)
self.__rename_port_button.grid(row=2, column=0, columnspan=2, sticky=tk.NSEW, pady=CONTROL_PAD_Y)
self.__remove_port_button.grid(row=3, column=0, columnspan=2, sticky=tk.NSEW, pady=CONTROL_PAD_Y)
self.__force_connection_checkbox_label.grid(row=4, column=0, sticky=tk.W, pady=CONTROL_PAD_Y)
self.__force_connection_checkbox.grid(row=4, column=1, sticky=tk.E, pady=CONTROL_PAD_Y)
self.__compatible_with_edit_button.grid(row=5, column=0, columnspan=2, sticky=tk.NSEW, pady=CONTROL_PAD_Y)
self.__compatible_with_listbox.grid(row=6, column=0, columnspan=2, sticky=tk.NSEW, pady=CONTROL_PAD_Y)
self.__cmp_label.grid(row=7, column=0, columnspan=2, pady=CONTROL_PAD_Y, sticky=tk.EW)
self.__amount_spinbox_label.grid(row=8, column=0, pady=CONTROL_PAD_Y, sticky=tk.W)
self.__amount_spinbox.grid(row=8, column=1, pady=CONTROL_PAD_Y, sticky=tk.NSEW)
self.__all_children_amount_spinbox_label.grid(row=8, column=0, pady=CONTROL_PAD_Y, sticky=tk.W)
self.__all_children_amount_spinbox.grid(row=8, column=1, pady=CONTROL_PAD_Y, sticky=tk.NSEW)
self.__apply_to_all_children_button.grid(row=9, column=0, columnspan=2, sticky=tk.NSEW, pady=CONTROL_PAD_Y)
self.columnconfigure(1, weight=1) # Give all the remaining space to taxonomy tree
self.rowconfigure(0, weight=1)
self.__left_frame.columnconfigure(1, weight=1)
# Hide widgets
self.__taxonomy_tree.grid_forget()
self.__cmp_label.grid_forget()
self.__amount_spinbox_label.grid_forget()
self.__amount_spinbox.grid_forget()
self.__all_children_amount_spinbox_label.grid_forget()
self.__all_children_amount_spinbox.grid_forget()
self.__apply_to_all_children_button.grid_forget()
# Taxonomy Treeview
def __on_select_tree_item(self, cmp_id: int) -> None:
"""Executed whenever a tree item is selected (by mouse click).
:param cmp_id: Id of the selected component.
"""
if self.__selected_port:
selected_cmp = self.__state.model.get_component(id_=cmp_id)
self.__selected_component = selected_cmp
self.__cmp_label.grid(row=7, column=0, columnspan=2, pady=CONTROL_PAD_Y, sticky=tk.EW) # Show cmp label
self.__cmp_label_var.set(trim_string(selected_cmp.name, length=30)) # Fill the label with component name
if selected_cmp.is_leaf:
self.__all_children_amount_spinbox_label.grid_forget() # Hide widgets (changing all children)
self.__all_children_amount_spinbox.grid_forget()
self.__apply_to_all_children_button.grid_forget()
self.__all_children_amount_spinbox_var.set(0)
amount = 0
if self.__selected_port.id_ in selected_cmp.ports:
amount = selected_cmp.ports[self.__selected_port.id_]
self.__amount_spinbox_var.set(amount)
self.__amount_spinbox_label.grid(row=8, column=0, pady=CONTROL_PAD_Y, sticky=tk.W) # Show widgets for leaves
self.__amount_spinbox.grid(row=8, column=1, pady=CONTROL_PAD_Y, sticky=tk.NSEW)
else:
self.__amount_spinbox_label.grid_forget() # Hide widgets for leaves
self.__amount_spinbox.grid_forget()
self.__all_children_amount_spinbox_label.grid(row=8, column=0, pady=CONTROL_PAD_Y, sticky=tk.W) # Show widgets (changing all children)
self.__all_children_amount_spinbox.grid(row=8, column=1, pady=CONTROL_PAD_Y, sticky=tk.NSEW)
self.__apply_to_all_children_button.grid(row=9, column=0, columnspan=2, sticky=tk.NSEW,
pady=CONTROL_PAD_Y)
def __extract_values(self, cmp: Component) -> Tuple[Any, ...]:
"""Extracts the data of the component to show in the taxonomy view.
:param cmp: Component from which to extract the data.
:return: Tuple containing data about component
(number of ports the of type __selected_port that the component has,).
"""
amount = ''
if self.__selected_port:
if self.__selected_port.id_ in cmp.ports:
amount = cmp.ports[self.__selected_port.id_]
return amount,
def __build_tree(self) -> None:
"""Fills the tree view with components from model."""
self.__taxonomy_tree.set_items(self.__state.model.taxonomy)
def __on_combobox_changed(self, *_) -> None:
"""Executed whenever the __port_combobox value changes."""
# Hide component-specific widgets
self.__cmp_label.grid_forget()
self.__amount_spinbox_label.grid_forget()
self.__amount_spinbox.grid_forget()
self.__all_children_amount_spinbox_label.grid_forget()
self.__all_children_amount_spinbox.grid_forget()
self.__apply_to_all_children_button.grid_forget()
prt_name = self.__port_combobox_var.get()
port = self.__state.model.get_port(name=prt_name)
buttons_to_change_state_of = [
self.__rename_port_button,
self.__remove_port_button,
self.__force_connection_checkbox,
self.__compatible_with_edit_button,
]
if port:
self.__selected_port = port
change_controls_state(tk.NORMAL, *buttons_to_change_state_of)
compatible_ports = self.__state.model.get_ports_by_ids(port.compatible_with)
self.__compatible_with_listbox.set_items(compatible_ports) # Fill the 'compatible with' listbox
self.__force_connection_checkbox_var.set(port.force_connection)
self.__update_tree() # Update tree values
self.__taxonomy_tree.grid(row=0, column=1, sticky=tk.NSEW) # Show tree
else:
self.__selected_port = None
self.__force_connection_checkbox_var.set(False)
self.__compatible_with_listbox.set_items([]) # Clear 'compatible with' listbox
change_controls_state(tk.DISABLED,
*buttons_to_change_state_of)
self.__taxonomy_tree.grid_forget() # Hide tree
def __update_tree(self) -> None:
"""Updates every leaf component in the taxonomy treeview."""
leaf_cmps = self.__state.model.get_components(is_leaf=True)
self.__taxonomy_tree.update_values(*leaf_cmps)
# SubscribesToListeners
def _subscribe_to_events(self) -> None:
pub.subscribe(self.__on_model_loaded, actions.MODEL_LOADED)
pub.subscribe(self.__on_taxonomy_edited, actions.TAXONOMY_EDITED)
pub.subscribe(self._reset, actions.RESET)
def __on_taxonomy_edited(self):
"""Executed whenever the structure of the taxonomy changes."""
self.__build_tree()
self.__selected_component = None
self.__cmp_label_var.set('')
self.__cmp_label.grid_forget()
self.__amount_spinbox_label.grid_forget()
self.__amount_spinbox.grid_forget()
self.__all_children_amount_spinbox_label.grid_forget()
self.__all_children_amount_spinbox.grid_forget()
self.__apply_to_all_children_button.grid_forget()
def __on_model_loaded(self):
"""Executed whenever a model is loaded from file."""
self._reset()
self.__build_tree()
ports_names = self.__state.model.get_all_ports_names()
self.__port_combobox['values'] = sorted(ports_names)
# Resetable
def _reset(self) -> None:
self.__selected_port = None
self.__selected_component = None
self.__taxonomy_tree.set_items([])
self.__port_combobox_var.set(SELECT_PORT)
self.__port_combobox['values'] = []
self.__taxonomy_tree.grid_forget() # Hide treeview
# Class-specific
# Ports
def __on_add(self) -> None:
"""Executed whenever the __add_port_button is pressed."""
AskStringWindow(self, self.__add, 'Add port', 'Enter name of a new port.')
def __add(self, name: str) -> None:
"""Executed after creating of a new port.
:param name: Name of a new port.
"""
prt = Port(name)
self.__state.model.add_port(prt)
self.__selected_port = prt
self.__port_combobox['values'] = sorted(self.__state.model.get_all_ports_names())
self.__port_combobox_var.set(prt.name)
change_controls_state(tk.NORMAL,
self.__rename_port_button,
self.__remove_port_button,
self.__force_connection_checkbox,
self.__compatible_with_edit_button)
def __on_rename(self) -> None:
"""Executed whenever the __rename_port_button is pressed."""
if self.__selected_port:
AskStringWindow(self, self.__rename, 'Rename port',
f'Enter new name for "{self.__selected_port.name}" port.',
string=self.__selected_port.name)
def __rename(self, new_name: str) -> None:
"""Executed after renaming of a port.
:param new_name: New name for the __selected_port.
"""
prt = self.__state.model.rename_port(self.__selected_port, new_name)
self.__port_combobox['values'] = sorted(self.__state.model.get_all_ports_names())
self.__port_combobox_var.set(prt.name)
def __remove(self) -> None:
"""Executed whenever the __remove_port_button is pressed.
Removes __selected_port from model.
"""
if self.__selected_port:
removed_prt = self.__state.model.remove_port(self.__selected_port)
updated_combobox_values = [val for val in [*self.__port_combobox['values']] if val != removed_prt.name]
self.__port_combobox['values'] = updated_combobox_values
self.__selected_port = None
self.__port_combobox_var.set(SELECT_PORT)
self.__compatible_with_listbox.set_items([]) # Clear the compatible with
change_controls_state(tk.DISABLED,
self.__rename_port_button,
self.__remove_port_button,
self.__force_connection_checkbox,
self.__compatible_with_edit_button)
# Hide component-specific widgets
self.__cmp_label.grid_forget()
self.__amount_spinbox_label.grid_forget()
self.__amount_spinbox.grid_forget()
self.__all_children_amount_spinbox_label.grid_forget()
self.__all_children_amount_spinbox.grid_forget()
self.__apply_to_all_children_button.grid_forget()
self.__update_tree()
def __on_edit_compatible_with(self):
"""Executed whenever the __compatible_with_edit_button is pressed."""
if self.__selected_port:
all_ports = self.__state.model.ports
compatible_ports = [p for p in all_ports if p.id_ in self.__selected_port.compatible_with]
ports_rest = [p for p in all_ports if p not in compatible_ports and p.id_ != self.__selected_port.id_]
SelectPortsWindow(self, self.__selected_port, compatible_ports, ports_rest,
callback=self.__edit_compatible_with)
def __edit_compatible_with(self, ports: List[Port]) -> None:
"""Executed after editing the list of ports compatible with __selected_port.
:param ports: New list of ports compatible with __selected_port.
"""
if self.__selected_port:
ports.sort(key=lambda x: x.name)
self.__compatible_with_listbox.set_items(ports)
self.__state.model.update_ports_compatibility(self.__selected_port, ports)
def __on_force_connection_toggled(self, *_):
"""Executed whenever the __force_connection_checkbox is toggled."""
if self.__selected_port:
value = self.__force_connection_checkbox_var.get()
self.__selected_port.force_connection = value
def __on_amount_changed(self, *_):
"""Executed whenever the __amount_spinbox value changes."""
if self.__selected_component and self.__selected_port:
value = None
try:
value = self.__amount_spinbox_var.get()
except tk.TclError:
pass
finally:
if value:
self.__selected_component.ports[self.__selected_port.id_] = value
elif self.__selected_port.id_ in self.__selected_component.ports:
del self.__selected_component.ports[self.__selected_port.id_]
self.__taxonomy_tree.update_values(self.__selected_component)
def __apply_to_all_children(self):
"""Executed whenever the __apply_to_all_children_button is pressed."""
if self.__selected_component and self.__selected_port:
value = None
try:
value = self.__all_children_amount_spinbox_var.get()
except tk.TclError:
pass
finally:
updated_components = self.__state.model.set_ports_amount_to_all_components_children(
self.__selected_component, self.__selected_port, value)
self.__taxonomy_tree.update_values(*updated_components)
class ResourcesTab(Tab, HasCommonSetup, SubscribesToEvents, Resetable):
"""Used to create, edit and remove components.
Attributes:
__selected_resource: Currently selected resource in the resources listview.
__selected_component: Currently selected component in the components taxonomy view.
"""
def __init__(self, parent_notebook):
self.__state = State()
self.__selected_resource: Optional[Resource] = None
self.__selected_component: Optional[Component] = None
Tab.__init__(self, parent_notebook, TAB_NAME)
HasCommonSetup.__init__(self)
SubscribesToEvents.__init__(self)
# HasCommonSetup
def _create_widgets(self):
self.__taxonomy_tree = ScrollbarListbox(
self,
on_select_callback=self.__on_select_tree_item,
heading=TREEVIEW_HEADING,
extract_id=lambda x: x.id_,
extract_text=lambda x: x.name,
extract_ancestor=lambda x: ''
if x.parent_id is None else x.parent_id,
extract_values=self.__extract_values,
columns=[Column('Produces')],
values=self.__state.model.taxonomy)
self.__left_frame = ttk.Frame(self)
# Resources combobox
self.__resource_combobox_var = tk.StringVar(value=SELECT_RESOURCE)
self.__resource_combobox_var.trace('w', self.__on_combobox_changed)
self.__resource_combobox = ttk.Combobox(
self.__left_frame,
state='readonly',
font=FONT,
textvariable=self.__resource_combobox_var)
# Fill the Combobox
resources_names = self.__state.model.get_all_resources_names()
self.__resource_combobox['values'] = sorted(resources_names)
self.__resource_combobox_var.set(SELECT_RESOURCE)
# C(r)ud Buttons
self.__add_resource_button = ttk.Button(self.__left_frame,
text='Add',
state=tk.NORMAL,
command=self.__on_add)
self.__rename_resource_button = ttk.Button(self.__left_frame,
text='Rename',
state=tk.DISABLED,
command=self.__on_rename)
self.__remove_resource_button = ttk.Button(self.__left_frame,
text='Remove',
state=tk.DISABLED,
command=self.__remove)
# Cmp label
self.__cmp_label_var = tk.StringVar(value='')
self.__cmp_label = ttk.Label(self.__left_frame,
textvariable=self.__cmp_label_var,
style='Big.TLabel',
anchor=tk.CENTER)
self.__produces_spinbox_label = ttk.Label(self.__left_frame,
text='Produces:')
self.__produces_spinbox_var = tk.IntVar(value='')
self.__produces_spinbox_var.trace('w', self.__on_produced_changed)
self.__produces_spinbox = ttk.Spinbox(
self.__left_frame,
from_=-math.inf,
to=math.inf,
textvariable=self.__produces_spinbox_var)
self.__all_children_produce_spinbox_label = ttk.Label(
self.__left_frame, text='Produces:')
self.__all_children_produce_spinbox_var = tk.IntVar(value=0)
self.__all_children_produce_spinbox = ttk.Spinbox(
self.__left_frame,
from_=-math.inf,
to=math.inf,
textvariable=self.__all_children_produce_spinbox_var)
self.__apply_to_all_children_button = ttk.Button(
self.__left_frame,
text='Apply to all children',
command=self.__apply_to_all_children)
def _setup_layout(self):
self.__taxonomy_tree.grid(row=0, column=1, sticky=tk.NSEW)
self.__left_frame.grid(row=0,
column=0,
sticky=tk.NSEW,
pady=FRAME_PAD_Y,
padx=FRAME_PAD_X)
self.__resource_combobox.grid(row=0,
column=0,
columnspan=2,
sticky=tk.NSEW,
pady=CONTROL_PAD_Y)
self.__add_resource_button.grid(row=1,
column=0,
columnspan=2,
sticky=tk.NSEW,
pady=CONTROL_PAD_Y)
self.__rename_resource_button.grid(row=2,
column=0,
columnspan=2,
sticky=tk.NSEW,
pady=CONTROL_PAD_Y)
self.__remove_resource_button.grid(row=3,
column=0,
columnspan=2,
sticky=tk.NSEW,
pady=CONTROL_PAD_Y)
self.__cmp_label.grid(row=4,
column=0,
columnspan=2,
pady=CONTROL_PAD_Y,
sticky=tk.EW)
self.__produces_spinbox_label.grid(row=5,
column=0,
pady=CONTROL_PAD_Y,
sticky=tk.W)
self.__produces_spinbox.grid(row=5,
column=1,
pady=CONTROL_PAD_Y,
sticky=tk.NSEW)
self.__all_children_produce_spinbox_label.grid(row=5,
column=0,
pady=CONTROL_PAD_Y,
sticky=tk.W)
self.__all_children_produce_spinbox.grid(row=5,
column=1,
pady=CONTROL_PAD_Y,
sticky=tk.NSEW)
self.__apply_to_all_children_button.grid(row=6,
column=0,
columnspan=2,
sticky=tk.NSEW,
pady=CONTROL_PAD_Y)
self.columnconfigure(0, weight=1, uniform='fred')
self.columnconfigure(1, weight=3, uniform='fred')
self.rowconfigure(0, weight=1)
self.__left_frame.columnconfigure(1, weight=1)
# Hide widgets
self.__taxonomy_tree.grid_forget()
self.__cmp_label.grid_forget()
self.__produces_spinbox_label.grid_forget()
self.__produces_spinbox.grid_forget()
self.__all_children_produce_spinbox_label.grid_forget()
self.__all_children_produce_spinbox.grid_forget()
self.__apply_to_all_children_button.grid_forget()
# SubscribesToListeners
def _subscribe_to_events(self):
pub.subscribe(self.__on_model_loaded, actions.MODEL_LOADED)
pub.subscribe(self.__on_taxonomy_edited, actions.TAXONOMY_EDITED)
pub.subscribe(self._reset, actions.RESET)
# HasTaxonomyTree
def __on_select_tree_item(self, cmp_id: int) -> None:
"""Executed whenever a tree item is selected (by mouse click).
:param cmp_id: Id of the selected component.
"""
if self.__selected_resource:
selected_cmp: Component = self.__state.model.get_component(
id_=cmp_id)
self.__selected_component = selected_cmp
self.__cmp_label.grid(row=4,
column=0,
columnspan=2,
pady=CONTROL_PAD_Y,
sticky=tk.EW)
self.__cmp_label_var.set(trim_string(selected_cmp.name, length=23))
if selected_cmp.is_leaf:
self.__all_children_produce_spinbox_label.grid_forget(
) # Hide widgets (changing all children)
self.__all_children_produce_spinbox.grid_forget()
self.__apply_to_all_children_button.grid_forget()
produced = 0
if self.__selected_resource.id_ in selected_cmp.produces:
produced = selected_cmp.produces[
self.__selected_resource.id_]
self.__produces_spinbox_var.set(produced)
self.__produces_spinbox_label.grid(row=5,
column=0,
pady=CONTROL_PAD_Y,
sticky=tk.W)
self.__produces_spinbox.grid(row=5,
column=1,
pady=CONTROL_PAD_Y,
sticky=tk.NSEW)
else:
self.__produces_spinbox_label.grid_forget(
) # Hide widgets for leaves
self.__produces_spinbox.grid_forget()
# Show widgets (changing all children)
self.__all_children_produce_spinbox_var.set(0)
self.__all_children_produce_spinbox_label.grid(
row=5, column=0, pady=CONTROL_PAD_Y, sticky=tk.W)
self.__all_children_produce_spinbox.grid(row=5,
column=1,
pady=CONTROL_PAD_Y,
sticky=tk.NSEW)
self.__apply_to_all_children_button.grid(row=6,
column=0,
columnspan=2,
sticky=tk.NSEW,
pady=CONTROL_PAD_Y)
def __extract_values(self, cmp: Component) -> Tuple[Any, ...]:
"""Extracts the data of the component to show in the taxonomy view.
:param cmp: Component from which to extract the data.
:return: Tuple containing data about component
(amount of produced resource of type __selected_resource,).
"""
produces = ''
if self.__selected_resource:
if self.__selected_resource.id_ in cmp.produces:
produces = cmp.produces[self.__selected_resource.id_]
return produces, # Coma means 1-element tuple
def __build_tree(self) -> None:
"""Fills the tree view with components from model."""
self.__taxonomy_tree.set_items(self.__state.model.taxonomy)
def __on_model_loaded(self):
"""Executed whenever a model is loaded from file."""
self._reset()
self.__build_tree()
resources_names = self.__state.model.get_all_resources_names()
self.__resource_combobox['values'] = sorted(resources_names)
self.__resource_combobox_var.set(SELECT_RESOURCE)
def __on_taxonomy_edited(self) -> None:
"""Executed whenever the structure of the taxonomy changes."""
self._reset()
self.__build_tree()
# Resetable
def _reset(self) -> None:
self.__selected_component = None
self.__selected_resource = None
# Hide widgets
self.__produces_spinbox_label.grid_forget()
self.__produces_spinbox.grid_forget()
self.__all_children_produce_spinbox_label.grid_forget()
self.__all_children_produce_spinbox.grid_forget()
self.__apply_to_all_children_button.grid_forget()
# Set entries to default
self.__resource_combobox['values'] = ()
self.__resource_combobox_var.set(SELECT_RESOURCE)
self.__produces_spinbox_var.set('')
self.__all_children_produce_spinbox_var.set('')
self.__cmp_label_var.set('')
# Disable buttons
change_controls_state(tk.DISABLED, self.__rename_resource_button,
self.__remove_resource_button)
self.__taxonomy_tree.set_items([])
self.__taxonomy_tree.grid_forget()
# Class-specific
def __on_combobox_changed(self, *_):
"""Executed whenever the __resource_combobox value changes."""
res_name = self.__resource_combobox_var.get()
resource = self.__state.model.get_resource(name=res_name)
if resource:
# Enable buttons
change_controls_state(tk.NORMAL, self.__rename_resource_button,
self.__remove_resource_button)
self.__selected_resource = resource
if self.__selected_component:
if self.__selected_component.is_leaf:
produced = 0
if resource.id_ in self.__selected_component.produces:
produced = self.__selected_component.produces[
self.__selected_resource.id_]
self.__produces_spinbox_var.set(produced)
else:
self.__all_children_produce_spinbox_var.set('')
# Show the taxonomy tree
self.__taxonomy_tree.grid(row=0, column=1, sticky=tk.NSEW)
self.__update_tree()
def __update_tree(self):
"""Updates every leaf component in the taxonomy treeview."""
leaf_cmps = self.__state.model.get_components(is_leaf=True)
self.__taxonomy_tree.update_values(*leaf_cmps)
def __on_add(self):
"""Executed whenever the __add_resource_button is pressed."""
AskStringWindow(self, self.__add, 'Add resource',
'Enter name of a new resource.')
def __add(self, name: str):
"""Executed after creating of a new resource.
:param name: Name of a new resource.
"""
res = Resource(name)
self.__state.model.add_resource(res)
self.__selected_resource = res
self.__resource_combobox['values'] = sorted(
self.__state.model.get_all_resources_names())
self.__resource_combobox_var.set(res.name)
# Enable buttons
change_controls_state(tk.NORMAL, self.__rename_resource_button,
self.__remove_resource_button)
# Show the taxonomy tree
self.__taxonomy_tree.grid(row=0, column=1, sticky=tk.NSEW)
self.__update_tree()
def __on_rename(self) -> None:
"""Executed whenever the __rename_resource_button is pressed."""
if self.__selected_resource:
AskStringWindow(
self,
self.__rename,
'Rename resource',
f'Enter new name for "{self.__selected_resource.name}" resource.',
string=self.__selected_resource.name)
def __rename(self, new_name: str) -> None:
"""Executed after renaming of a resource.
:param new_name: New name for the __selected_resource.
"""
res = self.__state.model.rename_resource(self.__selected_resource,
new_name)
self.__resource_combobox['values'] = sorted(
self.__state.model.get_all_resources_names())
self.__resource_combobox_var.set(res.name)
def __remove(self):
"""Executed whenever the __remove_resource_button is pressed.
Removes __selected_resource from model.
"""
if self.__selected_resource:
self.__state.model.remove_resource(self.__selected_resource)
self.__resource_combobox['values'] = sorted(
self.__state.model.get_all_resources_names())
self.__selected_resource = None
self.__resource_combobox_var.set(SELECT_RESOURCE)
# Disable buttons
change_controls_state(tk.DISABLED, self.__rename_resource_button,
self.__remove_resource_button)
# Hide widgets
self.__cmp_label.grid_forget()
self.__produces_spinbox_label.grid_forget()
self.__produces_spinbox.grid_forget()
self.__all_children_produce_spinbox_label.grid_forget()
self.__all_children_produce_spinbox.grid_forget()
self.__apply_to_all_children_button.grid_forget()
# Hide the taxonomy tree
self.__taxonomy_tree.grid_forget()
def __on_produced_changed(self, *_):
"""Executed whenever the __produces_spinbox value changes."""
if self.__selected_component and self.__selected_resource:
value = None
try:
value = self.__produces_spinbox_var.get()
except tk.TclError:
pass
finally:
if value:
self.__selected_component.produces[
self.__selected_resource.id_] = value
elif self.__selected_resource.id_ in self.__selected_component.produces:
del self.__selected_component.produces[
self.__selected_resource.id_]
self.__taxonomy_tree.update_values(self.__selected_component)
def __apply_to_all_children(self):
"""Executed whenever the __apply_to_all_children_button is pressed."""
if self.__selected_component and self.__selected_resource:
value = None
try:
value = self.__all_children_produce_spinbox_var.get()
except tk.TclError:
pass
finally:
updated_components = self.__state.model.set_resource_production_to_all_components_children(
self.__selected_component, self.__selected_resource, value)
self.__taxonomy_tree.update_values(*updated_components)