trainerbase 3.13.0__tar.gz → 3.14.1__tar.gz

{trainerbase-3.13.0 → trainerbase-3.14.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: trainerbase
-Version: 3.13.0
+Version: 3.14.1
 Summary: Framework for creating game trainers/mods
 Home-page: https://gitlab.com/python-trainers/trainer-base
 License: MIT

{trainerbase-3.13.0 → trainerbase-3.14.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "trainerbase"
-version = "3.13.0"
+version = "3.14.1"
 description = "Framework for creating game trainers/mods"
 authors = ["v01d <v01d@v01d.ru>", "doomayka <zd@doomayka.ru>"]
 license = "MIT"

{trainerbase-3.13.0 → trainerbase-3.14.1}/trainerbase/codeinjection.py

@@ -8,7 +8,7 @@ from typing import ClassVar, Final, Self, override
 
 from pymem.exception import MemoryWriteError
 
-from trainerbase.memory import ARCH, ConvertibleToAddress, ensure_address, pm
+from trainerbase.memory import ARCH, ConvertibleToAddress, allocate, ensure_address, pm
 
 
 FASM_EXECUTABLE_PATH: Final[Path] = Path(__file__).resolve().parent.parent / "trainerbase_vendor" / "FASM.EXE"
@@ -67,6 +67,7 @@ class AllocatingCodeInjection(AbstractCodeInjection):  # pylint: disable=too-man
         self,
         address: ConvertibleToAddress,
         code_to_inject: Code,
+        *,
         original_code_length: int = 0,
         new_memory_size: int = 1024,
         is_long_x64_jump_needed: bool | None = None,
@@ -92,7 +93,7 @@ class AllocatingCodeInjection(AbstractCodeInjection):  # pylint: disable=too-man
     def inject(self):
         self.injection_point = self.address.resolve()
 
-        self.new_memory_address: int = pm.allocate(self.new_memory_size)
+        self.new_memory_address: int = allocate(self.new_memory_size)
 
         self.original_code = pm.read_bytes(self.injection_point, self.original_code_length)
 

{trainerbase-3.13.0 → trainerbase-3.14.1}/trainerbase/common/helpers.py

@@ -1,4 +1,5 @@
 from collections.abc import Iterator, Sequence
+from typing import overload
 from uuid import uuid4
 
 from trainerbase.gameobject import AbstractReadableObject, GameObject
@@ -30,11 +31,17 @@ class ASMArrayManager(Sequence):
     def __len__(self) -> int:
         return self.array_length
 
-    def __getitem__(self, index: int) -> Address:  # type: ignore TODO: Support slices
-        element_address = self.get_pointer_to_element(index)
-        target_address = Address(element_address, self.offsets, self.add)
+    @overload
+    def __getitem__(self, index: int) -> Address: ...
 
-        return Address(target_address.resolve())
+    @overload
+    def __getitem__(self, index: slice) -> list[Address]: ...
+
+    def __getitem__(self, index: int | slice) -> Address | list[Address]:
+        if isinstance(index, int):
+            return self.get_address_of_element(index)
+
+        return [self.get_address_of_element(i) for i in range(index.start, index.stop, index.step)]
 
     def __iter__(self) -> Iterator[Address]:
         for index in range(len(self)):
@@ -43,6 +50,12 @@ class ASMArrayManager(Sequence):
     def get_pointer_to_element(self, index: int) -> int:
         return self.array_address + self.element_size * index
 
+    def get_address_of_element(self, index: int) -> Address:
+        element_address = self.get_pointer_to_element(index)
+        target_address = Address(element_address, self.offsets, self.add)
+
+        return Address(target_address.resolve())
+
     def generate_asm_append_code(self, from_register: str) -> str:
         array_address_register = "rbx" if pm.is_64_bit else "ebx"
         array_index_register = "rcx" if pm.is_64_bit else "ecx"

{trainerbase-3.13.0 → trainerbase-3.14.1}/trainerbase/gui/helpers.py

@@ -18,10 +18,9 @@ def simple_trainerbase_menu(window_title: str, width: int, height: int):
             dpg.create_context()
             dpg.create_viewport(
                 title=window_title,
-                min_width=width,
-                min_height=height,
                 width=width,
                 height=height,
+                resizable=False,
             )
             dpg.setup_dearpygui()
 
@@ -33,9 +32,11 @@ def simple_trainerbase_menu(window_title: str, width: int, height: int):
                 no_move=True,
                 no_title_bar=True,
                 horizontal_scrollbar=True,
+                autosize=True,
             ):
                 initializer()
 
+            dpg.set_primary_window("menu", value=True)
             dpg.show_viewport()
 
             on_initialized()