pyhw 0.6.6__py3-none-any.whl → 0.6.7__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,9 +20,13 @@ 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
+        print(self.__shellInfo.info)
         return self.__shellInfo
 
     def __getShell(self):
@@ -31,6 +36,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 +57,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 +72,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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: pyhw
-Version: 0.6.6
+Version: 0.6.7
 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.7.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=Yradj6aj4nqc9CNQvVqzOgwO-p2IIJ_Jt_okL7cF4as,2146
 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
@@ -80,12 +80,12 @@ pyhw/frontend/logo/ascii/ubuntu_small.pyhw,sha256=Xf8LSZdZUu9aEG3efhb1FUlUEuJ-3U
 pyhw/library/lib/iokitGPULib.dylib,sha256=DcJ0GZY79gTFckLFYtZgeKn1T0NFvdO_k_ccCa7od5Y,154808
 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=OY89sHr5LBjob7JMbCl-6x1AEPr-15UOE8UoQ04sLvI,5968
 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.7.dist-info/LICENSE,sha256=hJs6RBqSVCexbTsalkMLNFI5t06kekQEsSVaOt_-yLs,1497
+pyhw-0.6.7.dist-info/METADATA,sha256=NL0A7juUgOQRXCkDyg32z1NXzAFq8AF2drk4FSrkJi8,5225
+pyhw-0.6.7.dist-info/WHEEL,sha256=R06PA3UVYHThwHvxuRWMqaGcr-PuniXahwjmQRFMEkY,91
+pyhw-0.6.7.dist-info/entry_points.txt,sha256=q-AB8im_QahpmNrmy4aPTJRGi0LlbNlnI3kF7s6pKss,44
+pyhw-0.6.7.dist-info/top_level.txt,sha256=7Inxvxt1TngEricKZEex9_WJZS3DbKYFUXDz4v5WHYU,5
+pyhw-0.6.7.dist-info/RECORD,,