CompletedProcess 是 mypy 的泛型

Question

我用作subprocess.CompletedProcess函数的返回类型。当运行mypy时--strict，它会抱怨Missing type parameters for generic type "CompletedProcess"。这是出现问题的一个简短示例：

import subprocess

def run_command() -> subprocess.CompletedProcess:
    return subprocess.CompletedProcess(args=[], returncode=0, stdout='')

我不明白如何CompletedProcess是通用的。我检查了参考文献（https://docs.python.org/3.9/library/subprocess.html#subprocess.CompletedProcess），但找不到任何提及。

例如，随机添加通用参数-> subprocess.CompletedProcess[str]可以消除错误，但显然这不是重点。我希望能够在不定义泛型参数的情况下提示返回类型。

我正在使用 python 3.8.5 和 mypy 0.761 和 0.910 （两者给出相同的输出）。

Answer 1

子进程的输出 ( stdout/ ) 可以是或，具体取决于参数。需要通用参数 to来指示 this 的类型。stderrstrbytesCompletedProcess

import subprocess

def run_some_command() -> 'subprocess.CompletedProcess[str]':
    """Run some subprocess that captures output as ``str``"""
    return subprocess.CompletedProcess(args=[], returncode=0, stdout='')

def run_other_command() -> 'subprocess.CompletedProcess[bytes]':
    """Run some subprocess that captures output as ``bytes``"""
    return subprocess.CompletedProcess(args=[], returncode=0, stdout=b'')

reveal_type(run_some_command().stdout)  # note: Revealed type is 'builtins.str*'

由于只有类型提示subprocess.CompletedProcess是通用的，而运行时类型不是通用的，因此必须用引号引起来。

标准库的类型声明是typeshed. 具体来说，CompletedProcess目前定义如下：

class CompletedProcess(Generic[_T]):
    # morally: _CMD
    args: Any
    returncode: int
    # These are really both Optional, but requiring checks would be tedious
    # and writing all the overloads would be horrific.
    stdout: _T
    stderr: _T
    def __init__(self, args: _CMD, returncode: int, stdout: Optional[_T] = ..., stderr: Optional[_T] = ...) -> None: ...
    def check_returncode(self) -> None: ...
    if sys.version_info >= (3, 9):
        def __class_getitem__(cls, item: Any) -> GenericAlias: ...