ptb/tkinter_ui/prefer.py

import tkinter as tk
from tkinter import ttk
from utils.config import config


class PreferUI:
    def init_ui(self, root=None):
        """
        Init prefer UI
        """
        origin_type_prefer = [
            item.lower()
            for item in config.get(
                "Settings",
                "origin_type_prefer",
                fallback="hotel,multicast,subscribe,online_search",
            ).split(",")
        ]
        config_options = [
            {"label_text": f"结果来源优先{i+1}:", "combo_box_value": i}
            for i in range(len(origin_type_prefer))
        ]
        self.origin_type_prefer_options = []
        for config_option in config_options:
            option = ConfigOption(root, **config_option)
            option.combo_box.bind(
                "<<ComboboxSelected>>",
                option.update_select,
            )
            option.entry.bind("<KeyRelease>", option.update_input)
            self.origin_type_prefer_options.append(option)

        frame_prefer_ipv_type = tk.Frame(root)
        frame_prefer_ipv_type.pack(fill=tk.X)
        self.prefer_ipv_type_label = tk.Label(
            frame_prefer_ipv_type, text="结果协议优先:", width=12
        )
        self.prefer_ipv_type_label.pack(side=tk.LEFT, padx=4, pady=8)
        self.prefer_ipv_type_combo = ttk.Combobox(frame_prefer_ipv_type)
        self.prefer_ipv_type_combo.pack(side=tk.LEFT, padx=4, pady=8)
        self.prefer_ipv_type_combo["values"] = ("IPv4", "IPv6", "自动")
        ipv_type_prefer = config.get(
            "Settings", "ipv_type_prefer", fallback="IPv4"
        ).lower()
        if ipv_type_prefer == "ipv4":
            self.prefer_ipv_type_combo.current(0)
        elif ipv_type_prefer == "ipv6":
            self.prefer_ipv_type_combo.current(1)
        else:
            self.prefer_ipv_type_combo.current(2)
        self.prefer_ipv_type_combo.bind(
            "<<ComboboxSelected>>", self.update_ipv_type_prefer
        )
        self.ipv_type_input = []
        for ipv_type in ["ipv4", "ipv6"]:
            input = IpvNumInput(root, ipv_type)
            input.entry.bind("<KeyRelease>", input.update_input)
            self.ipv_type_input.append(input)

    def update_ipv_type_prefer(self, event):
        config.set(
            "Settings",
            "ipv_type_prefer",
            self.prefer_ipv_type_combo.get(),
        )

    def change_entry_state(self, state):
        for option in self.origin_type_prefer_options:
            option.change_state(state)
        self.prefer_ipv_type_combo.config(state=state)
        for input in self.ipv_type_input:
            input.change_state(state)


class IpvNumInput:
    def __init__(self, master, ipv_type):
        self.master = master
        self.ipv_type = ipv_type
        self.frame = tk.Frame(master)
        self.frame.pack(fill=tk.X)
        self.frame_column1 = tk.Frame(self.frame)
        self.frame_column1.pack(side=tk.LEFT, fill=tk.Y)
        self.frame_column2 = tk.Frame(self.frame)
        self.frame_column2.pack(side=tk.RIGHT, fill=tk.Y)

        ipv_type_text = "IPv4" if ipv_type == "ipv4" else "IPv6"
        self.entry_label = tk.Label(
            self.frame_column1, text=f"{ipv_type_text}数量:", width=12
        )
        self.entry_label.pack(side=tk.LEFT, padx=4, pady=8)

        self.entry = tk.Entry(self.frame_column1)
        self.entry.insert(
            0,
            config.getint(
                "Settings",
                f"{ipv_type}_num",
                fallback=15,
            ),
        )
        self.entry.pack(side=tk.LEFT, padx=4, pady=8)

    def update_input(self, event):
        config.set(
            "Settings",
            f"{ self.ipv_type}_num",
            self.entry.get(),
        )

    def change_state(self, state):
        self.entry.config(state=state)


class ConfigOption:
    def __init__(self, master, label_text, combo_box_value):
        self.master = master
        self.label_text = label_text
        self.combo_box_value = combo_box_value

        self.frame = tk.Frame(master)
        self.frame.pack(fill=tk.X)

        self.column1 = tk.Frame(self.frame)
        self.column1.pack(side=tk.LEFT, fill=tk.Y)

        self.column2 = tk.Frame(self.frame)
        self.column2.pack(side=tk.RIGHT, fill=tk.Y)

        self.label = tk.Label(self.column1, text=label_text, width=12)
        self.label.pack(side=tk.LEFT, padx=4, pady=8)

        self.combo_box = ttk.Combobox(self.column1)
        self.origin_type_prefer_obj = {
            "酒店源": "hotel",
            "组播源": "multicast",
            "订阅源": "subscribe",
            "关键字搜索": "online_search",
        }
        combo_box_values_name = list(self.origin_type_prefer_obj.keys())
        self.combo_box["values"] = combo_box_values_name
        self.combo_box.pack(side=tk.LEFT, padx=4, pady=8)
        self.combo_box.current(combo_box_value)

        self.entry_label = tk.Label(self.column2, text="数量:", width=12)
        self.entry_label.pack(side=tk.LEFT, padx=4, pady=8)

        self.entry = tk.Entry(self.column2)
        self.entry.insert(
            0,
            config.getint(
                "Settings",
                f"{self.origin_type_prefer_obj[self.combo_box.get()]}_num",
                fallback=10,
            ),
        )
        self.entry.pack(side=tk.LEFT, padx=4, pady=8)

    def update_select(self, key):
        origin_type_prefer_list = [
            item.lower()
            for item in config.get(
                "Settings",
                "origin_type_prefer",
                fallback="hotel,multicast,subscribe,online_search",
            ).split(",")
        ]
        origin_type_prefer_list[self.combo_box_value] = self.origin_type_prefer_obj[
            self.combo_box.get()
        ]
        config.set(
            "Settings",
            "origin_type_prefer",
            (",").join(origin_type_prefer_list),
        )

    def update_input(self, event):
        config.set(
            "Settings",
            f"{ self.origin_type_prefer_obj[self.combo_box.get()]}_num",
            self.entry.get(),
        )

    def change_state(self, state):
        self.combo_box.config(state=state)
        self.entry.config(state=state)