mods_panel.py

import tkinter as tk
from functools import partial

import settings as st
import view_helper as vh
import file_helper as fh


class ModsPanel:

    def __init__(self, parent, cont):
        """Panel for armor and resistance mods."""
        self.frame = tk.Frame(parent, bg=st.base_bg)
        self.cont = cont

        norm_font = st.norm_font

        armor_header_label = tk.Label(self.frame, text="Armor (0=weak)", font="Arial 12", fg='#221CD9', bg=st.base_bg)

        armor_labels = ['slash', 'pierce', 'bludge', 'cold', 'fire', 'acid', 'electric']
        self.armor_entries = vh.make_float_entry(self.frame, armor_labels)

        resist_header_label = tk.Label(self.frame, text="Resist (0=strong)", font="Arial 12", fg='#221CD9', bg=st.base_bg)

        resist_labels = ['slash', 'pierce', 'bludge', 'cold', 'fire', 'acid', 'electric', 'nether']
        self.resist_entries = vh.make_float_entry(self.frame, resist_labels)

        set_button = tk.Button(self.frame, text="Set", bg=st.button_bg, command=self.set_mods)
        batch_button = tk.Button(self.frame, text="Run Batch", command=partial(self.cont.run_sql_batch, self.set_mods))

        # layout

        r = 0
        c = 0

        armor_header_label.grid(row=r, column=c)
        r += 1

        for name, entry in self.armor_entries.items():
            label = tk.Label(self.frame, text=name, font=norm_font, bg=st.base_bg)
            label.grid(row=r, column=c)
            entry.grid(row=r, column=c + 1)
            r += 1

        resist_header_label.grid(row=r, column=c)
        r += 1

        for name, entry in self.resist_entries.items():
            label = tk.Label(self.frame, text=name, font=norm_font, bg=st.base_bg)
            label.grid(row=r, column=c)
            entry.grid(row=r, column=c + 1)
            r += 1

        set_button.grid(row=r, column=c, padx=5, pady=5, sticky="ew")
        r += 1
        batch_button.grid(row=r, column=c, padx=5, pady=5, sticky="ew")

    def show_mods(self):
        if self.cont.sql_commands is not None:

            # clear existing
            for name, entry in self.armor_entries.items():
                entry.delete(0, tk.END)
            for name, entry in self.resist_entries.items():
                entry.delete(0, tk.END)

            armor_mods = fh.get_armor_mods(self.cont.sql_commands)
            resist_mods = fh.get_resist_mods(self.cont.sql_commands)

            for name, entry in self.armor_entries.items():
                if name in armor_mods.keys():
                    result = armor_mods.get(name)
                    if result is not None:
                        entry.insert(0, result)

            for name, entry in self.resist_entries.items():
                if name in resist_mods.keys():
                    result = resist_mods.get(name)
                    if result is not None:
                        entry.insert(0, result)

    def set_mods(self):

        if self.cont.sql_commands is not None:
            armor_dict = {
                "slash": (13, "/* ArmorModVsSlash */"),
                "pierce": (14, "/* ArmorModVsPierce */"),
                "bludge": (15, "/* ArmorModVsBludgeon */"),
                "cold": (16, "/* ArmorModVsCold */"),
                "fire": (17, "/* ArmorModVsFire */"),
                "acid": (18, "/* ArmorModVsAcid */"),
                "electric": (19, "/* ArmorModVsElectric */")
            }
            self.cont.set_properties(armor_dict, self.armor_entries, 'float')

            resist_dict = {
                "slash": (64, "/* ResistSlash */"),
                "pierce": (65, "/* ResistPierce */"),
                "bludge": (66, "/* ResistBludgeon */"),
                "cold": (68, "/* ResistCold */"),
                "fire": (67, "/* ResistFire */"),
                "acid": (69, "/* ResistAcid */"),
                "electric": (70, "/* ResistElectric */"),
                "nether": (166, "/* ResistNether */")
            }
            self.cont.set_properties(resist_dict, self.resist_entries, 'float')

        else:
            self.cont.file_warning()