如何在Python中获取当前模块中所有类的列表？

import inspect

def get_classes_in_module(module):
    classes = []
    for name, obj in inspect.getmembers(module):
        if inspect.isclass(obj):
            classes.append(obj)
    return classes

# 示例
class TestClass1:
    pass


class TestClass2:
    pass


if __name__ == "__main__":
    classes = get_classes_in_module(__import__(__name__))
    print(classes)