如何在python中获取当前用户和脚本的工作目录?

时间:2017-06-14 00:11:35

标签: python linux windows

目前,我正在使用以下代码段静态指定脚本源数据的文件路径:

 def get_files():
     global thedir
     thedir = 'C:\\Users\\username\\Documents'
     list = os.listdir(thedir)
     for i in list:
         if i.endswith('.txt'):
             print("\n\n"+i)
             eat_file(thedir+'\\'+i)

我静态分配位置的原因是,当脚本在Eclipse&等调试环境中执行时,脚本无法正确执行。 Visual Studio代码。这些调试器假定脚本正在从其工作目录运行。

由于我无法修改可能正在运行此脚本的每个系统的本地设置,是否有推荐的模块强制脚本收集活动用户(linux和windows)和/或脚本的本地目录务实?

1 个答案:

答案 0 :(得分:2)

new-ish pathlib module(在Python> = 3.4中提供)非常适合处理类似路径的对象(Windows和其他操作系统)。用它。不要打扰过时的os模块。并且不要试图使用裸字符串来表示类似路径的对象。

It's Paths - Paths all the way down

简化:您可以构建任何路径(目录和文件路径对象被视为完全相同)作为对象,该对象可以是绝对路径对象相对路径对象

简单显示一些有用的路径 - 例如当前工作目录和用户主页 - 如下所示:

from pathlib import Path

# Current directory (relative):
cwd = Path() # or Path('.')
print(cwd)

# Current directory (absolute):
cwd = Path.cwd()
print(cwd)

# User home directory:
home = Path.home()
print(home)

# Something inside the current directory
file_path = Path('some_file.txt') # relative path; or 
file_path = Path()/'some_file.txt' # also relative path
file_path = Path().resolve()/Path('some_file.txt') # absolute path
print(file_path)

要在文件树中向下导航,您可以执行以下操作。请注意,第一个对象homePath,其余的只是字符串:

file_path = home/'Documents'/'project'/'data.txt' # or
file_path = home.join('Documents', 'project', 'data.txt')

要阅读位于路径的文件,您可以使用其open方法而不是open函数:

with file_path.open() as f:
    dostuff(f)

但你也可以直接抓住文字!

contents = file_path.read_text()
content_lines = contents.split('\n')

...并直接写文字!

data = '\n'.join(content_lines)
file_path.write_text(data) # overwrites existing file

通过这种方式检查它是文件还是目录(并且存在):

file_path.is_dir() # False
file_path.is_file() # True

创建一个新的空文件而不像这样打开它(静默替换任何现有文件):

file_path.touch()

要使文件仅在不存在时,请使用exist_ok=False

try:
    file_path.touch(exist_ok=False)
except FileExistsError:
    # file exists

创建一个新目录(在当前目录下,Path()),如下所示:

Path().mkdir('new/dir') # get errors if Path()/`new` doesn't exist
Path().mkdir('new/dir', parents=True) # will make Path()/`new` if it doesn't exist
Path().mkdir('new/dir', exist_ok=True) # errors ignored if `dir` already exists

以这种方式获取路径的文件扩展名或文件名:

file_path.suffix # empty string if no extension
file_path.stem # note: works on directories too

对路径的整个最后部分使用name(如果它们在那里,则为词干和扩展名):

file_path.name # note: works on directories too

使用with_name方法重命名文件(返回相同的路径对象但使用新文件名):

new_path = file_path.with_name('data.txt')

您可以使用iterdir

遍历目录中的所有“内容”
all_the_things = list(Path().iterdir()) # returns a list of Path objects