python - 在 Python 中，如何获取特定文件中定义的类列表？

Question

如果文件 myfile.py 包含:

class A(object):
  # Some implementation

class B (object):
  # Some implementation

如何定义一个方法，以便在给定 myfile.py 的情况下返回[A, B]?

这里，A 和 B 的返回值可以是类的名称，也可以是类的类型。

(即 type(A) = type(str) 或 type(A) = type(type))

Answer 1

你可以同时获得:

import importlib, inspect
for name, cls in inspect.getmembers(importlib.import_module("myfile"), inspect.isclass):

您可能还想检查:

if cls.__module__ == 'myfile'