In my code, I want to execute import filename for all the files that are present in a directory. I have a file abc.py under workspace. I am currently doing the following :
for filename in os.listdir(homedir/workspace) exec "import " + filename filename = eval(filename + '.function(variable)') The problem is that instead of doing import abc, it is doing import abc.py, and then showing the error no module named py
How can I resolve this?
Thanks in advance!
You can use os.path.splitext
os.path.splitext(filename)[0] Of the returned two element array, the first element is just the filename, the second element is the extension. To be extra safe, you could double check that you did indeed grab a .py file:
if os.path.splitext(filename)[1] == '.py': # do stuff To run .function(variable) for all Python files in a directory:
import os import sys from importlib import import_module dirpath = 'homedir/workspace' sys.path.append(dirpath) for filename in os.listdir(path) module_name, ext = os.path.splitext(filename) if ext == '.py': try: result = import_module(module_name).function(variable) except Exception as e: sys.stderr.write('error: %s: %s\n' % (filename, e)) continue else: print('%s: %s' % (filename, result)) 
exec. You can achieve exactly the same effect with:module = __import__(filename). Usingexecyou are opening a security issue (e.g. someone can name a fileos;os.sytem("killall cats")and when the code in your question executes all the cats will die! Using__import__you would only receive anImportError. Even better than__import__isimportlib.import_module, although it works in python2.7 and 3.1+ only.