python:将脚本工作目录更改为脚本自己的目录

Question

我每分钟从crontab运行一个python shell:

* * * * * /home/udi/foo/bar.py

/home/udi/foo有一些必要的子目录,比如/home/udi/foo/log和/home/udi/foo/config,/home/udi/foo/bar.py指的是.

问题是crontab从不同的工作目录运行脚本,因此尝试打开./log/bar.log失败.

有没有一种很好的方法告诉脚本将工作目录更改为脚本自己的目录？我想要一个适用于任何脚本位置的解决方案,而不是明确地告诉脚本它在哪里.

编辑:

os.chdir(os.path.dirname(sys.argv[0]))

是最紧凑的优雅解决方案.感谢您的回答和解释!

Answer 1

这会将您当前的工作目录更改为打开相对路径:

import os
os.chdir("/home/udi/foo")

但是,您询问如何更改为Python脚本所在的目录,即使您不知道编写脚本时将使用哪个目录.为此,您可以使用以下os.path功能:

import os

abspath = os.path.abspath(__file__)
dname = os.path.dirname(abspath)
os.chdir(dname)

这将获取脚本的文件名,将其转换为绝对路径,然后提取该路径的目录,然后更改为该目录.

Answer 2

你可以使用更短的版本sys.path[0].

os.chdir(sys.path[0])

来自http://docs.python.org/library/sys.html#sys.path

在程序启动时初始化,此列表的第一项 path[0]是包含用于调用Python解释器的脚本的目录.

Answer 3

不要这样做.

您的脚本和数据不应该被嵌入到一个大目录中.将代码放在与数据分开的某个已知位置(site-packages或某个位置/var/opt/udi).对代码使用良好的版本控制,以确保您将当前版本和先前版本相互分离,以便您可以回退到以前的版本并测试将来的版本.

底线:不要混淆代码和数据.

数据很珍贵.代码来来去去.

提供工作目录作为命令行参数值.您可以提供默认值作为环境变量.不要推断它(或猜测它)

使其成为必需的参数值并执行此操作.

import sys
import os
working= os.environ.get("WORKING_DIRECTORY","/some/default")
if len(sys.argv) > 1: working = sys.argv[1]
os.chdir( working )

不要"假设"基于软件位置的目录.从长远来看,它不会很好.

Answer 4

将您的crontab命令更改为

* * * * * (cd /home/udi/foo/ || exit 1; ./bar.py)

该(...)启动子shell,你的crond的执行作为一个单一的命令.将|| exit 1导致你的cronjob,该目录是不可用的情况下失败.

虽然从长远来看,其他解决方案对于特定脚本可能更为优雅,但在无法修改要执行的程序或命令的情况下,我的示例仍然有用.