How to find the list of all the class name in a file in python?

Question

I am currently writing a python script to display all name of all the python files in a package and also all the name of the class a file contains.

scenario

#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

out put A.py,apple,banana

I can use an old convention way where i could check for each line whether "class" string to is present and retrieve the className. But i want to know is there is a better way to retrieve all the class Name ?

Ashwini Chaudhary · Accepted Answer · 2013-06-18 15:02:55Z

16

Something like this:

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

Another alternative as @user2033511 suggested:

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

edited Jun 18, 2013 at 15:02

answered Jun 18, 2013 at 14:49

Ashwini Chaudhary

252k60 gold badges478 silver badges519 bronze badges

Sign up to request clarification or add additional context in comments.

5 Comments

Ashwini Chaudhary Over a year ago

@SanjayT.Sharma oops! typo.

enpenax Over a year ago

And I would use the classname A instead of so :)

enpenax Over a year ago

Instead of types and isinstance one can use inspect.isclass instead, which will probably boil down to the same, but is easier to read, imho.

metmirr Over a year ago

is that possible to get classes by defination order? default is sorted by name.

Ashwini Chaudhary Over a year ago

@metmirr You can get the line number using inspect.getsourcelines and sort the list based on it.

Jon Clements · Accepted Answer · 2013-06-18 15:38:22Z

13

An alternative that will also find nested classes and doesn't require importing of the file:

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']

answered Jun 18, 2013 at 15:38

Jon Clements

143k34 gold badges254 silver badges288 bronze badges

Collectives™ on Stack Overflow

How to find the list of all the class name in a file in python?

2 Answers 2

5 Comments

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

5 Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Related