pyhw 0.6.6__py3-none-any.whl → 0.6.8__py3-none-any.whl

pyhw/backend/host/linux.py

@@ -1,7 +1,7 @@
 """
     In dev.
 """
-from ...pyhwUtil import getArch
+from ...pyhwUtil import getArch, getDocker
 from .hostInfo import HostInfo
 import os
 
@@ -10,47 +10,53 @@ class HostDetectLinux:
     def __init__(self):
         self.__hostInfo = HostInfo()
         self.__arch = getArch()
+        self.__docker = getDocker()
 
     def getHostInfo(self):
         self.__getModel()
         return self.__hostInfo
 
     def __getModel(self):
-        if self.__arch in ["x86_64", "x86"]:
-            try:
-                with open("/sys/devices/virtual/dmi/id/product_name", "r") as f:
-                    product_name = f.read().strip()
-                    if product_name.startswith("To be filled by O.E.M."):
-                        self.__hostInfo.name = f"General {self.__arch} Host"
-                    else:
-                        self.__hostInfo.name = product_name
-                with open("/sys/devices/virtual/dmi/id/product_version", "r") as f:
-                    version = f.read().strip()
-                    if version.startswith("To be filled by O.E.M."):
-                        self.__hostInfo.version = ""
-                    else:
-                        self.__hostInfo.version = version
-                self.__hostInfo.model = self.__hostInfo.name + " " + self.__hostInfo.version
-            except FileNotFoundError:
-                pass
-        elif self.__arch in ["aarch64", "arm32"]:
-            # try to find dmi folder since some arm based desktops and servers may have same structure as x86_64 machines.
-            if os.path.exists("/sys/devices/virtual/dmi/id"):
+        if self.__docker:
+            self.__hostInfo.name = f"General {self.__arch} Docker Host"
+            self.__hostInfo.version = ""
+            self.__hostInfo.model = self.__hostInfo.name + " " + self.__hostInfo.version
+        else:
+            if self.__arch in ["x86_64", "x86"]:
                 try:
                     with open("/sys/devices/virtual/dmi/id/product_name", "r") as f:
-                        self.__hostInfo.name = f.read().strip()
+                        product_name = f.read().strip()
+                        if product_name.startswith("To be filled by O.E.M."):
+                            self.__hostInfo.name = f"General {self.__arch} Host"
+                        else:
+                            self.__hostInfo.name = product_name
                     with open("/sys/devices/virtual/dmi/id/product_version", "r") as f:
-                        self.__hostInfo.version = f.read().strip()
+                        version = f.read().strip()
+                        if version.startswith("To be filled by O.E.M."):
+                            self.__hostInfo.version = ""
+                        else:
+                            self.__hostInfo.version = version
                     self.__hostInfo.model = self.__hostInfo.name + " " + self.__hostInfo.version
                 except FileNotFoundError:
                     pass
-            else:
-                # some single board computers may not have dmi folder, try to find model name in device tree.
-                try:
-                    with open("/sys/firmware/devicetree/base/model", "r") as f:
-                        self.__hostInfo.model = f.read().strip()
-                except FileNotFoundError:
-                    pass
+            elif self.__arch in ["aarch64", "arm32"]:
+                # try to find dmi folder since some arm based desktops and servers may have same structure as x86_64 machines.
+                if os.path.exists("/sys/devices/virtual/dmi/id"):
+                    try:
+                        with open("/sys/devices/virtual/dmi/id/product_name", "r") as f:
+                            self.__hostInfo.name = f.read().strip()
+                        with open("/sys/devices/virtual/dmi/id/product_version", "r") as f:
+                            self.__hostInfo.version = f.read().strip()
+                        self.__hostInfo.model = self.__hostInfo.name + " " + self.__hostInfo.version
+                    except FileNotFoundError:
+                        pass
+                else:
+                    # some single board computers may not have dmi folder, try to find model name in device tree.
+                    try:
+                        with open("/sys/firmware/devicetree/base/model", "r") as f:
+                            self.__hostInfo.model = f.read().strip()
+                    except FileNotFoundError:
+                        pass
 
     def __getHostFamily(self):
         pass

pyhw/backend/shell/unix.py

@@ -1,6 +1,7 @@
 """
     In dev.
 """
+from ...pyhwUtil import getDocker
 from dataclasses import dataclass
 import os
 import subprocess
@@ -19,7 +20,10 @@ class ShellDetectUnix:
         self.__shellInfo = ShellInfoUnix()
 
     def getShellInfo(self):
-        self.__getShell()
+        if getDocker():
+            self.__getShellDocker()
+        else:
+            self.__getShell()
         self.__getVersion()
         self.__shellInfo.info = self.__shellInfo.shell + " " + self.__shellInfo.version
         return self.__shellInfo
@@ -31,6 +35,19 @@ class ShellDetectUnix:
             self.__shellInfo.shell = shell_env.split("/")[-1]
             self.__shellInfo.path = shell_env
 
+    def __getShellDocker(self):
+        try:
+            with open("/etc/passwd", "r") as f:
+                for line in f:
+                    if line.startswith("root:"):
+                        root_shell = line.strip().split(":")[-1]
+                        break
+        except:
+            root_shell = ""
+        if root_shell != "":
+            self.__shellInfo.shell = root_shell.split("/")[-1]
+            self.__shellInfo.path = root_shell
+
     def __getVersion(self):
         shell = self.__shellInfo.shell
         if shell != "":
@@ -39,7 +56,7 @@ class ShellDetectUnix:
             elif shell == "bash":
                 try:
                     result = subprocess.run(["bash", "-c", "echo $BASH_VERSION"], capture_output=True, text=True)
-                    self.__shellInfo.version = result.stdout.strip()
+                    self.__shellInfo.version = result.stdout.strip().split("(")[0]
                 except subprocess.SubprocessError:
                     pass
             elif shell == "zsh":
@@ -54,3 +71,4 @@ class ShellDetectUnix:
 
 
 
+

pyhw/pyhwUtil/__init__.py

@@ -1,4 +1,4 @@
-from .pyhwUtil import getOS, getArch, createDataString, selectOSLogo
+from .pyhwUtil import getOS, getArch, getDocker, createDataString, selectOSLogo
 from .sysctlUtil import sysctlGetString, sysctlGetInt
 
-__all__ = ["getOS", "getArch", "createDataString", "selectOSLogo", "sysctlGetString", "sysctlGetInt"]
+__all__ = ["getOS", "getArch", "getDocker", "createDataString", "selectOSLogo", "sysctlGetString", "sysctlGetInt"]

pyhw/pyhwUtil/pyhwUtil.py

@@ -38,6 +38,14 @@ def getArch():
         return "unknown"
 
 
+def getDocker() -> bool:
+    """
+    Check if the current environment is running in Docker.
+    :return: bool, True if running in Docker, False otherwise.
+    """
+    return os.path.exists("/.dockerenv")
+
+
 class DataStringProcessor:
     def __init__(self, data: Data):
         self.data = data
@@ -46,8 +54,11 @@ class DataStringProcessor:
     @staticmethod
     def __getENV() -> int:
         if getOS() == "linux":
-            _, columns_str = os.popen('stty size', 'r').read().split()
-            columns = int(columns_str)
+            try:
+                _, columns_str = os.popen('stty size', 'r').read().split()
+                columns = int(columns_str)
+            except:
+                columns = 80  # default terminal size is 80 columns
         else:
             # macOS default terminal size is 80 columns
             columns = 80
@@ -176,9 +187,14 @@ def selectOSLogo(os_id: str):
         pass
     else:
         return "linux"
-    rows_str, columns_str = os.popen('stty size', 'r').read().split()
-    rows = int(rows_str)
-    columns = int(columns_str)
+    try:
+        rows_str, columns_str = os.popen('stty size', 'r').read().split()
+        rows = int(rows_str)
+        columns = int(columns_str)
+    except:
+        # Default terminal size is 24 rows and 80 columns
+        rows = 24
+        columns = 80
     if columns <= 80:
         if os_id in ["fedora", "ubuntu"]:
             return f"{os_id}_small"

{pyhw-0.6.6.dist-info → pyhw-0.6.8.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: pyhw
-Version: 0.6.6
+Version: 0.6.8
 Summary: PyHw, a neofetch-like command line tool for fetching system information but written mostly in python.
 Author-email: Xiao Ran <xiaoran.007@icloud.com>
 License: BSD-3-Clause

{pyhw-0.6.6.dist-info → pyhw-0.6.8.dist-info}/RECORD

@@ -16,7 +16,7 @@ pyhw/backend/gpu/macos.py,sha256=SmTqqTrIMRW-GVPmDs8xAiOX7HsCjrGh9qkxLQCdvO8,385
 pyhw/backend/host/__init__.py,sha256=Efaj7-Oya7H8HdpZHQCLrwn-mcfPb-d6yfh4dzsE_7I,58
 pyhw/backend/host/hostBase.py,sha256=POyDW3f5JSWtEKyCfrVSBEddSwoywe_OBgUExCEuje8,563
 pyhw/backend/host/hostInfo.py,sha256=Xvz0LugPiCSWMkcDsp4p2rrojYFZauL6Q-gCZ6NLz5k,184
-pyhw/backend/host/linux.py,sha256=wp65o_BJUxq-rkts_1UVP6IuK6U54Cl066G3QfZksiQ,2424
+pyhw/backend/host/linux.py,sha256=5wfEbe4au4JqDazBuUTfWLC2sPPa_YuidSGKVwzyI-0,2853
 pyhw/backend/host/macos.py,sha256=KW-EJK9g1xHNrFsVxJv5GPLpc_ZevX0Zv1WjZUzRkzo,15369
 pyhw/backend/host/windows.py,sha256=rjDJaIs-5zspzFsNCMCh6m2yZXEXI0vccqeBpmAdEBk,53
 pyhw/backend/kernel/__init__.py,sha256=fGjwjpOhwA_PnsWbwoq102hwhTay2ufYKaTqxjSV2-I,65
@@ -47,7 +47,7 @@ pyhw/backend/os/osBase.py,sha256=AA17HIwmWy7E6nCtqbojTNsHKtcNuehf51FxGcfXu7A,462
 pyhw/backend/os/osInfo.py,sha256=iLPc7INFHH3izascwooj4JBVgvBsSgVPXWBlFXG47mQ,378
 pyhw/backend/shell/__init__.py,sha256=SeQ7OLNSl_V1JCCWnJGjLilAWiSe9e5kgsMEt63TMS0,62
 pyhw/backend/shell/shellBase.py,sha256=mylNazVtTbCSzus-IPe1QwnuIGSFNP-z5vYKwTIl_yg,377
-pyhw/backend/shell/unix.py,sha256=COPpSE_wnuuSWDckTvBT0nNwLzgh8iPLicY8XghTSYM,1530
+pyhw/backend/shell/unix.py,sha256=Vt2wSxAIrzN679v0MjJwBoK6L12OqZOichociFZ9lw8,2109
 pyhw/backend/title/__init__.py,sha256=l6_WqreSBeK7krapCrbrr4ndnydrXB1aZ50Ve-beV_w,62
 pyhw/backend/title/titleBase.py,sha256=pcnkiNn9N69yc4iicAmpMpC6G7HUMyGCh0V1jjGIP_c,495
 pyhw/backend/title/unix.py,sha256=9B-zLE8yhlWBnJ-6Aa-DmFmgMV9KdvvNU40drJIaNck,837
@@ -77,15 +77,16 @@ pyhw/frontend/logo/ascii/macOS.pyhw,sha256=HBGROtBb7wrNtRqAQ9ND7zxK0l17BRsmpgc2d
 pyhw/frontend/logo/ascii/raspbian.pyhw,sha256=jEl6WMKF-FGx359ipQ1kzweUJS-NuauYQCFaYime4PQ,159
 pyhw/frontend/logo/ascii/ubuntu.pyhw,sha256=l-Q0PfutxXYMwTojqeiM88063x4V0RBkZUHmi6YdsNc,833
 pyhw/frontend/logo/ascii/ubuntu_small.pyhw,sha256=Xf8LSZdZUu9aEG3efhb1FUlUEuJ-3UztcIOJISpKhPw,229
-pyhw/library/lib/iokitGPULib.dylib,sha256=DcJ0GZY79gTFckLFYtZgeKn1T0NFvdO_k_ccCa7od5Y,154808
+pyhw/library/lib/iokitGPULib.dylib,sha256=18A4SyThKQvzEEOijeWb4XsgXO-iTFYGa0XtrDoBHjY,154808
+pyhw/library/lib/iokitHostLib.dylib,sha256=PKNLMhHwB6-sKDaeo3izbPR88OgzjuRGaLV0NvNWVRk,149256
 pyhw/pyhwException/__init__.py,sha256=8JsFvtF13g0Y5t4z9fRndDXtfCzuBM59jDf6PhWSFSk,220
 pyhw/pyhwException/pyhwException.py,sha256=wxuzFQa9g7XB1q9TUKO_55lw7wMEJMpzG8w1GVTFVa0,197
-pyhw/pyhwUtil/__init__.py,sha256=PzeP9fXsIhvr3sUpJ4DxW9_H25DEIasBFfXd_NztfR4,226
-pyhw/pyhwUtil/pyhwUtil.py,sha256=0Pn-0MqgloQV_p5HrM3HwrGsA_eB-P2eW2hLJrHWmhk,5762
+pyhw/pyhwUtil/__init__.py,sha256=diIqlNUBfuHu-2VAOJk5nipGLafnWxR3chAAOmX8QRo,250
+pyhw/pyhwUtil/pyhwUtil.py,sha256=4qO3Ohc67U0TEXEEOiQpic7aqgeHDsc9-QhuG65j17w,6211
 pyhw/pyhwUtil/sysctlUtil.py,sha256=S-rUvqi7ZrMyMouIhxlyHEQ4agM7sCT1Y7uzs3Hu5-o,841
-pyhw-0.6.6.dist-info/LICENSE,sha256=hJs6RBqSVCexbTsalkMLNFI5t06kekQEsSVaOt_-yLs,1497
-pyhw-0.6.6.dist-info/METADATA,sha256=lDfujbVf0CPbvs5_n97qzeRBA5Y3xmjQ6hWAsuMIeV4,5225
-pyhw-0.6.6.dist-info/WHEEL,sha256=R06PA3UVYHThwHvxuRWMqaGcr-PuniXahwjmQRFMEkY,91
-pyhw-0.6.6.dist-info/entry_points.txt,sha256=q-AB8im_QahpmNrmy4aPTJRGi0LlbNlnI3kF7s6pKss,44
-pyhw-0.6.6.dist-info/top_level.txt,sha256=7Inxvxt1TngEricKZEex9_WJZS3DbKYFUXDz4v5WHYU,5
-pyhw-0.6.6.dist-info/RECORD,,
+pyhw-0.6.8.dist-info/LICENSE,sha256=hJs6RBqSVCexbTsalkMLNFI5t06kekQEsSVaOt_-yLs,1497
+pyhw-0.6.8.dist-info/METADATA,sha256=IWl0AR6pOJrrwgd_7wMo3NBtEJF3h4b9Hlt_CnneGjY,5225
+pyhw-0.6.8.dist-info/WHEEL,sha256=R06PA3UVYHThwHvxuRWMqaGcr-PuniXahwjmQRFMEkY,91
+pyhw-0.6.8.dist-info/entry_points.txt,sha256=q-AB8im_QahpmNrmy4aPTJRGi0LlbNlnI3kF7s6pKss,44
+pyhw-0.6.8.dist-info/top_level.txt,sha256=7Inxvxt1TngEricKZEex9_WJZS3DbKYFUXDz4v5WHYU,5
+pyhw-0.6.8.dist-info/RECORD,,