python - 装饰器更改返回类型时键入函数

Question

如何正确编写返回类型被装饰器修改的函数的类型？

简单的例子:

def example_decorator(fn):
    def wrapper(data):
        res = fn(data)
        return ', '.join(res)

    return wrapper


@example_decorator
def func(data: list):  # -> ???? str ? list ?
    new_data = data
    new_data.append('XYZ')
    return new_data


# When we type func -> list

def test() -> str:
    result = func(['ABC', 'EFG'])
    print(type(result))  # <class 'str'>
    return result  # Incompatible return type [7]: Expected str but got list.


test()

Answer 1

可能是类型检查器问题。解决方案如下。使用 mypy 没问题，但 pycharm 和 pyre-check 正在提示。

from typing import *

def example_decorator(
    fn: Callable[[List[str]], List[str]]
) -> Callable[[List[str]], str]:
    def wrapper(data: List[str]) -> str:
        res = fn(data)
        return ', '.join(res)

    return wrapper


@example_decorator
def func(data: List[str]) -> List[str]:  
    data.append('XYZ')
    return data


def test() -> str:
    result = func(['ABC', 'EFG'])
    print(type(result))  # <class 'str'>
    return result


test()