[CORE] Prevents inconsistent PATH from breaking subprocess executions

It appears "broken" `PATH` environment variable (e.g. ending with a
file instead of a directory) may lead to `NotADirectoryError` Python
exception. Let's get bulletproof by extending most of subprocess calls
expected exceptions to parent `OSError` (or at least by actually
catching `NotADirectoryError` explicitly).

> closes #164

Co-Authored-By: Michael Bromilow <developer@bromilow.uk>

Author: HorlogeSkynet

archey/entries/cpu.py

@@ -101,7 +101,7 @@ def _parse_lscpu_output(cls) -> List[Dict[str, int]]:
         """Same operation but from `lscpu` output"""
         try:
             cpu_info = check_output("lscpu", env={"LANG": "C"}, universal_newlines=True)
-        except FileNotFoundError:
+        except OSError:
             return []
 
         nb_threads = cls._THREADS_PER_CORE_REGEXP.findall(cpu_info)
@@ -129,7 +129,7 @@ def _parse_system_profiler() -> List[Dict[str, int]]:
                 stderr=DEVNULL,
                 universal_newlines=True,
             )
-        except (FileNotFoundError, CalledProcessError):
+        except (OSError, CalledProcessError):
             # `-json` is not available before Catalina.
             return []
 
@@ -165,7 +165,7 @@ def _parse_sysctl_machdep() -> List[Dict[str, int]]:
                 stderr=DEVNULL,
                 universal_newlines=True,
             )
-        except (FileNotFoundError, CalledProcessError):
+        except (OSError, CalledProcessError):
             return []
 
         # `sysctl_output` should exactly contains two lines.
@@ -179,7 +179,7 @@ def _parse_sysctl_cpu_model() -> List[Dict[str, int]]:
             sysctl_output = check_output(
                 ["sysctl", "-n", "hw.model", "hw.ncpu"], stderr=DEVNULL, universal_newlines=True
             )
-        except (FileNotFoundError, CalledProcessError):
+        except (OSError, CalledProcessError):
             return []
 
         # `sysctl_output` should exactly contains two lines.

archey/entries/disk.py

@@ -71,7 +71,7 @@ def _replace_apfs_volumes_by_their_containers(self) -> Dict[str, dict]:
         # Call `diskutil` to generate a property list (PList) of all APFS containers
         try:
             property_list = plistlib.loads(check_output(["diskutil", "apfs", "list", "-plist"]))
-        except FileNotFoundError:
+        except OSError:
             self._logger.warning(
                 "APFS volumes cannot be deduplicated as diskutil program could not be found."
             )
@@ -186,7 +186,7 @@ def _get_df_output_dict() -> Dict[str, dict]:
                 check=False,
                 encoding="utf-8",
             ).stdout
-        except FileNotFoundError:
+        except OSError:
             # `df` isn't available on this system.
             return {}
 

archey/entries/distro.py

@@ -30,7 +30,7 @@ def _fetch_android_release() -> Optional[str]:
             release = check_output(
                 ["getprop", "ro.build.version.release"], universal_newlines=True
             ).rstrip()
-        except FileNotFoundError:
+        except OSError:
             return None
 
         return f"Android {release}"

archey/entries/gpu.py

@@ -49,7 +49,7 @@ def _parse_lspci_output() -> List[str]:
         """Based on `lspci` output, return a list of video controllers names"""
         try:
             lspci_output = check_output(["lspci", "-m"], universal_newlines=True).splitlines()
-        except (FileNotFoundError, CalledProcessError):
+        except (OSError, CalledProcessError):
             return []
 
         gpus_list = []
@@ -122,7 +122,7 @@ def _parse_system_profiler() -> List[str]:
             profiler_output = check_output(
                 ["system_profiler", "SPDisplaysDataType"], stderr=DEVNULL, universal_newlines=True
             )
-        except FileNotFoundError:
+        except OSError:
             return []
 
         return re.findall(r"Chipset Model: (.*)", profiler_output, re.MULTILINE)
@@ -134,7 +134,7 @@ def _parse_pciconf_output() -> List[str]:
             pciconf_output = check_output(
                 ["pciconf", "-lv"], stderr=DEVNULL, universal_newlines=True
             )
-        except (FileNotFoundError, CalledProcessError):
+        except (OSError, CalledProcessError):
             return []
 
         gpus_list = []

archey/entries/model.py

@@ -60,7 +60,7 @@ def _fetch_virtual_env_info(self) -> Optional[str]:
         except CalledProcessError:
             # Not a virtual environment.
             return None
-        except FileNotFoundError:
+        except OSError:
             # If not available, let's query `virt-what` (privileges usually required).
             try:
                 return (
@@ -130,7 +130,7 @@ def _fetch_sysctl_hw() -> Optional[str]:
                 model = check_output(
                     ["sysctl", "-n", "hw.model"], stderr=DEVNULL, universal_newlines=True
                 )
-            except FileNotFoundError:
+            except OSError:
                 return None
             except CalledProcessError:
                 pass
@@ -144,7 +144,7 @@ def _fetch_sysctl_hw() -> Optional[str]:
                 sysctl_output = check_output(
                     ["sysctl", "-n", f"hw.{hw_oid}"], stderr=DEVNULL, universal_newlines=True
                 )
-            except FileNotFoundError:
+            except OSError:
                 return None
             except CalledProcessError:
                 pass
@@ -189,7 +189,7 @@ def _fetch_android_device_model() -> Optional[str]:
         try:
             brand = check_output(["getprop", "ro.product.brand"], universal_newlines=True).rstrip()
             model = check_output(["getprop", "ro.product.model"], universal_newlines=True).rstrip()
-        except FileNotFoundError:
+        except OSError:
             return None
 
         return f"{brand} ({model})"
@@ -202,7 +202,7 @@ def _fetch_freebsd_model() -> Optional[str]:
             product = check_output(
                 ["kenv", "smbios.system.version"], universal_newlines=True
             ).rstrip()
-        except (FileNotFoundError, CalledProcessError):
+        except (OSError, CalledProcessError):
             return None
 
         return f"{vendor} ({product})"

archey/entries/ram.py

@@ -39,14 +39,14 @@ def _get_used_total_values(self) -> Tuple[float, float]:
         Tries a variety of methods, increasing compatibility for a wide range of systems.
         """
         if platform.system() == "Linux":
-            with suppress(IndexError, FileNotFoundError):
+            with suppress(IndexError, OSError):
                 return self._run_free_dash_m()
         elif platform.system() == "FreeBSD":
-            with suppress(FileNotFoundError):
+            with suppress(OSError):
                 return self._run_sysctl_mem()
         else:
             # Darwin or any other BSD-based system.
-            with suppress(FileNotFoundError):
+            with suppress(OSError):
                 return self._run_sysctl_and_vmstat()
 
         with suppress(OSError):

archey/entries/temperature.py

@@ -82,7 +82,7 @@ def _get_sensors_output(whitelisted_chip: Optional[str]) -> Optional[str]:
                 sensors_output = run(
                     sensors_args, universal_newlines=True, stdout=PIPE, stderr=PIPE, check=True
                 )
-            except FileNotFoundError:
+            except OSError:
                 return None
             except CalledProcessError as called_process_error:
                 error_message = called_process_error.stderr
@@ -187,7 +187,7 @@ def _run_sysctl_dev_cpu(self) -> None:
                 stderr=PIPE,
                 universal_newlines=True,
             )
-        except FileNotFoundError:
+        except OSError:
             # `sysctl` does not seem to be available on this system.
             return
         except CalledProcessError as error_message:

archey/entries/uptime.py

@@ -72,7 +72,7 @@ def _parse_uptime_cmd(self) -> timedelta:
         """Tries to get uptime by parsing the `uptime` command"""
         try:
             uptime_output = run("uptime", env={"LANG": "C"}, stdout=PIPE, stderr=PIPE, check=True)
-        except FileNotFoundError as error:
+        except (FileNotFoundError, NotADirectoryError) as error:
             raise ArcheyException("Couldn't find `uptime` command on this system.") from error
 
         # Log any `uptime` error messages at warning level.

archey/entries/wan_ip.py

@@ -82,7 +82,7 @@ def _run_dns_query(query: str, resolver: str, ip_version: int, timeout: float) -
                 stderr=DEVNULL,
                 universal_newlines=True,
             ).rstrip()
-        except (FileNotFoundError, TimeoutExpired, CalledProcessError):
+        except (OSError, TimeoutExpired, CalledProcessError):
             return None
 
         # `ip_address` might be empty here.

archey/entries/window_manager.py

@@ -76,7 +76,7 @@ def __init__(self, *args, **kwargs):
                 r"(?<=Name: ).*",
                 check_output(["wmctrl", "-m"], stderr=DEVNULL, universal_newlines=True),
             ).group(0)
-        except (FileNotFoundError, CalledProcessError):
+        except (OSError, CalledProcessError):
             processes = Processes().list
             for wm_id, wm_name in WM_DICT.items():
                 if wm_id in processes: