trainerbase 4.6.5__tar.gz → 4.7.0__tar.gz

{trainerbase-4.6.5 → trainerbase-4.7.0}/PKG-INFO

@@ -1,9 +1,9 @@
 Metadata-Version: 2.4
 Name: trainerbase
-Version: 4.6.5
+Version: 4.7.0
 Summary: Framework for creating game trainers/mods
 Author: v01d, doomayka
-Author-email: v01d <v01d@v01d.ru>, doomayka <zd@doomayka.ru>
+Author-email: v01d <v01d@v01d.ru>, doomayka <doomaykaka@gmail.com>
 License-Expression: MIT
 Classifier: Programming Language :: Python :: Implementation :: CPython
 Classifier: Programming Language :: Python :: 3.14

{trainerbase-4.6.5 → trainerbase-4.7.0}/pyproject.toml

@@ -1,11 +1,11 @@
 [project]
 name = "trainerbase"
-version = "4.6.5"
+version = "4.7.0"
 description = "Framework for creating game trainers/mods"
 license = "MIT"
 authors = [
     { name = "v01d", email = "v01d@v01d.ru" },
-    { name = "doomayka", email = "zd@doomayka.ru" },
+    { name = "doomayka", email = "doomaykaka@gmail.com" },
 ]
 requires-python = ">=3.14"
 readme = "README.md"

{trainerbase-4.6.5 → trainerbase-4.7.0}/trainerbase/gui/objects.py

@@ -1,5 +1,6 @@
 from collections.abc import Callable
 from contextlib import suppress
+from enum import Enum
 from typing import Any, ClassVar, Self, override
 from uuid import uuid4
 
@@ -43,6 +44,7 @@ class GameObjectUI(AbstractUIComponent):
         before_set: Callable | None = None,
         tts_on_hotkey: bool = True,
         setter_input_width: int = 220,
+        enum: type[Enum] | None = None,
     ):
         if not gameobject.is_tracked:
             freeze_hotkey = None
@@ -56,6 +58,7 @@ class GameObjectUI(AbstractUIComponent):
         self.before_set = before_set
         self.tts_on_hotkey = tts_on_hotkey
         self.setter_input_width = setter_input_width
+        self.enum = enum
 
         dpg_tag = f"{self.DPG_TAG_PREFIX}{uuid4()}"
         self.dpg_tag_frozen = f"{dpg_tag}{self.DPG_TAG_POSTFIX_IS_FROZEN}"
@@ -89,9 +92,15 @@ class GameObjectUI(AbstractUIComponent):
         default_kwargs = {
             "tag": self.dpg_tag_setter,
             "width": self.setter_input_width,
-            "default_value": self.default_setter_input_value,
+            "default_value": self.default_setter_input_value.name
+            if isinstance(self.default_setter_input_value, Enum)
+            else self.default_setter_input_value,
         }
 
+        if self.enum is not None:
+            dpg.add_combo(list(self.enum.__members__.keys()), **default_kwargs)
+            return
+
         if self.gameobject.value_range is not None:
             min_value, max_value = self.gameobject.value_range
             default_kwargs["min_clamped"] = True
@@ -141,7 +150,12 @@ class GameObjectUI(AbstractUIComponent):
 
     def on_value_set(self):
         raw_new_value = dpg.get_value(self.dpg_tag_setter)
+        if self.enum is not None:
+            raw_new_value = self.enum[raw_new_value]
+
         new_value = raw_new_value if self.before_set is None else self.before_set(raw_new_value)
+        if isinstance(new_value, Enum):
+            new_value = new_value.value
 
         if self.gameobject.frozen is None:
             with suppress(MemoryWriteError, ValueError):
@@ -164,6 +178,13 @@ def update_displayed_objects():
         try:
             new_value = game_object_ui.gameobject.value
         except MemoryReadError:
-            new_value = "<Unresolved>"
+            dpg.set_value(game_object_ui.dpg_tag_getter, "<Unresolved>")
+            continue
+
+        if game_object_ui.enum is not None:
+            try:
+                new_value = game_object_ui.enum(new_value).name
+            except ValueError:
+                new_value = "<InvalidEnumValue>"
 
         dpg.set_value(game_object_ui.dpg_tag_getter, new_value)