trainerbase 3.12.1__tar.gz → 3.13.0__tar.gz

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

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

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

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

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

@@ -4,7 +4,7 @@ from pathlib import Path
 from subprocess import DEVNULL
 from subprocess import run as run_executable
 from tempfile import TemporaryDirectory
-from typing import ClassVar, Final, Never, Self, override
+from typing import ClassVar, Final, Self, override
 
 from pymem.exception import MemoryWriteError
 
@@ -69,21 +69,16 @@ class AllocatingCodeInjection(AbstractCodeInjection):  # pylint: disable=too-man
         code_to_inject: Code,
         original_code_length: int = 0,
         new_memory_size: int = 1024,
-        is_long_x64_jump_needed: Never = Never,
+        is_long_x64_jump_needed: bool | None = None,
     ):
-        if is_long_x64_jump_needed is not Never:
-            raise Exception(  # pylint: disable=broad-exception-raised
-                "Since TrainerBase 3.4.0 is_long_x64_jump_needed has been deprecated."
-                " Now there is not need to manually pop and push rax. That's why you must"
-                " rewrite injections based on manual pop rax, push rax and then remove is_long_x64_jump_needed"
-                " argument"
-            )
+        if is_long_x64_jump_needed is None:
+            is_long_x64_jump_needed = pm.is_64_bit
 
+        self.is_long_x64_jump_needed = is_long_x64_jump_needed
         self.address = ensure_address(address)
-
         self.code_to_inject = compile_asm(code_to_inject)
 
-        if pm.is_64_bit:
+        if self.is_long_x64_jump_needed:
             self.code_to_inject = compile_asm("pop rax") + self.code_to_inject + compile_asm("push rax")
 
         self.original_code: bytes = b""
@@ -101,7 +96,7 @@ class AllocatingCodeInjection(AbstractCodeInjection):  # pylint: disable=too-man
 
         self.original_code = pm.read_bytes(self.injection_point, self.original_code_length)
 
-        if pm.is_64_bit:
+        if self.is_long_x64_jump_needed:
             jump_to_new_mem = compile_asm(get_long_x64_jump_to_new_mem_code(self.new_memory_address))
             jumper_length = len(jump_to_new_mem)
             jump_back_address = self.injection_point + jumper_length - 1

{trainerbase-3.12.1 → trainerbase-3.13.0}/trainerbase/process.py

@@ -24,7 +24,9 @@ def attach_to_process() -> Pymem:
         for process_name in cycle(process_names):
             try:
                 pm = Pymem(
-                    process_name, exact_match=pymem_config["exact_match"], ignore_case=pymem_config["ignore_case"]
+                    process_name,
+                    exact_match=pymem_config["exact_match"],
+                    ignore_case=pymem_config["ignore_case"],
                 )
             except ProcessNotFound:
                 continue