更改目录后找不到模块

时间:2019-09-10 12:08:46

标签: python python-os python-importlib

因此,我必须创建一个运行文件夹,即一个包含一堆需要运行的python文件的文件夹。我能够轻松创建此文件夹,并且所有文件都在那里。但是,当我尝试使用importlib运行文件时,python找不到它。

我想确保正在运行的代码实际上是运行文件夹代码,因此我将目录更改为运行文件夹的位置。

abs_path = os.path.abspath("{}".format(run_location))
os.chdir(abs_path)
files = [f for f in os.listdir('.') if os.path.isfile(f)]
print(files)
try:
    driver_module = importlib.import_module("main_driver.py")
    driver_module.main(config, logger)
except Exception as e:
    logger.error("error", str(e))
    finish_fail(config, logger)
finish_success(config, logger)

在上面的示例中,我只想运行main_driver.py。上面的输出是:

['PrepareDataframe.py', 'categorical_encoding.py', 'extra_files.zip', 'build_features.py', 'spot_extractor.py', 'dev.ini', 'featuriser.py', 'main_driver.py', 'time_features_extract.py']

很显然,main_driver.py在当前工作目录中,但是出现此错误。

No module named 'main_driver'

跟踪:

Traceback (most recent call last):
  File "./utils/submit.py", line 292, in <module>
    driver_module = importlib.import_module("main_driver")
  File "/usr/local/Cellar/python/3.7.4_1/Frameworks/Python.framework/Versions/3.7/lib/python3.7/importlib/__init__.py", line 127, in import_module
    return _bootstrap._gcd_import(name[level:], package, level)
  File "<frozen importlib._bootstrap>", line 1006, in _gcd_import
  File "<frozen importlib._bootstrap>", line 983, in _find_and_load
  File "<frozen importlib._bootstrap>", line 965, in _find_and_load_unlocked
ModuleNotFoundError: No module named 'main_driver'

1 个答案:

答案 0 :(得分:1)

这似乎是一个PYTHONPATH问题。

运行Python交互式解释器时,sys.path中的第一项是空字符串,实际上表示“当前目录”。因此,每次尝试导入模块时,它都会首先在当前目录中查找。

但是,如果您运行python myscript.py,则不是这样-sys.path中的第一项是您运行脚本时所在的实际目录。因此,如果使用os.chdir()来更改当前目录,则新的当前目录不在sys.path中,因此import_module('foo')会因为找不到而失败。

如果您想模仿总是在当前目录中查找模块的交互式python行为,请将sys.path.insert(0, '')添加到脚本中。