jumpserver/apps/terminal/applets/chrome/code_dialog.py

import functools
import threading
import tkinter as tk
from tkinter import StringVar, messagebox
from tkinter import ttk


class CodeDialog(object):

    def __init__(self, title=None, label="Code Dialog"):
        self.root = tk.Tk()
        self.root.title(title)
        self.code = StringVar()
        mainframe = ttk.Frame(self.root, padding="12 12 12 12")
        mainframe.grid(column=0, row=0, )
        self.label = ttk.Label(mainframe, text=label, width=10)
        self.input = ttk.Entry(mainframe, textvariable=self.code, width=20)
        self.button = ttk.Button(mainframe, text="ok", command=self.click_ok, width=5, )
        self.label.grid(row=1, column=0)
        self.input.grid(row=1, column=1)
        self.button.grid(row=2, column=1, sticky=tk.E)
        self.root.bind('<Return>', self.click_ok)

    def wait_string(self):
        # 局中
        self.root.eval('tk::PlaceWindow . center')
        self.root.mainloop()
        return self.code.get()

    def click_ok(self, *args, **kwargs):
        if not self.code.get():
            messagebox.showwarning(title="warning", message="code is empty")
            return
        self.root.destroy()


class TkProgressBar(object):
    # 30s
    wait_max_time = 3000 * 10

    def __init__(self, wait_func=None):
        self._wait_func = wait_func
        self._done = threading.Event()
        self._root = None
        self._wait_time = 0

    def _check(self):
        if self._done.isSet():
            self._root.destroy()
            return
        if self._wait_time >= self.wait_max_time:
            self._root.destroy()
            return
        self._wait_time += 100
        self._root.after(100, self._check)

    def stop(self):
        self._done.set()

    def show(self):
        if not self._wait_func:
            return
        root = tk.Tk()
        width, height = root.winfo_screenwidth(), root.winfo_screenheight()
        root.geometry('%dx%d' % (width, height))
        root.title('Progress')
        root.grid()
        pb_length = width - 20
        pb = ttk.Progressbar(root, orient='horizontal', length=pb_length, mode='indeterminate')
        pb.pack(expand=True, padx=10, pady=10)

        self._root = root
        self._root.after(60, pb.start)
        self._root.after(100, self._check)

        def _wait_run_func():
            self._wait_func()
            self.stop()
            print('wait func done')

        self._root.attributes("-topmost", True)
        self._root.attributes("-fullscreen", True)
        threading.Thread(target=_wait_run_func).start()
        self._root.mainloop()


# progress bar 装饰器，用于显示进度动画
# 此方法会创建一个全屏的进度条窗口
def wrapper_progress_bar(func):
    def inner(*args, **kwargs):
        wait_func = functools.partial(func, *args, **kwargs)
        tk_process = TkProgressBar(wait_func=wait_func)
        tk_process.show()

    return inner


if __name__ == '__main__':
    # code = CodeDialog(title="Code Dialog", label="Code: ").wait_string()
    # print(code)
    import time
    progress = TkProgressBar(wait_func=lambda: time.sleep(15))
    progress.show()
    print('end')