python - 如何在python中找到文件中所有类名的列表？

Question

我目前正在编写一个 python 脚本来显示包中所有 python 文件的所有名称以及文件包含的所有类的名称。

设想

#A.py

class apple:
      .
      .
class banana:
      .
      .

# Extracting_FilesName_className.py

 f=open("c:/package/A.py','r')     
 className=[]
 "Here i need to use some command to get all the class name within the file A.py and append it to className'  

  print file_name,className 

输出 A.py,apple,banana

我可以使用旧的约定方式，我可以检查每一行是否存在“类”字符串并检索类名。但我想知道是否有更好的方法来检索所有类名？

Answer 1

像这样的东西：

>>> from types import ClassType
>>> import A
>>> classes = [x for x in dir(A) if isinstance(getattr(A, x), ClassType)]

@user2033511 建议的另一种选择：

>>> from inspect import isclass
>>> classes = [x for x in dir(A) if isclass(getattr(A, x))]

Answer 2

也可以找到嵌套类并且不需要导入文件的替代方法：

source = """
class test:
    class inner_class:
        pass
    pass

class test2:
    pass
"""

import ast
p = ast.parse(source)
classes = [node.name for node in ast.walk(p) if isinstance(node, ast.ClassDef)]
# ['test', 'test2', 'inner_class']