從同一模塊中的類名字符串中獲取 Python 類對象

Question

我有一堂課

class Foo():
    def some_method():
        pass

和同一模塊中的另一個類：

class Bar():
    def some_other_method():
        class_name = "Foo"
        # Can I access the class Foo above using the string "Foo"?

我希望能夠使用字符串“Foo”訪問Foo類。

如果我在另一個模塊中，我可以使用以下方法執行此操作：

from project import foo_module
foo_class = getattr(foo_module, "Foo")

我可以在同一個模塊中做同樣的事情嗎？

IRC 中的人建議我使用映射字典將字符串類名映射到類，但如果有更簡單的方法，我不想這樣做。

Answer 1

import sys
getattr(sys.modules[__name__], "Foo")

# or 

globals()['Foo']

Answer 2

您可以在sys模塊的幫助下完成：

import sys

def str2Class(str):
    return getattr(sys.modules[__name__], str)

Answer 3

globals()[class_name]

請注意，如果這不是絕對必要的，您可能需要重構代碼以不使用它。