Include memory information

pyarchitecture/__init__.py

@@ -4,7 +4,7 @@
 import time
 from typing import Any, Dict
 
-from pyarchitecture import cpu, disks, gpu
+from pyarchitecture import cpu, disks, gpu, memory
 
 version = "0.0.0-a0"
 
@@ -20,6 +20,7 @@ def all_components() -> Dict[str, Any]:
         "Disks": disks.get_all_disks(),
         "CPU": cpu.get_cpu_info(),
         "GPU": gpu.get_gpu_names(),
+        "Memory": memory.get_memory_info(),
     }
 
 
@@ -107,10 +108,9 @@ def commandline() -> None:
                 data["GPU"] = gpu.get_gpu_names()
             if disk_info:
                 data["Disks"] = disks.get_all_disks()
-        import io
 
         with open(filename, "w") as json_file:
-            json.dump(data, json_file, indent=2)  # type: io.TextIOBase
+            json.dump(data, json_file, indent=2)
         print(f"Architecture information has been stored in {filename!r}")
         sys.exit(0)
     else:

pyarchitecture/cpu/__init__.py

@@ -2,6 +2,8 @@
 import os
 from typing import Dict
 
+import psutil
+
 from pyarchitecture import models
 from pyarchitecture.cpu import main
 
@@ -35,9 +37,8 @@ def get_cpu_info(cpu_lib: str | os.PathLike = None) -> Dict[str, int | str]:
         Returns CPU name.
     """
     cpu_name = main.get_name(_get_cpu_lib(cpu_lib))
-    cpu_count = os.cpu_count()
     return {
         "name": cpu_name,
-        "logical_cores": cpu_count,
-        "physical_cores": int(cpu_count / 2) if cpu_count >= 2 else 1,
+        "logical_cores": psutil.cpu_count(logical=True),
+        "physical_cores": psutil.cpu_count(logical=False),
     }

pyarchitecture/disks/macOS.py

@@ -5,7 +5,7 @@
 from collections import defaultdict
 from typing import Dict, List
 
-from pyarchitecture.disks import squire
+from pyarchitecture import squire
 
 LOGGER = logging.getLogger(__name__)
 

pyarchitecture/disks/windows.py

@@ -6,7 +6,7 @@
 import subprocess
 from typing import Dict, List, Tuple
 
-from pyarchitecture.disks import squire
+from pyarchitecture import squire
 
 LOGGER = logging.getLogger(__name__)
 

pyarchitecture/memory.py

@@ -0,0 +1,25 @@
+import psutil
+
+from pyarchitecture import squire
+
+
+def get_memory_info(raw: bool = False) -> dict[str, int | str]:
+    """Get memory information for the host system.
+
+    Returns:
+        Dict[str, str]:
+        Returns memory information.
+    """
+    if raw:
+        return {
+            "total": psutil.virtual_memory().total,
+            "available": psutil.virtual_memory().available,
+            "used": psutil.virtual_memory().used,
+            "free": psutil.virtual_memory().free,
+        }
+    return {
+        "total": squire.size_converter(psutil.virtual_memory().total),
+        "available": squire.size_converter(psutil.virtual_memory().available),
+        "used": squire.size_converter(psutil.virtual_memory().used),
+        "free": squire.size_converter(psutil.virtual_memory().free),
+    }

pyproject.toml

@@ -14,6 +14,9 @@ classifiers = [
 ]
 keywords = ["physical-drives", "PyArchitecture"]
 requires-python = ">=3.10"
+dependencies = [
+    "psutil>=6.1.1"
+]
 
 [tool.setuptools]
 packages = [