python3,在子文件夹中导入模块时目录不正确

时间:2017-02-27 07:48:02

标签: python-3.x module getcwd

我有一个名为'test'的主文件夹,内部结构是:

# folders and files in the main folder 'test'
Desktop\test\use_try.py
Desktop\test\cond\__init__.py   # empty file.
Desktop\test\cond\tryme.py
Desktop\test\db\

现在在tryme.py文件中。我想在'db'

文件夹中生成一个文件
# content in the file of tryme.py
import os

def main():
    cwd = os.getcwd()    # the directory of the folder 'Desktop\test\cond'
    folder_test = cwd[:-4]   # -4 since 'cond' has 4 letters
    folder_db = folder_test + 'db/'  # the directory of folder 'db'

    with open(folder_db + 'db01.txt', 'w') as wfile:
        wfile.writelines(['This is a test.'])

if __name__ == '__main__':
    main()

如果我直接运行此文件没问题,则文件'db01.txt'位于'db'文件夹中。 但是,如果我运行use_try.py文件,它将无法正常工作。

# content in the file of use_try.py
from cond import tryme

tryme.main()

我得到的错误是指tryme.py文件。在'打开......'的命令中

FileNotFoundError: [Error 2] No such file or directory: 'Desktop\db\db01.txt'

似乎是代码

'os.getcwd()' 

仅指调用tryme.py文件的文件,而不是tryme.py文件本身。

你知道如何修复它,以便我可以使用文件use_try.py在'db'文件夹中生成'db01.txt'吗?我正在使用Python3

由于

2 个答案:

答案 0 :(得分:2)

似乎您需要 working directory,而是tryme.py文件的目录。

这可以使用__file__魔法来解决:

curdir = os.path.dirname(__file__)

答案 1 :(得分:0)

使用环境变量中的绝对文件名,或者期望db /目录是当前工作目录的子目录。

此行为符合预期。当前工作目录是您调用代码的位置,而不是代码存储的位置。

folder_test = cwd   # assume working directory will have the db/ subdir

或     folder_test = os.getEnv(' TEST_DIR')#use $ {TEST_DIR} / db /