iptv-py/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
        """
        frame_location_isp = tk.Frame(root)
        frame_location_isp.pack(fill=tk.X)
        frame_location_isp_column1 = tk.Frame(frame_location_isp)
        frame_location_isp_column1.pack(side=tk.LEFT, fill=tk.Y)
        frame_location_isp_column2 = tk.Frame(frame_location_isp)
        frame_location_isp_column2.pack(side=tk.RIGHT, fill=tk.Y)

        self.location_label = tk.Label(
            frame_location_isp_column1, text="归属地:", width=12
        )
        self.location_label.pack(side=tk.LEFT, padx=4, pady=8)
        self.location_entry = tk.Entry(frame_location_isp_column1, width=24)
        self.location_entry.pack(side=tk.LEFT, padx=4, pady=8)
        self.location_entry.insert(0, ",".join(config.location))
        self.location_entry.bind("<KeyRelease>", self.update_location)

        self.isp_label = tk.Label(
            frame_location_isp_column2, text="运营商:", width=12
        )
        self.isp_label.pack(side=tk.LEFT, padx=4, pady=8)
        self.isp_entry = tk.Entry(frame_location_isp_column2, width=24)
        self.isp_entry.pack(side=tk.LEFT, padx=4, pady=8)
        self.isp_entry.insert(0, ",".join(config.isp))
        self.isp_entry.bind("<KeyRelease>", self.update_isp)

        config_options = [
            {"label_text": f"结果来源优先{i + 1}:", "combo_box_value": value}
            for i, value in enumerate(self.get_origin_type_prefer_index(config.origin_type_prefer))
        ]
        self.origin_type_prefer_options = []
        for i, config_option in enumerate(config_options):
            option = ConfigOption(root, **config_option)
            option.combo_box.bind(
                "<<ComboboxSelected>>",
                lambda event, opt=option, index=i: opt.update_select(event, index),
            )
            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, width=22)
        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.ipv_type_prefer
        if ipv_type_prefer:
            first_ipv_type_prefer = ipv_type_prefer[0]
            prefer_map = {"ipv4": 0, "ipv6": 1, "自动": 2, "auto": 2}
            if first_ipv_type_prefer in prefer_map:
                self.prefer_ipv_type_combo.current(prefer_map[first_ipv_type_prefer])
        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)

        frame_prefer_open_supply = tk.Frame(root)
        frame_prefer_open_supply.pack(fill=tk.X)
        self.open_supply_label = tk.Label(
            frame_prefer_open_supply, text="开启补偿模式:", width=12
        )
        self.open_supply_label.pack(side=tk.LEFT, padx=4, pady=8)
        self.open_supply_var = tk.BooleanVar(value=config.open_supply)
        self.open_supply_checkbutton = ttk.Checkbutton(
            frame_prefer_open_supply,
            variable=self.open_supply_var,
            onvalue=True,
            offvalue=False,
            command=self.update_open_supply,
        )
        self.open_supply_checkbutton.pack(side=tk.LEFT, padx=4, pady=8)

    def get_origin_type_prefer_index(self, origin_type_prefer):
        index_list = [None, None, None, None, None]
        origin_type_prefer_obj = {
            "local": 0,
            "hotel": 1,
            "multicast": 2,
            "subscribe": 3,
            "online_search": 4,
        }
        for i, item in enumerate(origin_type_prefer):
            index_list[i] = origin_type_prefer_obj[item]
        return index_list

    def update_location(self, event):
        config.set(
            "Settings",
            "location",
            self.location_entry.get(),
        )

    def update_isp(self, event):
        config.set(
            "Settings",
            "isp",
            self.isp_entry.get(),
        )

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

    def update_open_supply(self):
        config.set("Settings", "open_supply", str(self.open_supply_var.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)
        self.location_entry.config(state=state)
        self.isp_entry.config(state=state)
        self.open_supply_checkbutton.config(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, width=24)
        self.entry.insert(0, config.ipv_limit[ipv_type])
        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, width=22)
        self.origin_type_prefer_obj = {
            "本地源": "local",
            "酒店源": "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)
        origin_type_prefer = config.origin_type_prefer
        if not origin_type_prefer or combo_box_value is None:
            self.combo_box.current(None)
        else:
            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, width=24)
        if origin_type_prefer and combo_box_value is not None:
            self.entry.insert(
                0,
                config.source_limits[self.origin_type_prefer_obj[self.combo_box.get()]],
            )
        self.entry.pack(side=tk.LEFT, padx=4, pady=8)

    def update_select(self, event, index):
        origin_type_prefer_list = [''] * 5
        prefer_list = [
            origin.strip().lower()
            for origin in config.get(
                "Settings",
                "origin_type_prefer",
                fallback="",
            ).split(",")
        ]
        for i, value in enumerate(prefer_list):
            origin_type_prefer_list[i] = value
        select_value = self.origin_type_prefer_obj[
            self.combo_box.get()
        ]
        origin_type_prefer_list[index] = select_value
        self.entry.delete(0, tk.END)
        self.entry.insert(
            0,
            config.source_limits[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)