[psutil] Use more strict pyright settings

Author: donbarbos

pyrightconfig.stricter.json

@@ -72,7 +72,7 @@
         "stubs/pika",
         "stubs/pony",
         "stubs/protobuf",
-        "stubs/psutil",
+        "stubs/psutil/psutil/__init__.pyi",
         "stubs/psycopg2",
         "stubs/pyasn1",
         "stubs/pycurl",

stubs/psutil/psutil/__init__.pyi

@@ -199,7 +199,7 @@ class Process:
         def io_counters(self) -> pio: ...
         def ionice(self, ioclass: int | None = None, value: int | None = None) -> pionice: ...
         def cpu_affinity(self, cpus: list[int] | None = None) -> list[int] | None: ...
-        def memory_maps(self, grouped: bool = True): ...
+        def memory_maps(self, grouped: bool = True) -> list[Incomplete]: ...
     if sys.platform == "linux":
         def rlimit(self, resource: int, limits: tuple[int, int] | None = ...) -> tuple[int, int]: ...
         def cpu_num(self) -> int: ...

stubs/psutil/psutil/_common.pyi

@@ -1,7 +1,9 @@
 import enum
 import io
+import sys
 import threading
 from _typeshed import ConvertibleToFloat, FileDescriptorOrPath, Incomplete, StrOrBytesPath, SupportsWrite
+from collections import defaultdict
 from collections.abc import Callable
 from socket import AF_INET6 as AF_INET6, AddressFamily, SocketKind
 from typing import BinaryIO, Final, NamedTuple, SupportsIndex, TypeVar, overload
@@ -213,9 +215,7 @@ class addr(NamedTuple):
 
 conn_tmap: dict[str, tuple[list[AddressFamily], list[SocketKind]]]
 
-class Error(Exception):
-    msg: str
-    def __init__(self, msg: str = ...) -> None: ...
+class Error(Exception): ...
 
 class NoSuchProcess(Error):
     pid: int
@@ -242,6 +242,7 @@ class TimeoutExpired(Error):
 
 _P = ParamSpec("_P")
 _R = TypeVar("_R")
+_T = TypeVar("_T")
 
 def usage_percent(used: ConvertibleToFloat, total: float, round_: SupportsIndex | None = None) -> float: ...
 
@@ -257,32 +258,64 @@ def parse_environ_block(data: str) -> dict[str, str]: ...
 def sockfam_to_enum(num: int) -> AddressFamily: ...
 def socktype_to_enum(num: int) -> SocketKind: ...
 @overload
-def conn_to_ntuple(fd: int, fam: int, type_: int, laddr, raddr, status: str, status_map, pid: int) -> sconn: ...
+def conn_to_ntuple(
+    fd: int,
+    fam: int,
+    type_: int,
+    laddr: addr | tuple[str, int] | tuple[()],
+    raddr: addr | tuple[str, int] | tuple[()],
+    status: int | str,
+    status_map: dict[int, str] | dict[str, str],
+    pid: int,
+) -> sconn: ...
 @overload
-def conn_to_ntuple(fd: int, fam: int, type_: int, laddr, raddr, status: str, status_map, pid: None = None) -> pconn: ...
+def conn_to_ntuple(
+    fd: int,
+    fam: int,
+    type_: int,
+    laddr: addr | tuple[str, int] | tuple[()],
+    raddr: addr | tuple[str, int] | tuple[()],
+    status: int | str,
+    status_map: dict[int, str] | dict[str, str],
+    pid: None = None,
+) -> pconn: ...
 def deprecated_method(replacement: str) -> Callable[[Callable[_P, _R]], Callable[_P, _R]]: ...
 
 class _WrapNumbers:
     lock: threading.Lock
-    cache: dict[Incomplete, Incomplete]
-    reminders: dict[Incomplete, Incomplete]
-    reminder_keys: dict[Incomplete, Incomplete]
+    cache: dict[str, dict[str, tuple[int, ...]]]
+    reminders: dict[str, defaultdict[Incomplete, int]]
+    reminder_keys: dict[str, defaultdict[Incomplete, set[Incomplete]]]
     def __init__(self) -> None: ...
-    def run(self, input_dict, name): ...
-    def cache_clear(self, name=None) -> None: ...
-    def cache_info(self) -> tuple[dict[Incomplete, Incomplete], dict[Incomplete, Incomplete], dict[Incomplete, Incomplete]]: ...
-
-def wrap_numbers(input_dict, name: str): ...
+    def run(self, input_dict: dict[str, tuple[int, ...]], name: str) -> dict[str, tuple[int, ...]]: ...
+    def cache_clear(self, name: str | None = None) -> None: ...
+    def cache_info(
+        self,
+    ) -> tuple[
+        dict[str, dict[str, tuple[int, ...]]],
+        dict[str, defaultdict[Incomplete, int]],
+        dict[str, defaultdict[Incomplete, set[Incomplete]]],
+    ]: ...
+
+def wrap_numbers(input_dict: dict[str, tuple[int, ...]], name: str) -> dict[str, tuple[int, ...]]: ...
 def open_binary(fname: FileDescriptorOrPath) -> BinaryIO: ...
 def open_text(fname: FileDescriptorOrPath) -> io.TextIOWrapper: ...
-def cat(fname: FileDescriptorOrPath, fallback=..., _open=...): ...
-def bcat(fname: FileDescriptorOrPath, fallback=...): ...
+@overload
+def cat(fname: FileDescriptorOrPath, _open: Callable[[FileDescriptorOrPath], io.TextIOWrapper] = ...) -> str: ...
+@overload
+def cat(
+    fname: FileDescriptorOrPath, fallback: _T = ..., _open: Callable[[FileDescriptorOrPath], io.TextIOWrapper] = ...
+) -> str | _T: ...
+@overload
+def bcat(fname: FileDescriptorOrPath) -> str: ...
+@overload
+def bcat(fname: FileDescriptorOrPath, fallback: _T = ...) -> str | _T: ...
 def bytes2human(n: int, format: str = "%(value).1f%(symbol)s") -> str: ...
 def get_procfs_path() -> str: ...
 def decode(s: bytes) -> str: ...
-def term_supports_colors(file: SupportsWrite[str] = ...) -> bool: ...
+def term_supports_colors(file: SupportsWrite[str] = sys.stdout) -> bool: ...
 def hilite(s: str, color: str | None = None, bold: bool = False) -> str: ...
-def print_color(s: str, color: str | None = None, bold: bool = False, file: SupportsWrite[str] = ...) -> None: ...
+def print_color(s: str, color: str | None = None, bold: bool = False, file: SupportsWrite[str] = sys.stdout) -> None: ...
 def debug(msg: str | Exception) -> None: ...
 
 __all__ = [