ntmemoryapi 2.3.0__tar.gz → 2.4.0__tar.gz

{ntmemoryapi-2.3.0 → ntmemoryapi-2.4.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: ntmemoryapi
-Version: 2.3.0
+Version: 2.4.0
 Summary: Add your description here
 Requires-Dist: psutil>=7.2.2
 Requires-Python: >=3.13

{ntmemoryapi-2.3.0 → ntmemoryapi-2.4.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "ntmemoryapi"
-version = "2.3.0"
+version = "2.4.0"
 description = "Add your description here"
 readme = "README.md"
 requires-python = ">=3.13"

{ntmemoryapi-2.3.0 → ntmemoryapi-2.4.0}/src/ntmemoryapi/__init__.py

@@ -215,7 +215,7 @@ class Process:
             if not self.is_64bit():
                 raise RuntimeError("Unable to list modules for not 64 bit process using `%s` method, use another one method instead")
 
-            # If procewss PEB retrieve failed
+            # If process PEB retrieve failed
             if not (process_peb := self.get_information().peb):
                 raise RuntimeError("Process `%s` PEB address is null-value" % self.name)
 
@@ -380,11 +380,15 @@ class Process:
 
         return process_basic_information
 
-    def get_module(self, name: str) -> structs.MODULEENTRY32:
+    def get_module(self, name: str, method: typing.Literal["pebwalk", "snapshot"] = "pebwalk") -> structs.MODULEENTRY32:
         """Get process module information."""
 
+        # If method is not allowed
+        if method.lower() not in [item.lower() for item in allowed_methods] if (allowed_methods := typing.get_args(self.list_modules.__annotations__["method"])) else True:
+            raise ValueError("Method literal is invalid, expected one of: `%s`" % ", ".join(allowed_methods))
+
         # Process modules enumeration
-        for module in self.list_modules():
+        for module in self.list_modules(method):
 
             # If module have a required name
             if module.name.lower() == name.strip().lower():