时间戳

首页 > 解决方案 > CompletedProcess 是 mypy 的泛型

python - CompletedProcess 是 mypy 的泛型

问题描述

我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（都给出相同的输出）。

标签： pythonsubprocessmypy

解决方案

子进程的输出 ( stdout/ stderr) 可以是str或bytes，具体取决于参数。需要泛型参数 toCompletedProcess来指示 this 的类型。

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: ...

推荐阅读