Added option to customize enemy image path, and customize weapons

menus/main.py

@@ -8,7 +8,7 @@ class MenuButton(Button):
         super().__init__(text, scale=(.25, .075), highlight_color=color.azure, **kwargs)
 
         for key, value in kwargs.items():
-            setattr(self, key ,value)
+            setattr(self, key, value)
 
 class Main():
     def __init__(self, pypresence_client=None) -> None:

menus/settings.py

@@ -2,8 +2,9 @@ from ursina import *
 from ursina.prefabs.slider import ThinSlider
 from ursina.prefabs.dropdown_menu import DropdownMenuButton
 from ursina.prefabs.button_group import ButtonGroup
+
 import pypresence, json, copy
-from utils.utils import FakePyPresence, Dropdown
+from utils.utils import FakePyPresence, Dropdown, FileManager
 from utils.constants import discord_presence_id, settings, settings_start_category
 from utils.preload import music_sound
 
@@ -27,6 +28,12 @@ class Settings:
 
         self.ui += [self.main, self.back, self.category_group]
 
+        self.dmg_inputs = {}
+        self.atk_speed_inputs = {}
+        self.weapon_img_paths = {}
+        self.img_path_buttons = {}
+        self.save_buttons = {}
+
         self.show(self.category)
 
     def show(self, category):
@@ -36,6 +43,9 @@ class Settings:
         if category == "Credits":
             self.credits()
             return
+        elif category == "Weapons":
+            self.weapons()
+            return
 
         y = .2
 
@@ -57,7 +67,7 @@ class Settings:
                 slider.on_value_changed = lambda slider=slider, n=name: self.update(n, int(slider.value))
                 self.ui.append(slider)
 
-            else:
+            elif type == "option":
                 menu_buttons = []
                 for opt in info['options']:
                     menu_button = DropdownMenuButton(opt)
@@ -71,11 +81,34 @@ class Settings:
                 dropdown_menu.position = (.2, y)
                 self.ui.append(dropdown_menu)
 
+            elif type == "directory_select":
+                directory_select_button = Button(text=f"Select Directory ({val})", scale_x=1.1, scale_y=0.1, text_size=.7, position = (.33, y))
+                directory_select_button.on_click = lambda btn=directory_select_button, name=name: self.select_directory(btn, name)
+
+                self.ui.append(directory_select_button)
+
             y -= .08
 
         self.apply_button = Button('Apply', parent=camera.ui, color=color.green, scale=(.15, .08), position=(.6, -.4), on_click=self.apply_changes)
         self.ui.append(self.apply_button)
 
+    def directory_selected(self, btn, name, value):
+        btn.text = f"Select Directory ({value})"
+
+        self.update(name, value)
+
+    def select_directory(self, btn, name):
+        self.dir_file_manager = FileManager(return_folders=True, z=-1)
+        self.dir_file_manager.on_submit = lambda value, btn=btn, name=name: self.directory_selected(btn, name, str(value[0]))
+
+    def image_file_selected(self, btn, name, value):
+        btn.text = f"Select File ({value})"
+        self.weapon_img_paths[name] = value
+
+    def select_image_file(self, btn, name):
+        self.file_manager = FileManager(z=-1)
+        self.file_manager.on_submit = lambda value, btn=btn, name=name: self.image_file_selected(btn, name, str(value[0]))
+
     def dropdown_update(self, n, dropdown_menu, btn):
         dropdown_menu.text = btn.text
 
@@ -120,6 +153,7 @@ class Settings:
             music_sound.stop()
 
         json.dump(self.data, open('settings.json', 'w'), indent=4)
+
         self.hide()
         self.__init__(self.rpc)
 
@@ -168,3 +202,38 @@ class Settings:
         self.credits_label = Text(text=text, parent=camera.ui, position=(0, 0), origin=(0, 0), scale=font_size, color=color.white)
         self.credits_label.type = 'credits_text'
         self.ui.append(self.credits_label)
+
+    def save_weapon(self, name):
+        dmg, attack_speed, image = self.dmg_inputs[name].text, self.atk_speed_inputs[name].text, self.weapon_img_paths[name]
+
+        self.edits["weapons"] = self.edits.get("weapons", settings["Weapons"]["default"])
+        self.edits["weapons"][name] = {"dmg": float(dmg), "atk_speed": float(attack_speed), "image": image}
+
+        self.apply_changes()
+
+    def weapons(self):
+        y = .2
+
+        for weapon_name, weapon_dict in self.data.get("weapons", settings["Weapons"]["default"]).items():
+            dmg, atk_speed, image = weapon_dict["dmg"], weapon_dict["atk_speed"], weapon_dict["image"]
+
+            self.ui.append(Text(weapon_name, parent=camera.ui, position=(-.8, y), scale=1.2))
+
+            self.ui.append(Text("DMG: ", parent=camera.ui, position=(-.6, y), scale=1.2))
+            self.dmg_inputs[weapon_name] = InputField(default_value=str(round(dmg, 2)), parent=camera.ui, position=(-.45, y - .01), scale_x=0.125, scale_y=.05)
+            self.ui.append(self.dmg_inputs[weapon_name])
+
+            self.ui.append(Text("Attack Speed: ", parent=camera.ui, position=(-.35, y), scale=1.2))
+            self.atk_speed_inputs[weapon_name] = InputField(default_value=str(round(atk_speed, 2)), parent=camera.ui, position=(-0.075, y - .01), scale_x=0.125, scale_y=.05)
+            self.ui.append(self.atk_speed_inputs[weapon_name])
+
+            self.ui.append(Text("Image Path: ", parent=camera.ui, position=(0.05, y), scale=1.2))
+            self.img_path_buttons[weapon_name] = Button(text=f"Select File ({image})", scale_x=.3, scale_y=0.05, text_size=.5, position=(0.4, y - .01))
+            self.img_path_buttons[weapon_name].on_click = lambda name=weapon_name, btn=self.img_path_buttons[weapon_name]: self.select_image_file(btn, name)
+            self.ui.append(self.img_path_buttons[weapon_name])
+
+            self.save_buttons[weapon_name] = Button(text="Save", scale_x=.1, scale_y=0.05, text_size=.7, position=(0.7, y - .01))
+            self.save_buttons[weapon_name].on_click = lambda name=weapon_name: self.save_weapon(name)
+            self.ui.append(self.save_buttons[weapon_name])
+
+            y -= 0.08