前言
先让我们来看看一个用到相对文件路径的函数调用的问题。假设现在有两个脚本文件main.py和func.py,他们的路径关系是:
1.
2 |--dir1
3 |--main.py
4 |--dir2
5 |--func.py
6 |--test.txt
func.py的作用是提供load_txt()函数,读取同级目录下test.txt文件中的内容并返回。
1 # func.py
2 def load_txt()
3 filename = './test.txt'
4 return open(filenamem, 'r').read()
假设现在在main.py中调用load_txt()函数:
1 # main.py
2 from dir2 import func
3 if __name__ == '__main__':
4 print func.load_txt()
这个时候会报类似找不到文件test.txt的错误。
为什么会这样呢?这是因为在函数调用的过程中,当前路径.代表的是被执行的脚本文件的所在路径。在这个情况中,.表示的就是main.py的所在路径,所以load_txt()函数会在dir1文件夹中寻找test.txt文件。
那么怎么样才能在函数调用的过程中保持相对路径的不变呢?
方法
在网上有相当多的教程都有提到这个Python中相对文件路径的问题,但是大部分都没有提及到在这种情况下的解决办法。
在以下的三个函数中,第一个和第二个是大部分教程中的解决办法,但是这样是错误的,因为第一个和第二个函数所获取的"当前文件路径"都是被执行的脚本文件的所在路径,只有第三个函数返回的当前文件路径才是真正的、该函数所在的脚本文件的所在路径
1 def get_cur_path1():
2 import os
3 return os.path.abspath(os.curdir)
4 def get_cur_path2():
5 import sys
6 return sys.argv[0]
7 def get_cur_path3():
8 import os
9 return os.path.dirname(__file__)
因此,解决办法如下。修改func.py中的读取函数如下即可:
1 # func.py
2 import os
3 def load_txt()
4 module_path = os.path.dirname(__file__)
5 filename = modelu_path + '/test.txt'
6 return open(filenamem, 'r').read()
登录 | 立即注册