Question

我有一个基于我传入的类返回动态类型的方法：

def foo(cls):
    return cls()

如何设置此功能的输入？

Answer 1

阅读本文https://blog.yuo.be/2016/05/08/python-3-5-getting-to-grips-with-type-hints/后，我自己找到了解决方案：

from typing import TypeVar, Type

class A:

    def a(self):
        return 'a'


class B(A):

    def b(self):
        return 'b'


T = TypeVar('T')


def foo(a: T) -> T:
    return a()

这个模板适合我上面的问题，但实际上，我的需求有点不同，我需要更多工作。下面我包括我的问题和解决方案：

问题：我想使用with这样的关键字：

with open_page(PageX) as page:
    page.method_x() # method x is from PageX

<强>解决方案

from typing import TypeVar, Type, Generic

T = TypeVar('T')

def open_page(cls: Type[T]):
    class __F__(Generic[T]):

        def __init__(self, cls: Type[T]):
            self._cls = cls

        def __enter__(self) -> T:
            return self._cls()

        def __exit__(self, exc_type, exc_val, exc_tb):
            pass

    return __F__(cls)

因此，当我使用PyCharm时，当我将method_x传递到PageX

时，它能够建议with open_page(PageX) as page:

Python根据参数动态返回输入

1 个答案: