162 lines
6.1 KiB
Python
162 lines
6.1 KiB
Python
"""
|
|
core node services
|
|
"""
|
|
import logging
|
|
import tkinter as tk
|
|
from tkinter import messagebox, ttk
|
|
from typing import TYPE_CHECKING, Any, Set
|
|
|
|
from core.gui.dialogs.configserviceconfig import ConfigServiceConfigDialog
|
|
from core.gui.dialogs.dialog import Dialog
|
|
from core.gui.themes import FRAME_PAD, PADX, PADY
|
|
from core.gui.widgets import CheckboxList, ListboxScroll
|
|
|
|
if TYPE_CHECKING:
|
|
from core.gui.app import Application
|
|
from core.gui.graph.node import CanvasNode
|
|
|
|
|
|
class NodeConfigServiceDialog(Dialog):
|
|
def __init__(
|
|
self,
|
|
master: Any,
|
|
app: "Application",
|
|
canvas_node: "CanvasNode",
|
|
services: Set[str] = None,
|
|
):
|
|
title = f"{canvas_node.core_node.name} Config Services"
|
|
super().__init__(master, app, title)
|
|
self.app = app
|
|
self.canvas_node = canvas_node
|
|
self.node_id = canvas_node.core_node.id
|
|
self.groups = None
|
|
self.services = None
|
|
self.current = None
|
|
if services is None:
|
|
services = set(canvas_node.core_node.config_services)
|
|
self.current_services = services
|
|
self.draw()
|
|
|
|
def draw(self):
|
|
self.top.columnconfigure(0, weight=1)
|
|
self.top.rowconfigure(0, weight=1)
|
|
|
|
frame = ttk.Frame(self.top)
|
|
frame.grid(stick="nsew", pady=PADY)
|
|
frame.rowconfigure(0, weight=1)
|
|
for i in range(3):
|
|
frame.columnconfigure(i, weight=1)
|
|
label_frame = ttk.LabelFrame(frame, text="Groups", padding=FRAME_PAD)
|
|
label_frame.grid(row=0, column=0, sticky="nsew")
|
|
label_frame.rowconfigure(0, weight=1)
|
|
label_frame.columnconfigure(0, weight=1)
|
|
self.groups = ListboxScroll(label_frame)
|
|
self.groups.grid(sticky="nsew")
|
|
for group in sorted(self.app.core.config_services_groups):
|
|
self.groups.listbox.insert(tk.END, group)
|
|
self.groups.listbox.bind("<<ListboxSelect>>", self.handle_group_change)
|
|
self.groups.listbox.selection_set(0)
|
|
|
|
label_frame = ttk.LabelFrame(frame, text="Services")
|
|
label_frame.grid(row=0, column=1, sticky="nsew")
|
|
label_frame.columnconfigure(0, weight=1)
|
|
label_frame.rowconfigure(0, weight=1)
|
|
self.services = CheckboxList(
|
|
label_frame, self.app, clicked=self.service_clicked, padding=FRAME_PAD
|
|
)
|
|
self.services.grid(sticky="nsew")
|
|
|
|
label_frame = ttk.LabelFrame(frame, text="Selected", padding=FRAME_PAD)
|
|
label_frame.grid(row=0, column=2, sticky="nsew")
|
|
label_frame.rowconfigure(0, weight=1)
|
|
label_frame.columnconfigure(0, weight=1)
|
|
|
|
self.current = ListboxScroll(label_frame)
|
|
self.current.grid(sticky="nsew")
|
|
self.draw_current_services()
|
|
|
|
frame = ttk.Frame(self.top)
|
|
frame.grid(stick="ew")
|
|
for i in range(4):
|
|
frame.columnconfigure(i, weight=1)
|
|
button = ttk.Button(frame, text="Configure", command=self.click_configure)
|
|
button.grid(row=0, column=0, sticky="ew", padx=PADX)
|
|
button = ttk.Button(frame, text="Save", command=self.click_save)
|
|
button.grid(row=0, column=1, sticky="ew", padx=PADX)
|
|
button = ttk.Button(frame, text="Remove", command=self.click_remove)
|
|
button.grid(row=0, column=2, sticky="ew", padx=PADX)
|
|
button = ttk.Button(frame, text="Cancel", command=self.click_cancel)
|
|
button.grid(row=0, column=3, sticky="ew")
|
|
|
|
# trigger group change
|
|
self.groups.listbox.event_generate("<<ListboxSelect>>")
|
|
|
|
def handle_group_change(self, event: tk.Event = None):
|
|
selection = self.groups.listbox.curselection()
|
|
if selection:
|
|
index = selection[0]
|
|
group = self.groups.listbox.get(index)
|
|
self.services.clear()
|
|
for name in sorted(self.app.core.config_services_groups[group]):
|
|
checked = name in self.current_services
|
|
self.services.add(name, checked)
|
|
|
|
def service_clicked(self, name: str, var: tk.IntVar):
|
|
if var.get() and name not in self.current_services:
|
|
self.current_services.add(name)
|
|
elif not var.get() and name in self.current_services:
|
|
self.current_services.remove(name)
|
|
self.draw_current_services()
|
|
self.canvas_node.core_node.config_services[:] = self.current_services
|
|
|
|
def click_configure(self):
|
|
current_selection = self.current.listbox.curselection()
|
|
if len(current_selection):
|
|
dialog = ConfigServiceConfigDialog(
|
|
self,
|
|
self.app,
|
|
self.current.listbox.get(current_selection[0]),
|
|
self.canvas_node,
|
|
self.node_id,
|
|
)
|
|
if not dialog.has_error:
|
|
dialog.show()
|
|
self.draw_current_services()
|
|
else:
|
|
messagebox.showinfo(
|
|
"Config Service Configuration",
|
|
"Select a service to configure",
|
|
parent=self,
|
|
)
|
|
|
|
def draw_current_services(self):
|
|
self.current.listbox.delete(0, tk.END)
|
|
for name in sorted(self.current_services):
|
|
self.current.listbox.insert(tk.END, name)
|
|
if self.is_custom_service(name):
|
|
self.current.listbox.itemconfig(tk.END, bg="green")
|
|
|
|
def click_save(self):
|
|
self.canvas_node.core_node.config_services[:] = self.current_services
|
|
logging.info(
|
|
"saved node config services: %s", self.canvas_node.core_node.config_services
|
|
)
|
|
self.destroy()
|
|
|
|
def click_cancel(self):
|
|
self.current_services = None
|
|
self.destroy()
|
|
|
|
def click_remove(self):
|
|
cur = self.current.listbox.curselection()
|
|
if cur:
|
|
service = self.current.listbox.get(cur[0])
|
|
self.current.listbox.delete(cur[0])
|
|
self.current_services.remove(service)
|
|
for checkbutton in self.services.frame.winfo_children():
|
|
if checkbutton["text"] == service:
|
|
checkbutton.invoke()
|
|
return
|
|
|
|
def is_custom_service(self, service: str) -> bool:
|
|
return service in self.canvas_node.config_service_configs
|