pyhw 0.14.1__tar.gz → 0.14.3__tar.gz

{pyhw-0.14.1/src/pyhw.egg-info → pyhw-0.14.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pyhw
-Version: 0.14.1
+Version: 0.14.3
 Summary: PyHw, a neofetch-like command line tool for fetching system information but written mostly in python.
 Author: Xiao Ran
 Maintainer-email: Xiao Ran <xiaoran.007@icloud.com>

{pyhw-0.14.1 → pyhw-0.14.3}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "pyhw"
-version = "0.14.1"
+version = "0.14.3"
 description = "PyHw, a neofetch-like command line tool for fetching system information but written mostly in python."
 keywords = ["neofetch", "system information", "command line tool", "python", "hardware information", "fastfetch", "fetching"]
 authors = [

{pyhw-0.14.1 → pyhw-0.14.3}/src/pyhw/__init__.py

@@ -1,2 +1,2 @@
-__version__ = '0.14.1'
+__version__ = '0.14.3'
 __author__ = 'xiaoran007'

{pyhw-0.14.1 → pyhw-0.14.3}/src/pyhw/__main__.py

@@ -154,13 +154,21 @@ def main():
         multiprocessing.Process(target=detect_os, args=(debug_info, current_os, result_dict)),
         multiprocessing.Process(target=detect_cpu, args=(debug_info, current_os, result_dict)),
         multiprocessing.Process(target=detect_memory, args=(debug_info, current_os, result_dict)),
-        multiprocessing.Process(target=detect_pci_related, args=(debug_info, current_os, result_dict)),
     ]
 
+    if current_os == "macos":
+        processes.append(multiprocessing.Process(target=detect_gpu, args=(debug_info, current_os, result_dict)))
+        processes.append(multiprocessing.Process(target=detect_nic, args=(debug_info, current_os, result_dict)))
+        processes.append(multiprocessing.Process(target=detect_npu, args=(debug_info, current_os, result_dict)))
+    else:
+        multiprocessing.Process(target=detect_pci_related, args=(debug_info, current_os, result_dict)),
+
     start_time = time.time()
     for process in processes:
         process.start()
 
+    detection_time = time.time() - start_time
+
     for process in processes[1:]:
         process.join()
 
@@ -179,6 +187,8 @@ def main():
         for func_name, elapsed in debug_dict.items():
             detection_name = func_name.replace("detect_", "")
             print(f"{detection_name:<10}: {elapsed:.4f} s")
+        print("-" * 50)
+        print(f"Total create time: {detection_time:.4f} s")
         print("-"*50)
         print(f"Total execution time: {total_time:.4f} s")
         print("="*50)

{pyhw-0.14.1 → pyhw-0.14.3}/src/pyhw/backend/gpu/macos.py

@@ -1,5 +1,6 @@
 from .gpuInfo import GPUInfo
 from ...pyhwUtil import getArch
+from ..cpu import CPUDetect
 import json
 import subprocess
 import ctypes
@@ -19,22 +20,25 @@ class GPUDetectMacOS:
         return self.__gpuInfo
 
     def __getGPUAppleSilicon(self):
-        gpus = list()
-        try:
-            gpu_info_dict = json.loads(subprocess.check_output(["system_profiler", "SPDisplaysDataType", "-json"]))
-            if 'SPDisplaysDataType' in gpu_info_dict:
-                gpus = gpu_info_dict['SPDisplaysDataType']
-                self.__gpuInfo.number = len(gpus)
-            else:
-                pass
-        except Exception:
-            return
+        if self.__getGPUIOKitAppleSilicon():
+            pass
+        else:
+            gpus = list()
+            try:
+                gpu_info_dict = json.loads(subprocess.check_output(["system_profiler", "SPDisplaysDataType", "-json"]))
+                if 'SPDisplaysDataType' in gpu_info_dict:
+                    gpus = gpu_info_dict['SPDisplaysDataType']
+                    self.__gpuInfo.number = len(gpus)
+                else:
+                    pass
+            except Exception:
+                return
 
-        for gpu in gpus:
-            self.__gpuInfo.gpus.append(f'{gpu.get("sppci_model")} ({gpu.get("sppci_cores")} Cores) [SOC Integrated]')
+            for gpu in gpus:
+                self.__gpuInfo.gpus.append(f'{gpu.get("sppci_model")} ({gpu.get("sppci_cores")} Cores) [SOC Integrated]')
 
     def __getGPUIntel(self):
-        if self.__getGPUIOKit():
+        if self.__getGPUIOKitIntel():
             pass
         else:  # fallback to the default implementation
             gpus = list()
@@ -56,7 +60,7 @@ class GPUDetectMacOS:
                 elif self.__handleVendor(gpu.get("spdisplays_vendor")) == "Nvidia":    # Since current macOS does not support NVIDIA GPUs, this condition is not applicable
                     pass
 
-    def __getGPUIOKit(self):
+    def __getGPUIOKitIntel(self):
         try:
             package_root = Path(__file__).resolve().parent.parent.parent
             lib = ctypes.CDLL(f"{package_root}/library/lib/iokitGPULib.dylib")
@@ -82,6 +86,22 @@ class GPUDetectMacOS:
             # print(f"An error occurred while getting GPU info using IOKit: {e}")
             return False
 
+    def __getGPUIOKitAppleSilicon(self):
+        try:
+            package_root = Path(__file__).resolve().parent.parent.parent
+            lib = ctypes.CDLL(f"{package_root}/library/lib/iokitGPULib.dylib")
+            lib.getAppleSiliconGPUInfo.restype = ctypes.c_char_p
+            gpu_cores = lib.getAppleSiliconGPUInfo().decode('utf-8')
+
+            if gpu_cores == "0":
+                return False
+            self.__gpuInfo.number = 1
+            self.__gpuInfo.gpus.append(f'{CPUDetect(os="macos").getCPUInfo().model} ({gpu_cores} Cores) [SOC Integrated]')
+            return True
+        except Exception as e:
+            # print(f"An error occurred while getting GPU info using IOKit: {e}")
+            return False
+
     @staticmethod
     def __handleVendor(vendor):
         if vendor == "sppci_vendor_Apple":

pyhw-0.14.3/src/pyhw/backend/nic/macos.py

@@ -0,0 +1,192 @@
+from .nicInfo import NICInfo
+import subprocess
+import ctypes
+from ctypes import c_char_p, c_bool, c_int32, c_char, byref, create_string_buffer
+from pathlib import Path
+
+
+class NICDetectMacOS:
+    def __init__(self):
+        self.__nicInfo = NICInfo()
+
+    def getNICInfo(self):
+        self.__getNICInfo()
+        return self.__nicInfo
+
+    def __getNICInfo(self):
+        # Try to get NIC info using IOKit first
+        if not self.__getNICInfoIOKit():
+            try:
+                # Get default interface
+                cmd_get_interface = "route get default | grep interface"
+                interface_output = subprocess.run(["bash", "-c", cmd_get_interface],
+                                                  capture_output=True, text=True).stdout.strip()
+                interface = interface_output.split(":")[1].strip()
+
+                # Get IP address
+                if_ip = subprocess.run(["bash", "-c", f"ipconfig getifaddr {interface}"],
+                                       capture_output=True, text=True).stdout.strip()
+
+                # Get interface type and link speed
+                link_info = self.__getLinkInfo(interface)
+
+                # Add all information to nicInfo
+                self.__nicInfo.nics.append(f"{interface} @ {if_ip} - {link_info}")
+                self.__nicInfo.number += 1
+            except Exception as e:
+                self.__handleError()
+        else:
+            pass
+
+    def __getNICInfoIOKit(self):
+        try:
+            package_root = Path(__file__).resolve().parent.parent.parent
+            lib = ctypes.CDLL(f"{package_root}/library/lib/iokitNICLib.dylib")
+
+            lib.getDefaultInterface.argtypes = [c_char_p]
+            lib.getDefaultInterface.restype = c_bool
+
+            lib.getNetworkInfo.argtypes = [c_char_p, ctypes.POINTER(c_bool),
+                                           c_char_p, ctypes.POINTER(c_int32),
+                                           c_char_p, c_char_p, c_char_p, c_char_p]
+            lib.getNetworkInfo.restype = c_bool
+            interface = create_string_buffer(32)
+            if lib.getDefaultInterface(interface):
+                interface_str = interface.value.decode('utf-8')
+
+                is_wifi = c_bool(False)
+                ip_address = create_string_buffer(32)
+                speed = c_int32(0)
+                band = create_string_buffer(16)
+                channel = create_string_buffer(32)
+                conn_type = create_string_buffer(32)
+                wifi_standard = create_string_buffer(16)
+
+                if lib.getNetworkInfo(interface_str.encode('utf-8'),
+                                      byref(is_wifi),
+                                      ip_address,
+                                      byref(speed),
+                                      band,
+                                      channel,
+                                      conn_type,
+                                      wifi_standard):
+
+                    if is_wifi.value:
+                        conn_info = f"{conn_type.value.decode('utf-8')} 802.11{wifi_standard.value.decode('utf-8')} ({band.value.decode('utf-8')} {speed.value} Mbps)"
+                    else:
+                        conn_info = conn_type.value.decode('utf-8')
+
+                    formatted_output = f"{interface_str} @ {ip_address.value.decode('utf-8')} - {conn_info}"
+
+                    self.__nicInfo.nics.append(formatted_output)
+                    self.__nicInfo.number += 1
+
+                    return True
+
+                else:
+                    return False
+            else:
+                return False
+
+        except Exception as e:
+            # print(f"An error occurred while getting NIC info using IOKit: {e}")
+            return False
+
+
+    def __getLinkInfo(self, interface):
+        try:
+            # Check if interface is wireless
+            is_wifi_cmd = f"networksetup -getairportpower {interface} 2>/dev/null"
+            is_wifi_result = subprocess.run(["bash", "-c", is_wifi_cmd],
+                                            capture_output=True, text=True).returncode
+
+            if is_wifi_result == 0:  # This is a Wi-Fi interface
+                speed, channel, band = self.__getWifiInfo(interface)
+                return f"Wi-Fi {band} ({speed} Mbps)"
+            else:
+                # For wired connections, get link speed using networksetup
+                media_cmd = f"networksetup -getmedia {interface} | grep 'Active'"
+                media_info = subprocess.run(["bash", "-c", media_cmd],
+                                            capture_output=True, text=True).stdout.strip()
+
+                if media_info:
+                    parts = media_info.split(':', 1)
+                    if len(parts) > 1:
+                        # Extract "2500Base-T <full-duplex>" part
+                        media_details = parts[1].strip()
+
+                        # Parse the speed part (e.g., "2500Base-T")
+                        speed_part = media_details.split()[0] if media_details else ""
+
+                        if "2500Base-T" in speed_part:
+                            return "Wired (2.5 Gbps)"
+                        elif "1000Base-T" in speed_part:
+                            return "Wired (1 Gbps)"
+                        elif "100Base-T" in speed_part:
+                            return "Wired (100 Mbps)"
+                        elif "10000Base-T" in speed_part:
+                            return "Wired (10 Gbps)"
+                        elif "Base-T" in speed_part or "base-T" in speed_part:
+                            # Extract the numeric part from formats like "5000Base-T"
+                            import re
+                            speed_match = re.search(r'(\d+)(?:Base-T|base-T)', speed_part)
+                            if speed_match:
+                                speed_value = int(speed_match.group(1))
+                                if speed_value >= 1000:
+                                    return f"Wired ({speed_value / 1000:.1f} Gbps)"
+                                else:
+                                    return f"Wired ({speed_value} Mbps)"
+                            return f"Wired ({speed_part})"
+        except:
+            return "Unknown connection type"
+
+    def __getWifiInfo(self, interface):
+        # 使用 system_profiler 获取 WiFi 信息
+        profiler_cmd = f"system_profiler SPAirPortDataType -json"
+        try:
+            wifi_data = subprocess.run(["bash", "-c", profiler_cmd],
+                                       capture_output=True, text=True, timeout=5).stdout
+
+            import json
+            data = json.loads(wifi_data)
+
+            # 根据实际 JSON 结构提取信息
+            if 'SPAirPortDataType' in data and isinstance(data['SPAirPortDataType'], list):
+                airport_data = data['SPAirPortDataType'][0]
+
+                # 获取接口列表
+                if 'spairport_airport_interfaces' in airport_data:
+                    interfaces = airport_data['spairport_airport_interfaces']
+
+                    # 查找匹配的接口
+                    for iface in interfaces:
+                        if interface == iface.get('_name', ''):
+                            # 获取当前网络信息
+                            current_network = iface.get('spairport_current_network_information', {})
+
+                            # 从当前连接中提取速率
+                            speed = current_network.get('spairport_network_rate', 'Unknown')
+
+                            # 获取信道信息
+                            channel_info = current_network.get('spairport_network_channel', 'Unknown')
+
+                            # 确定频段信息
+                            band = "Unknown"
+                            if isinstance(channel_info, str) and "GHz" in channel_info:
+                                if "2GHz" in channel_info:
+                                    band = "2.4GHz"
+                                elif "5GHz" in channel_info:
+                                    band = "5GHz"
+                                elif "6GHz" in channel_info:
+                                    band = "6GHz"
+
+                            return speed, channel_info, band
+
+            return "Unknown", "Unknown", "Unknown"
+        except Exception as e:
+            # 可以考虑添加日志记录 print(f"WiFi info error: {str(e)}")
+            return "Unknown", "Unknown", "Unknown"
+
+    def __handleError(self):
+        self.__nicInfo.nics.append("en0 - Unknown connection")
+        self.__nicInfo.number = 1

pyhw-0.14.3/src/pyhw/library/test/iokitGPULibTest.py

@@ -0,0 +1,17 @@
+import ctypes
+
+
+def main():
+    lib = ctypes.CDLL(f"../lib/iokitGPULib.dylib")
+    lib.getGPUInfo.restype = ctypes.c_char_p
+    lib.getAppleSiliconGPUInfo.restype = ctypes.c_char_p
+    gpu_info = lib.getAppleSiliconGPUInfo()
+    print(gpu_info.decode('utf-8'))
+
+
+if __name__ == "__main__":
+    main()
+
+
+
+

pyhw-0.14.3/src/pyhw/library/test/iokitNICLibTest.py

@@ -0,0 +1,76 @@
+import ctypes
+from ctypes import c_char_p, c_bool, c_int32, c_char, byref, create_string_buffer
+import os
+
+
+def main():
+    current_dir = os.path.dirname(os.path.abspath(__file__))
+    lib_path = os.path.join(current_dir, "../lib/iokitNICLib.dylib")
+
+    try:
+        lib = ctypes.CDLL(lib_path)
+
+        lib.getDefaultInterface.argtypes = [c_char_p]
+        lib.getDefaultInterface.restype = c_bool
+
+        lib.getNetworkInfo.argtypes = [c_char_p, ctypes.POINTER(c_bool),
+                                       c_char_p, ctypes.POINTER(c_int32),
+                                       c_char_p, c_char_p, c_char_p, c_char_p]
+        lib.getNetworkInfo.restype = c_bool
+
+        interface = create_string_buffer(32)
+        if lib.getDefaultInterface(interface):
+            interface_str = interface.value.decode('utf-8')
+
+            # interface_str = "en1"
+            print(f"默认网络接口: {interface_str}")
+
+            is_wifi = c_bool(False)
+            ip_address = create_string_buffer(32)
+            speed = c_int32(0)
+            band = create_string_buffer(16)
+            channel = create_string_buffer(32)
+            conn_type = create_string_buffer(32)
+            wifi_standard = create_string_buffer(16)
+
+            if lib.getNetworkInfo(interface_str.encode('utf-8'),
+                                  byref(is_wifi),
+                                  ip_address,
+                                  byref(speed),
+                                  band,
+                                  channel,
+                                  conn_type,
+                                  wifi_standard):
+
+                print("=== 网络信息 ===")
+                print(f"接口: {interface_str}")
+                print(f"IP地址: {ip_address.value.decode('utf-8')}")
+                print(f"连接类型: {conn_type.value.decode('utf-8')}")
+                print(f"速度: {speed.value} Mbps")
+
+                if is_wifi.value:
+                    print(f"Wi-Fi频段: {band.value.decode('utf-8')}")
+                    print(f"信道: {channel.value.decode('utf-8')}")
+                    print(f"Wi-Fi标准: {wifi_standard.value.decode('utf-8')}")
+
+                if is_wifi.value:
+                    wifi_info = f"{wifi_standard.value.decode('utf-8')} {band.value.decode('utf-8')}"
+                    conn_info = f"{conn_type.value.decode('utf-8')} ({wifi_info}, {speed.value} Mbps)"
+                else:
+                    conn_info = conn_type.value.decode('utf-8')
+
+                formatted_output = f"{interface_str} @ {ip_address.value.decode('utf-8')} - {conn_info}"
+                print("\n格式化输出:")
+                print(formatted_output)
+            else:
+                print(f"无法获取接口 {interface_str} 的网络信息")
+        else:
+            print("无法获取默认网络接口")
+
+    except Exception as e:
+        print(f"错误: {e}")
+
+
+if __name__ == "__main__":
+    main()
+

{pyhw-0.14.1 → pyhw-0.14.3/src/pyhw.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pyhw
-Version: 0.14.1
+Version: 0.14.3
 Summary: PyHw, a neofetch-like command line tool for fetching system information but written mostly in python.
 Author: Xiao Ran
 Maintainer-email: Xiao Ran <xiaoran.007@icloud.com>

{pyhw-0.14.1 → pyhw-0.14.3}/src/pyhw.egg-info/SOURCES.txt

@@ -115,10 +115,13 @@ src/pyhw/frontend/logo/ascii/windows_old.pyhw
 src/pyhw/library/lib/iokitCPULib.dylib
 src/pyhw/library/lib/iokitGPULib.dylib
 src/pyhw/library/lib/iokitHostLib.dylib
+src/pyhw/library/lib/iokitNICLib.dylib
 src/pyhw/library/lib/nvmlGPULib_amd64.so
 src/pyhw/library/lib/nvmlGPULib_arm64.so
 src/pyhw/library/test/iokitCPULibTest.py
+src/pyhw/library/test/iokitGPULibTest.py
 src/pyhw/library/test/iokitHostLibTest.py
+src/pyhw/library/test/iokitNICLibTest.py
 src/pyhw/library/test/nvmlGPULibTest.py
 src/pyhw/pyhwException/__init__.py
 src/pyhw/pyhwException/pyhwException.py

pyhw-0.14.1/src/pyhw/backend/nic/macos.py

@@ -1,25 +0,0 @@
-from .nicInfo import NICInfo
-import subprocess
-
-
-class NICDetectMacOS:
-    def __init__(self):
-        self.__nicInfo = NICInfo()
-
-    def getNICInfo(self):
-        self.__getNICInfo()
-        return self.__nicInfo
-
-    def __getNICInfo(self):
-        # Placeholder for a more advanced method.
-        try:
-            interface = subprocess.run(["bash", "-c", "route get default | grep interface"], capture_output=True, text=True).stdout.strip().split(":")[1]
-            if_ip = subprocess.run(["bash", "-c", f"ipconfig getifaddr {interface}"], capture_output=True, text=True).stdout.strip()
-            self.__nicInfo.nics.append(f"{interface} @ {if_ip}")
-            self.__nicInfo.number += 1
-        except:
-            self.__handleError()
-
-    def __handleError(self):
-        self.__nicInfo.nics.append("en0")
-        self.__nicInfo.number = 1