CoTan/system/gui.py

import tkinter
from tkinter.filedialog import askopenfilename
from tkinter.messagebox import showwarning, askokcancel
from tkinter.scrolledtext import ScrolledText

from system.controller import Plugin, NamingError, ConflictError

SCREEN = tkinter.Tk()
plugin = Plugin()
SCREEN.title('插件管理')
SCREEN.resizable(width=False, height=False)
SCREEN.geometry(f'+10+10')
bg_color = "#FFFAFA"  # 主颜色
SCREEN["bg"] = bg_color
botton_color = "#FFFAFA"  # 按钮颜色
word_color = "#000000"  # 文字颜色
FONT = ("黑体", 11)  # 设置字体
gui_width = 13  # 标准宽度
gui_height = 2
row = 0
column = 0


def code_window(name):
    global bg_color
    cli_screen = tkinter.Toplevel(bg=bg_color)
    cli_screen.title(f'插件查看器:{name}')
    cli_screen.geometry("+10+10")  # 设置所在位置
    cli_screen.resizable(width=False, height=False)

    class ScrolledCli(ScrolledText):
        def __init__(self, *args, **kwargs):
            super(ScrolledCli, self).__init__(*args, **kwargs)

        def insert(self, index, chars, *args):
            text.config(state=tkinter.NORMAL)
            super(ScrolledCli, self).insert(index, chars, *args)
            text.config(state=tkinter.DISABLED)

        def clear(self):
            text.config(state=tkinter.NORMAL)
            self.delete("0.0", tkinter.END)
            text.config(state=tkinter.DISABLED)

    text = ScrolledCli(cli_screen, font=("黑体", 11), height=30, width=100)
    text.grid(column=0, row=0, columnspan=5, sticky=tkinter.E + tkinter.W)
    text.config(state=tkinter.DISABLED)
    cli_screen.update()
    return text, cli_screen


def get_dir():
    plugin_dir_box.delete(0, tkinter.END)
    plugin_dir_box.insert(0, *plugin.get_dir())


def get_all_plugin():
    plugin_box.delete(0, tkinter.END)
    plugin_box.insert(0, *plugin.get_all_plugin())


def get_plugin():
    try:
        index = plugin_dir_box.curselection()[0]
    except IndexError:
        return False
    plugin_box.delete(0, tkinter.END)
    plugin_box.insert(0, *plugin.get_plugin(index))


def add_plugin():
    index = plugin_dir_box.curselection()[0]
    plugin_dir = askopenfilename(title='选择插件文件', filetypes=[("Python", ".py")])
    try:
        plugin_list = plugin.add_plugin(index, plugin_dir)
    except NamingError:
        showwarning('文件错误', '插件命名错误，命名规则:\ntemplate_[类\\方法名].py')
    except ConflictError:
        if askokcancel('提示', f'已经存在插件，是否需要尝试合并插件?\n[合并失败将产生不可逆的后果]'):
            plugin.merge_plugin(index, plugin_dir)
    except BaseException:
        showwarning('文件错误', '插件导入遇到了未知错误')
    else:
        plugin_box.delete(0, tkinter.END)
        plugin_box.insert(0, *plugin_list)


def del_plugin():
    index = plugin_box.curselection()[0]
    try:
        plugin_list = plugin.del_plugin(index)
    except BaseException:
        pass
    else:
        plugin_box.delete(0, tkinter.END)
        plugin_box.insert(0, *plugin_list)


def show_plugin():
    index = plugin_box.curselection()[0]
    try:
        code, name = plugin.show_plugin(index)
    except BaseException:
        pass
    else:
        code_window(name)[0].insert(tkinter.END, code)


def system_main():
    global SCREEN
    SCREEN.mainloop()


(tkinter.Label(SCREEN, text='【插件管理】', bg=bg_color, fg=word_color, font=FONT, width=gui_width*3, height=gui_height)
    .grid(column=column, row=row, columnspan=3))
row += 1
(tkinter.Button(SCREEN, bg=botton_color, fg=word_color, command=get_dir, text='查看插件列表', font=FONT, width=gui_width, height=gui_height)
 .grid(column=column, row=row, sticky=tkinter.E + tkinter.W))
(tkinter.Button(SCREEN, bg=botton_color, fg=word_color, command=get_all_plugin, text='查看所有插件', font=FONT, width=gui_width, height=gui_height)
 .grid(column=column + 1, row=row, sticky=tkinter.E + tkinter.W))
(tkinter.Button(SCREEN, bg=botton_color, fg=word_color, command=get_plugin, text='查看仓库插件', font=FONT, width=gui_width, height=gui_height)
 .grid(column=column + 2, row=row, sticky=tkinter.E + tkinter.W))
row += 1
plugin_box = tkinter.Listbox(SCREEN, width=gui_width * 3, height=gui_height * 5)
plugin_box.grid(column=column, row=row, columnspan=3, rowspan=5, sticky=tkinter.E + tkinter.W + tkinter.S + tkinter.N)
row += 5
(tkinter.Label(SCREEN, text='【插件仓库】', bg=bg_color, fg=word_color, font=FONT, width=gui_width*3, height=gui_height)
    .grid(column=column, row=row, columnspan=3))
row += 1
plugin_dir_box = tkinter.Listbox(SCREEN, width=gui_width * 3, height=gui_height * 5)
plugin_dir_box.grid(column=column, row=row, columnspan=3, rowspan=5, sticky=tkinter.E + tkinter.W + tkinter.S + tkinter.N)
row += 5
(tkinter.Button(SCREEN, bg=botton_color, fg=word_color, command=add_plugin, text='新增插件', font=FONT, width=gui_width, height=gui_height)
 .grid(column=column, row=row, sticky=tkinter.E + tkinter.W))
(tkinter.Button(SCREEN, bg=botton_color, fg=word_color, command=del_plugin, text='删除插件', font=FONT, width=gui_width, height=gui_height)
 .grid(column=column + 1, row=row, sticky=tkinter.E + tkinter.W))
(tkinter.Button(SCREEN, bg=botton_color, fg=word_color, command=show_plugin, text='查看插件', font=FONT, width=gui_width, height=gui_height)
 .grid(column=column + 2, row=row, sticky=tkinter.E + tkinter.W))