Merge pull request #2 from thomrommens/patch-1

Remove resources lib from module import - make conditional per OS
thevickypedia · Jul 29, 2023 · 1f4ece2 · 1f4ece2
2 parents ad577d4 + 664e846
commit 1f4ece2
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/stressinjector/memory.py b/stressinjector/memory.py
@@ -1,6 +1,5 @@
 import logging
 import math
-import resource
 import time
 from typing import Union
 
@@ -106,10 +105,13 @@ def _memory_util_check(cls) -> Union[int, float]:
                 `memory_info <https://psutil.readthedocs.io/en/latest/#psutil.Process.memory_info>`__
         """
         if settings.os == operating_system.macOS:
+            import resource
             return resource.getrusage(resource.RUSAGE_SELF).ru_maxrss
         if settings.os == operating_system.linux:
+            import resource
             return resource.getrusage(resource.RUSAGE_SELF).ru_maxrss * 1e+3
         if settings.os == operating_system.windows:
+            import psutil
             return psutil.Process(settings.pid).memory_info().peak_wset
 
     def _run(self) -> None: