core-extra/coretk/coretk/dialogs/sessions.py

import logging
import tkinter as tk
from tkinter import ttk

from core.api.grpc import core_pb2
from coretk.dialogs.dialog import Dialog
from coretk.images import ImageEnum, Images


class SessionsDialog(Dialog):
    def __init__(self, master, app):
        super().__init__(master, app, "Sessions", modal=True)
        self.selected = False
        self.selected_id = None
        self.tree = None
        self.draw()

    def draw(self):
        self.columnconfigure(0, weight=1)
        self.draw_description()
        self.draw_tree()
        self.draw_buttons()

    def draw_description(self):
        """
        write a short description
        :return: nothing
        """
        label = ttk.Label(
            self,
            text="Below is a list of active CORE sessions. Double-click to \n"
            "connect to an existing session. Usually, only sessions in \n"
            "the RUNTIME state persist in the daemon, except for the \n"
            "one you might be concurrently editting.",
        )
        label.grid(row=0, sticky="ew", pady=5)

    def draw_tree(self):
        self.tree = ttk.Treeview(
            self, columns=("id", "state", "nodes"), show="headings"
        )
        self.tree.grid(row=1, sticky="nsew")
        self.tree.column("id", stretch=tk.YES)
        self.tree.heading("id", text="ID")
        self.tree.column("state", stretch=tk.YES)
        self.tree.heading("state", text="State")
        self.tree.column("nodes", stretch=tk.YES)
        self.tree.heading("nodes", text="Node Count")

        response = self.app.core.client.get_sessions()
        logging.info("sessions: %s", response)
        for index, session in enumerate(response.sessions):
            state_name = core_pb2.SessionState.Enum.Name(session.state)
            self.tree.insert(
                "",
                tk.END,
                text=str(session.id),
                values=(session.id, state_name, session.nodes),
            )
        self.tree.bind("<Double-1>", self.on_selected)
        self.tree.bind("<<TreeviewSelect>>", self.click_select)

        yscrollbar = ttk.Scrollbar(self, orient="vertical", command=self.tree.yview)
        yscrollbar.grid(row=1, column=1, sticky="ns")
        self.tree.configure(yscrollcommand=yscrollbar.set)

        xscrollbar = ttk.Scrollbar(self, orient="horizontal", command=self.tree.xview)
        xscrollbar.grid(row=2, sticky="ew", pady=5)
        self.tree.configure(xscrollcommand=xscrollbar.set)

    def draw_buttons(self):
        frame = ttk.Frame(self)
        for i in range(4):
            frame.columnconfigure(i, weight=1)
        frame.grid(row=3, sticky="ew")

        image = Images.get(ImageEnum.DOCUMENTNEW, 16)
        b = ttk.Button(
            frame, image=image, text="New", compound=tk.LEFT, command=self.click_new
        )
        b.image = image
        b.grid(row=0, padx=2, sticky="ew")

        image = Images.get(ImageEnum.FILEOPEN, 16)
        b = ttk.Button(
            frame,
            image=image,
            text="Connect",
            compound=tk.LEFT,
            command=self.click_connect,
        )
        b.image = image
        b.grid(row=0, column=1, padx=2, sticky="ew")

        image = Images.get(ImageEnum.EDITDELETE, 16)
        b = ttk.Button(
            frame,
            image=image,
            text="Shutdown",
            compound=tk.LEFT,
            command=self.click_shutdown,
        )
        b.image = image
        b.grid(row=0, column=2, padx=2, sticky="ew")

        b = ttk.Button(frame, text="Cancel", command=self.click_new)
        b.grid(row=0, column=3, padx=2, sticky="ew")

    def click_new(self):
        self.app.core.create_new_session()
        self.destroy()

    def click_select(self, event):
        item = self.tree.selection()
        session_id = int(self.tree.item(item, "text"))
        self.selected = True
        self.selected_id = session_id

    def click_connect(self):
        """
        if no session is selected yet, create a new one else join that session

        :return: nothing
        """
        if self.selected and self.selected_id is not None:
            self.join_session(self.selected_id)
        elif not self.selected and self.selected_id is None:
            self.click_new()
        else:
            logging.error("sessions invalid state")

    def click_shutdown(self):
        """
        if no session is currently selected create a new session else shut the selected
        session down.

        :return: nothing
        """
        if self.selected and self.selected_id is not None:
            self.shutdown_session(self.selected_id)
        elif not self.selected and self.selected_id is None:
            self.click_new()
        else:
            logging.error("querysessiondrawing.py invalid state")

    def join_session(self, session_id):
        self.app.core.join_session(session_id)
        self.destroy()

    def on_selected(self, event):
        item = self.tree.selection()
        sid = int(self.tree.item(item, "text"))
        self.join_session(sid)

    def shutdown_session(self, sid):
        self.app.core.shutdown_session(sid)
        self.click_new()
        self.destroy()