2013-03-24 175 views
0

我想要做的是创建一个设置为主文件夹脚本的变量homedir。我该怎么做呢?就像说脚本位于C:\ blah \,我会希望将homedir分配给它。将目录设置为脚本路径

homedir = (current directory) 
pdfList = [] 

def getfiles(): 
    directory = os.listdir(homedir) 
    for file in directory: 
     if "pdf" in file: 
      pdfList.append(file) 

回答

0
homedir = '.' 

import os 
homedir = os.getcwd() 

将导致os.listdir(homedir)列出在当前工作目录中的文件(和子目录)(即,从执行该脚本的目录)。请注意,这是而不是必须与包含该脚本的目录相同。对于目录使用

import os 
homedir = os.path.dirname(__file__) 
5

__file__模块属性包含脚本位置,但它可能是相对的。下面的文件是c:\test\x.py

import os 
print(__file__) 
homedir = os.path.abspath(os.path.dirname(__file__)) 
print(homedir) 

输出:

.\x.py 
c:\test 
相关问题