在 Python 中使用 typing.get_type_hints 判断在另一个方法类中定义的类的前向引用

发布于05月25日

对于将前向引用作为字符串的类，我在调用tying.get_type_hints()时遇到了问题.我的代码可以在函数内部未定义的情况下运行.我在下面的Python3.10中复制了一个最小的例子:

import typing
class B:
  pass
class A:
  some_b: "B" 
print(typing.get_type_hints(A)) # prints {'some_b': <class '__main__.B'>}

import typing
def func():
  class B:
    pass
  class A:
    some_b: "B" 
  print(typing.get_type_hints(A)) 
func() # NameError: name 'B' is not defined

这是预期中的行为吗？有没有办法绕过这个问题，并确保对字符串的正向引用在正确的作用域中求值？

推荐答案

typing.get_type_hints允许您显式地传递本地命名空间，以用于通过localns参数解析引用.

from typing import get_type_hints


def func():
    class A:
        some_b: "B"

    class B:
        pass

    print(get_type_hints(A, localns=locals()))


func()

输出:{'some_b': <class '__main__.func.<locals>.B'>}

请参阅locals英镑的文档.

附注:通过使用延迟的批注求值(PEP 563)，您可以省略引号:

from __future__ import annotations
from typing import get_type_hints


def func():
    class A:
        some_b: B

    class B:
        pass

    print(get_type_hints(A, localns=locals()))