Import class dynamically

[voltron]

I have two base classes FirstBaseClass and SecondBaseClass and lot of theirs subclasses. Each subclass defined in the separate file, location of these files is known. I need to import and instantiate subclasses programmatically. Now I use following approach:

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16

import inspect import importlib   class_name = "FirstChildClass" file_path = "/path/to/class.py"   try:     spec = importlib.util.spec_from_file_location(class_name, file_path)     module = importlib.util.module_from_spec(spec)     spec.loader.exec_module(module)     for x in dir(module):         obj = getattr(module, x)         if inspect.isclass(obj) and (issubclass(obj, FirstBaseClass) or issubclass(obj, SecondBaseClass)) and obj.__name__ == class_name:             return obj() except ImportError as e:     print(e)

But problem is that I need to know class name. Is it possible to import class from file without knowing its name, assuming that only one class defined in that file?

[Gribouillis]

But problem is that I need to know class name.

If there is only one class, only one object will match inspect.isclass(obj) so you don't need to know the class name. Is it your code?

[voltron]

If there is only one class, only one object will match inspect.isclass(obj) so you don't need to know the class name. Is it your code?

Maybe I'm doing something wrong, but this does not work. In class code I have something like this

1 2 3 4 5 6 7 8 9

import os from program.package.FirstBaseClass import FirstBaseClass from program.package.SomeClass import SomeClass ... (more imports here)   class FirstChildClass(FirstBaseClass):     ...     (implementation here)

When I run import code without class name check, e.g. just inspect.isclass(obj) it returns FirstBaseClass, then SomeClass... Class defined in the code returned in the end.

[Gribouillis]

You can use eg inspect.getmodule(obj), or inspect.getsourcefile(obj) to differentiate between classes defined here and imported classes.

[voltron]

Using inspect.getmodule(obj) and inspect.getsourcefile(obj) does not worked for me, but I managed to solve this by checking class name like this

1 2	if inspect.isclass(obj) and issubclass(obj, (FirstBaseClass, SecondBaseClass)) and obj.__name__ not in ("FirstBaseClass", "SecondBaseClass"):     # do something

Thanks a lot for your help!