importing files in python code example

Example 1: import from another file python

----FILE_1----
def mult(a, b):
  return a * b

----FILE_2----
from FILE_1 import mult

print(mult(2, 4))
# >>> 8

Example 2: how to import all functions from another python file

from scripts import *

Example 3: python3 import all files in directory

from inspect import isclass
from pkgutil import iter_modules
from pathlib import Path
from importlib import import_module

# iterate through the modules in the current package
package_dir = Path(__file__).resolve().parent
for (_, module_name, _) in iter_modules([package_dir]):

    # import the module and iterate through its attributes
    module = import_module(f"{__name__}.{module_name}")
    for attribute_name in dir(module):
        attribute = getattr(module, attribute_name)

        if isclass(attribute):            
            # Add the class to this package's variables
            globals()[attribute_name] = attribute

Tags: