python: Altere o diretório de trabalho dos scripts para o próprio diretório do script

Eu corro um shell python do crontab a cada minuto:

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

/home/udi/foopossui alguns subdiretórios necessários, como /home/udi/foo/loge /home/udi/foo/config, ao qual /home/udi/foo/bar.pyse refere.

O problema é que crontabo script é executado a partir de um diretório de trabalho diferente, portanto, a tentativa de abrir ./log/bar.logfalha.

Existe uma boa maneira de dizer ao script para alterar o diretório de trabalho para o próprio diretório do script? Gostaria de uma solução que funcionasse para qualquer local de script, em vez de dizer explicitamente onde está o script.

EDITAR:

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

Foi a solução elegante mais compacta. Obrigado por suas respostas e explicações!

Isso mudará seu diretório de trabalho atual para, para que a abertura de caminhos relativos funcione:

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

No entanto, você perguntou como mudar para qualquer diretório em que seu script Python esteja localizado, mesmo que você não saiba qual diretório será esse quando estiver escrevendo seu script. Para fazer isso, você pode usar as os.pathfunções:

import os

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

Isso pega o nome do arquivo do seu script, converte-o para um caminho absoluto, extrai o diretório desse caminho e depois muda para esse diretório.

Você pode obter uma versão mais curta usando sys.path[0].

os.chdir(sys.path[0])

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

Conforme inicializado na inicialização do programa, o primeiro item desta lista path[0], é o diretório que contém o script usado para chamar o interpretador Python.

Não faça isso.

Seus scripts e seus dados não devem ser compactados em um grande diretório. Coloque seu código em algum local conhecido ( site-packagesou /var/opt/udiou algo assim) separar de seus dados. Use um bom controle de versão no seu código para garantir que as versões atual e anterior sejam separadas uma da outra, para que você possa voltar às versões anteriores e testar versões futuras.

Conclusão: não misture código e dados.

Dados são preciosos. O código vem e vai.

Forneça o diretório ativo como um valor do argumento da linha de comandos. Você pode fornecer um padrão como uma variável de ambiente. Não deduza (ou adivinhe)

Crie um valor de argumento necessário e faça isso.

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

Não "assuma" um diretório com base na localização do seu software. Não vai funcionar bem a longo prazo.

Mude seu comando crontab para

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

O (...)inicia um sub-shell que sua crond executa como um único comando. O || exit 1faz com que seu cronjob a falhar no caso de que o diretório não está disponível.

Embora as outras soluções possam ser mais elegantes a longo prazo para seus scripts específicos, meu exemplo ainda pode ser útil nos casos em que você não pode modificar o programa ou comando que deseja executar.