pyhw 0.14.0__tar.gz → 0.14.2__tar.gz

{pyhw-0.14.0/src/pyhw.egg-info → pyhw-0.14.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pyhw
-Version: 0.14.0
+Version: 0.14.2
 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.0 → pyhw-0.14.2}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "pyhw"
-version = "0.14.0"
+version = "0.14.2"
 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.0 → pyhw-0.14.2}/src/pyhw/__init__.py

@@ -1,2 +1,2 @@
-__version__ = '0.14.0'
+__version__ = '0.14.2'
 __author__ = 'xiaoran007'

pyhw-0.14.2/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.0 → pyhw-0.14.2}/src/pyhw/backend/npu/macos.py

@@ -44,20 +44,20 @@ class NPUDetectMacOS:
         # see https://apple.fandom.com/wiki/Neural_Engine for more details.
         model_name = CPUDetect(os=getOS()).getCPUInfo().model
         npu = {
-            "Apple M1": "Apple Neural Engine 16 Cores (5nm) [SOC Integrated]",
-            "Apple M1 Pro": "Apple Neural Engine 16 Cores (5nm) [SOC Integrated]",
-            "Apple M1 Max": "Apple Neural Engine 16 Cores (5nm) [SOC Integrated]",
-            "Apple M1 Ultra": "Apple Neural Engine 32 Cores (5nm) [SOC Integrated]",
-            "Apple M2": "Apple Neural Engine 16 Cores (5nm) [SOC Integrated]",
-            "Apple M2 Pro": "Apple Neural Engine 16 Cores (5nm) [SOC Integrated]",
-            "Apple M2 Max": "Apple Neural Engine 16 Cores (5nm) [SOC Integrated]",
-            "Apple M2 Ultra": "Apple Neural Engine 32 Cores (5nm) [SOC Integrated]",
-            "Apple M3": "Apple Neural Engine 16 Cores (3nm) [SOC Integrated]",
-            "Apple M3 Pro": "Apple Neural Engine 16 Cores (3nm) [SOC Integrated]",
-            "Apple M3 Max": "Apple Neural Engine 16 Cores (3nm) [SOC Integrated]",
-            "Apple M3 Ultra": "Apple Neural Engine 32 Cores (3nm) [SOC Integrated]",
-            "Apple M4": "Apple Neural Engine 16 Cores (3nm) [SOC Integrated]",
-            "Apple M4 Pro": "Apple Neural Engine 16 Cores (3nm) [SOC Integrated]",
-            "Apple M4 Max": "Apple Neural Engine 16 Cores (3nm) [SOC Integrated]"
+            "Apple M1": "Apple Neural Engine 16 Cores (11 TOPS) [SOC Integrated]",
+            "Apple M1 Pro": "Apple Neural Engine 16 Cores (11 TOPS [SOC Integrated]",
+            "Apple M1 Max": "Apple Neural Engine 16 Cores (11 TOPS) [SOC Integrated]",
+            "Apple M1 Ultra": "Apple Neural Engine 32 Cores (22 TOPS) [SOC Integrated]",
+            "Apple M2": "Apple Neural Engine 16 Cores (15.8 TOPS) [SOC Integrated]",
+            "Apple M2 Pro": "Apple Neural Engine 16 Cores (15.8 TOPS) [SOC Integrated]",
+            "Apple M2 Max": "Apple Neural Engine 16 Cores (15.8 TOPS) [SOC Integrated]",
+            "Apple M2 Ultra": "Apple Neural Engine 32 Cores (31.6 TOPS) [SOC Integrated]",
+            "Apple M3": "Apple Neural Engine 16 Cores (18 TOPS) [SOC Integrated]",
+            "Apple M3 Pro": "Apple Neural Engine 16 Cores (18 TOPS) [SOC Integrated]",
+            "Apple M3 Max": "Apple Neural Engine 16 Cores (18 TOPS) [SOC Integrated]",
+            "Apple M3 Ultra": "Apple Neural Engine 32 Cores (36 TOPS) [SOC Integrated]",
+            "Apple M4": "Apple Neural Engine 16 Cores (38 TOPS) [SOC Integrated]",
+            "Apple M4 Pro": "Apple Neural Engine 16 Cores (38 TOPS) [SOC Integrated]",
+            "Apple M4 Max": "Apple Neural Engine 16 Cores (38 TOPS) [SOC Integrated]"
         }
         return npu.get(model_name, "Not Found")

pyhw-0.14.2/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.0 → pyhw-0.14.2/src/pyhw.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pyhw
-Version: 0.14.0
+Version: 0.14.2
 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.0 → pyhw-0.14.2}/src/pyhw.egg-info/SOURCES.txt

@@ -115,10 +115,12 @@ 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/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.0/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