HGSSystem/tk_ui/admin_menu.py

import abc
import tkinter as tk

from tool.typing import *
from tool.tk import make_font, set_tk_disable_from_list

from . import admin


class AdminMenu(metaclass=abc.ABCMeta):
    def __init__(self, station: "admin.AdminStationBase", win: Union[tk.Frame, tk.Toplevel, tk.Tk], color: str,
                 title: str):
        self.station = station
        self.win = win
        self.color = color
        self.frame = tk.Frame(self.win)
        self.frame['bg'] = color
        self.menu_title = title
        self.btn: List[tk.Button] = []
        self.btn_name: List[str] = []
        self.__conf_font()

    def __conf_font(self, n: int = 1):
        self.btn_font_size = int(16 * n)

    def set_disable(self):
        set_tk_disable_from_list(self.btn)

    def reset_disable(self):
        set_tk_disable_from_list(self.btn, flat='normal')

    def conf_gui(self, color: str, n: int = 1):
        self.__conf_font(n)

        btn_font = make_font(size=self.btn_font_size, weight="bold")
        height = 0.02
        for btn, text in zip(self.btn, self.btn_name):
            btn['font'] = btn_font
            btn['text'] = text
            btn['bg'] = color
            btn.place(relx=0.02, rely=height, relwidth=0.96, relheight=0.1)
            height += 0.1 + 0.02

    def get_menu_title(self) -> str:
        return self.menu_title

    def get_menu_frame(self) -> Tuple[str, tk.Frame]:
        return self.menu_title, self.frame


class MainMenu(AdminMenu):
    def __init__(self, station, win, color):
        super().__init__(station, win, color, "主页")
        self.btn: List[tk.Button] = [tk.Button(self.frame) for _ in range(6)]
        self.btn_name = ["创建", "删除", "搜索", "更新", "数据分析", "退出登录"]

    def conf_gui(self, color: str, n: int = 1):
        super().conf_gui(color, n)
        self.btn[0]['command'] = lambda: self.create_command()
        self.btn[1]['command'] = lambda: self.delete_command()
        self.btn[2]['command'] = lambda: self.search_command()
        self.btn[3]['command'] = lambda: self.update_command()
        self.btn[4]['command'] = lambda: self.statistics_command()
        self.btn[5]['command'] = lambda: self.logout_command()

    def create_command(self):
        self.station.to_menu("创建")

    def delete_command(self):
        self.station.to_menu("删除")

    def search_command(self):
        self.station.to_menu("搜索")

    def update_command(self):
        self.station.to_menu("更新")

    def statistics_command(self):
        self.station.to_menu("数据分析")

    def logout_command(self):
        self.station.logout()


class CreateMenu(AdminMenu):
    def __init__(self, station, win, color):
        super().__init__(station, win, color, "创建")
        self.btn: List[tk.Button] = [tk.Button(self.frame) for _ in range(7)]
        self.btn_name = ["普通用户", "自动创建", "管理员用户", "垃圾袋",
                         "导出用户二维码", "导出垃圾袋二维码", "从CSV导入用户"]

    def conf_gui(self, color: str, n: int = 1):
        super().conf_gui(color, n)
        self.btn[0]['command'] = lambda: self.create_normal_user()
        self.btn[1]['command'] = lambda: self.create_auto_user()
        self.btn[2]['command'] = lambda: self.create_manager_user()
        self.btn[3]['command'] = lambda: self.create_garbage()
        self.btn[4]['command'] = lambda: self.export_user()
        self.btn[5]['command'] = lambda: self.export_garbage()
        self.btn[6]['command'] = lambda: self.create_user_from_csv()

    def create_normal_user(self):
        self.station.to_program("创建普通用户")

    def create_auto_user(self):
        self.station.to_program("创建自动用户")

    def create_manager_user(self):
        self.station.to_program("创建管理员")

    def create_garbage(self):
        self.station.to_program("创建垃圾袋")

    def export_user(self):
        self.station.to_program("导出用户二维码")

    def export_garbage(self):
        self.station.to_program("导出垃圾袋二维码")

    def create_user_from_csv(self):
        self.station.to_program("从CSV导入用户")


class DeleteMenu(AdminMenu):
    def __init__(self, station, win, color):
        super().__init__(station, win, color, "删除")
        self.btn: List[tk.Button] = [tk.Button(self.frame) for _ in range(5)]
        self.btn_name = ["用户", "多个用户", "垃圾袋", "多个垃圾袋", "所有垃圾袋"]

    def conf_gui(self, color: str, n: int = 1):
        super().conf_gui(color, n)
        self.btn[0]['command'] = lambda: self.del_user()
        self.btn[1]['command'] = lambda: self.del_users()
        self.btn[2]['command'] = lambda: self.del_garbage()
        self.btn[3]['command'] = lambda: self.del_garbage_more()
        self.btn[4]['command'] = lambda: self.del_all_garbage()

    def del_user(self):
        self.station.to_program("删除用户")

    def del_users(self):
        self.station.to_program("删除多个用户")

    def del_garbage(self):
        self.station.to_program("删除垃圾袋")

    def del_garbage_more(self):
        self.station.to_program("删除多个垃圾袋")

    def del_all_garbage(self):
        self.station.to_program("删除所有垃圾袋")


class SearchMenu(AdminMenu):
    def __init__(self, station, win, color):
        super().__init__(station, win, color, "搜索")
        self.btn: List[tk.Button] = [tk.Button(self.frame) for _ in range(6)]
        self.btn_name = ["用户", "高级搜索-用户", "垃圾袋", "高级搜索-垃圾袋", "高级搜索"]

    def conf_gui(self, color: str, n: int = 1):
        super().conf_gui(color, n)
        self.btn[0]['command'] = lambda: self.user_command()
        self.btn[1]['command'] = lambda: self.user_advanced_command()
        self.btn[2]['command'] = lambda: self.garbage_command()
        self.btn[3]['command'] = lambda: self.garbage_advanced_command()
        self.btn[4]['command'] = lambda: self.advanced_command()

    def user_command(self):
        self.station.to_program("搜索用户")

    def user_advanced_command(self):
        self.station.to_program("高级搜索-用户")

    def garbage_command(self):
        self.station.to_program("搜索垃圾袋")

    def garbage_advanced_command(self):
        self.station.to_program("高级搜索-垃圾袋")

    def advanced_command(self):
        self.station.to_program("高级搜索")


class UpdateMenu(AdminMenu):
    def __init__(self, station, win, color):
        super().__init__(station, win, color, "更新")
        self.btn: List[tk.Button] = [tk.Button(self.frame) for _ in range(4)]
        self.btn_name = ["用户-积分", "用户-累计分类信用", "垃圾袋-垃圾类型", "垃圾袋-检测结果"]

    def conf_gui(self, color: str, n: int = 1):
        super().conf_gui(color, n)
        self.btn[0]['command'] = lambda: self.update_score_command()
        self.btn[1]['command'] = lambda: self.update_reputation_command()
        self.btn[2]['command'] = lambda: self.update_garbage_type_command()
        self.btn[3]['command'] = lambda: self.update_garbage_result_command()

    def update_reputation_command(self):
        self.station.to_program("更新用户-垃圾分类信用")

    def update_score_command(self):
        self.station.to_program("更新用户-积分")

    def update_garbage_type_command(self):
        self.station.to_program("更新垃圾袋-垃圾类型")

    def update_garbage_result_command(self):
        self.station.to_program("更新垃圾袋-检测结果")


class StatisticsMenu(AdminMenu):
    def __init__(self, station, win, color):
        super().__init__(station, win, color, "数据分析")
        self.btn: List[tk.Button] = [tk.Button(self.frame) for _ in range(4)]
        self.btn_name = ["时段分析", "日期分析", "积分信用分析", "通过率"]

    def conf_gui(self, color: str, n: int = 1):
        super().conf_gui(color, n)
        self.btn[0]['command'] = self.statistics_time_command
        self.btn[1]['command'] = self.statistics_date_command
        self.btn[2]['command'] = self.statistics_user_command
        self.btn[3]['command'] = self.statistics_pass_command

    def statistics_time_command(self):
        self.station.to_menu("时段分析")

    def statistics_date_command(self):
        self.station.to_menu("日期分析")

    def statistics_user_command(self):
        self.station.to_menu("积分信用分析")

    def statistics_pass_command(self):
        self.station.to_menu("通过率")


class StatisticsTimeMenu(AdminMenu):
    def __init__(self, station, win, color):
        super().__init__(station, win, color, "时段分析")
        self.btn: List[tk.Button] = [tk.Button(self.frame) for _ in range(7)]
        self.btn_name = ["按投放区域", "按投放类型", "投放类型和区域", "按检查结果", "按检查结果和类型", "按检查结果和区域",
                         "详细分类"]

    def conf_gui(self, color: str, n: int = 1):
        super().conf_gui(color, n)
        self.btn[0]['command'] = self.by_loc
        self.btn[1]['command'] = self.by_type
        self.btn[2]['command'] = self.by_type_and_type
        self.btn[3]['command'] = self.by_result
        self.btn[4]['command'] = self.by_result_and_type
        self.btn[5]['command'] = self.by_result_and_loc
        self.btn[6]['command'] = self.by_detail

    def by_loc(self):
        self.station.to_program("时段分析-按投放区域")

    def by_type(self):
        self.station.to_program("时段分析-按投放类型")

    def by_type_and_type(self):
        self.station.to_program("时段分析-按投放类型和区域")

    def by_result(self):
        self.station.to_program("时段分析-按检查结果")

    def by_result_and_type(self):
        self.station.to_program("时段分析-按检查结果和类型")

    def by_result_and_loc(self):
        self.station.to_program("时段分析-按检查结果和区域")

    def by_detail(self):
        self.station.to_program("时段分析-详细分类")


class StatisticsDateMenu(AdminMenu):
    def __init__(self, station, win, color):
        super().__init__(station, win, color, "日期分析")
        self.btn: List[tk.Button] = [tk.Button(self.frame) for _ in range(2)]
        self.btn_name = ["最近7日", "最近30日"]

    def conf_gui(self, color: str, n: int = 1):
        super().conf_gui(color, n)
        self.btn[0]['command'] = self.by_7
        self.btn[1]['command'] = self.by_30

    def by_7(self):
        self.station.to_menu("日期分析-最近7日")

    def by_30(self):
        self.station.to_menu("日期分析-最近30日")


class StatisticsDate7Menu(AdminMenu):
    def __init__(self, station, win, color):
        super().__init__(station, win, color, "日期分析-最近7日")
        self.btn: List[tk.Button] = [tk.Button(self.frame) for _ in range(7)]
        self.btn_name = ["按投放区域", "按投放类型", "投放类型和区域", "按检查结果", "按检查结果和类型", "按检查结果和区域",
                         "详细分类"]

    def conf_gui(self, color: str, n: int = 1):
        super().conf_gui(color, n)
        self.btn[0]['command'] = self.by_loc
        self.btn[1]['command'] = self.by_type
        self.btn[2]['command'] = self.by_type_and_type
        self.btn[3]['command'] = self.by_result
        self.btn[4]['command'] = self.by_result_and_type
        self.btn[5]['command'] = self.by_result_and_loc
        self.btn[6]['command'] = self.by_detail

    def by_loc(self):
        self.station.to_program("最近7日-按投放区域")

    def by_type(self):
        self.station.to_program("最近7日-按投放类型")

    def by_type_and_type(self):
        self.station.to_program("最近7日-按投放类型和区域")

    def by_result(self):
        self.station.to_program("最近7日-按检查结果")

    def by_result_and_type(self):
        self.station.to_program("最近7日-按检查结果和类型")

    def by_result_and_loc(self):
        self.station.to_program("最近7日-按检查结果和区域")

    def by_detail(self):
        self.station.to_program("最近7日-详细分类")


class StatisticsDate30Menu(AdminMenu):
    def __init__(self, station, win, color):
        super().__init__(station, win, color, "日期分析-最近30日")
        self.btn: List[tk.Button] = [tk.Button(self.frame) for _ in range(7)]
        self.btn_name = ["按投放区域", "按投放类型", "投放类型和区域", "按检查结果", "按检查结果和类型", "按检查结果和区域",
                         "详细分类"]

    def conf_gui(self, color: str, n: int = 1):
        super().conf_gui(color, n)
        self.btn[0]['command'] = self.by_loc
        self.btn[1]['command'] = self.by_type
        self.btn[2]['command'] = self.by_type_and_type
        self.btn[3]['command'] = self.by_result
        self.btn[4]['command'] = self.by_result_and_type
        self.btn[5]['command'] = self.by_result_and_loc
        self.btn[6]['command'] = self.by_detail

    def by_loc(self):
        self.station.to_program("最近30日-按投放区域")

    def by_type(self):
        self.station.to_program("最近30日-按投放类型")

    def by_type_and_type(self):
        self.station.to_program("最近30日-按投放类型和区域")

    def by_result(self):
        self.station.to_program("最近30日-按检查结果")

    def by_result_and_type(self):
        self.station.to_program("最近30日-按检查结果和类型")

    def by_result_and_loc(self):
        self.station.to_program("最近30日-按检查结果和区域")

    def by_detail(self):
        self.station.to_program("最近30日-详细分类")


class StatisticsUserMenu(AdminMenu):
    def __init__(self, station, win, color):
        super().__init__(station, win, color, "积分信用分析")
        self.btn: List[tk.Button] = [tk.Button(self.frame) for _ in range(6)]
        self.btn_name = ["积分,信用-细致", "积分,信用-大致", "积分分布", "信用分布"]

    def conf_gui(self, color: str, n: int = 1):
        super().conf_gui(color, n)
        self.btn[0]['command'] = self.by_tiny
        self.btn[1]['command'] = self.by_large
        self.btn[2]['command'] = self.score
        self.btn[3]['command'] = self.reputation

    def by_tiny(self):
        self.station.to_program("积分信用分析-细致")

    def by_large(self):
        self.station.to_program("积分信用分析-大致")

    def score(self):
        self.station.to_program("积分分布")

    def reputation(self):
        self.station.to_program("垃圾分类信用分布")


class StatisticsPassMenu(AdminMenu):
    def __init__(self, station, win, color):
        super().__init__(station, win, color, "通过率")
        self.btn: List[tk.Button] = [tk.Button(self.frame) for _ in range(4)]
        self.btn_name = ["全局", "按类型", "按区域", "按类型和区域"]

    def conf_gui(self, color: str, n: int = 1):
        super().conf_gui(color, n)
        self.btn[0]['command'] = self.by_global
        self.btn[1]['command'] = self.by_type
        self.btn[2]['command'] = self.by_loc
        self.btn[3]['command'] = self.by_type_loc

    def by_global(self):
        self.station.to_program("通过率-全局")

    def by_type(self):
        self.station.to_program("通过率-按类型")

    def by_loc(self):
        self.station.to_program("通过率-按区域")

    def by_type_loc(self):
        self.station.to_program("通过率-按类型和区域")


all_menu = [MainMenu, CreateMenu, DeleteMenu, SearchMenu, UpdateMenu, StatisticsMenu, StatisticsTimeMenu,
            StatisticsUserMenu, StatisticsPassMenu, StatisticsDateMenu, StatisticsDate7Menu, StatisticsDate30Menu]