ntmemoryapi 1.7.0__tar.gz → 1.7.2__tar.gz

{ntmemoryapi-1.7.0 → ntmemoryapi-1.7.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: ntmemoryapi
-Version: 1.7.0
+Version: 1.7.2
 Summary: Simple library for Windows to manipulate process virtual memory with stelthy syscall wraps
 Author: Xenely
 Requires-Dist: psutil>=7.1.3

{ntmemoryapi-1.7.0 → ntmemoryapi-1.7.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "ntmemoryapi"
-version = "1.7.0"
+version = "1.7.2"
 description = "Simple library for Windows to manipulate process virtual memory with stelthy syscall wraps"
 authors = [
     {name = "Xenely"}

{ntmemoryapi-1.7.0 → ntmemoryapi-1.7.2}/src/ntmemoryapi/__init__.py

@@ -283,14 +283,14 @@ class Process:
         # Open process by it's ID or it's name
         match name_or_pid:
 
-            case pid if type(name_or_pid) is int:
-                self.handle, self.name, self.pid = self.__init_with_pid(pid, access)
+            case int():
+                self.handle, self.pid, self.name = self.__init_with_pid(name_or_pid, access)
 
-            case name if type(name_or_pid) is str:
-                self.handle, self.name, self.pid = self.__init_with_name(name, access)
+            case str():
+                self.handle, self.pid, self.name = self.__init_with_name(name_or_pid, access)
 
             case _:
-                raise Exception("Invalid `pid_or_name` argument value, have to be `int` or `str` type")
+                raise Exception("Invalid `name_or_pid` argument value, have to be `int` or `str` type")
 
         # Try create file at temp directory to load SIMD KMP .dll (Module to blazingly fast pattern scaning)
         try:
@@ -746,11 +746,11 @@ class Process:
     # ==-------------------------------------------------------------------== #
     # Private methods                                                         #
     # ==-------------------------------------------------------------------== #
-    def __init_with_pid(self, pid: int, access: int, process_name: str | None = None) -> int:
+    def __init_with_pid(self, pid: int, access: int, listed_processes: list[dict[str, int | str]] | None = None) -> tuple[int, int, str]:
         """Open process handle by it's ID with desired access."""
 
-        # Iterate all of the processes if name not defined
-        for process in list_processes():
+        # Iterate all of the processes
+        for process in listed_processes if listed_processes is not None else list_processes():
 
             # If process have a reqired name
             if process["id"] == pid:
@@ -778,16 +778,16 @@ class Process:
             else:
                 raise Exception("NtOpenProcess failed with status: 0x%s" % hex(result)[2:].upper())
 
-        return handle.value, process_name, pid
+        return handle.value, pid, process_name
 
-    def __init_with_name(self, name: str, access: int, current_username_only: bool) -> int:
+    def __init_with_name(self, name: str, access: int) -> tuple[int, int, str]:
         """Open process hanle by it's name with desired access."""
 
         # Iterate all of the processes using snapshot
-        for process in list_processes():
+        for process in (listed_processes := list_processes()):
 
             # If process have a reqired name
             if process["name"].lower() == name.strip().lower():
-                return self.__init_with_pid(process["id"], access, current_username_only, process["name"])
+                return self.__init_with_pid(process["id"], access, listed_processes)
 
         raise Exception("Process with `%s` name not found" % name)