python: Ubah direktori kerja skrip ke direktori skrip itu sendiri

Saya menjalankan shell python dari crontab setiap menit:

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

/home/udi/foomemiliki beberapa subdirektori yang diperlukan, seperti /home/udi/foo/logdan /home/udi/foo/config, yang /home/udi/foo/bar.pymerujuk pada.

Masalahnya adalah crontabmenjalankan skrip dari direktori kerja yang berbeda, jadi mencoba membuka ./log/bar.loggagal.

Apakah ada cara yang bagus untuk memberitahu skrip untuk mengubah direktori kerja ke direktori skrip sendiri? Saya lebih suka solusi yang akan bekerja untuk lokasi skrip apa pun, daripada secara eksplisit memberitahu skrip di mana skrip itu berada.

EDIT:

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

Merupakan solusi elegan paling kompak. Terima kasih atas jawaban dan penjelasan Anda!

Ini akan mengubah direktori kerja Anda saat ini sehingga membuka jalur relatif akan berfungsi:

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

Namun, Anda bertanya bagaimana cara mengubah ke direktori apa pun skrip Python Anda berada, bahkan jika Anda tidak tahu direktori apa yang akan terjadi ketika Anda sedang menulis skrip Anda. Untuk melakukan ini, Anda dapat menggunakan os.pathfungsi:

import os

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

Ini mengambil nama file skrip Anda, mengonversinya menjadi jalur absolut, lalu mengekstrak direktori jalur itu, lalu berubah ke direktori itu.

Anda bisa mendapatkan versi yang lebih pendek dengan menggunakan sys.path[0].

os.chdir(sys.path[0])

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

Seperti diinisialisasi pada saat startup program, item pertama dari daftar ini path[0],, adalah direktori yang berisi skrip yang digunakan untuk memanggil juru bahasa Python

Jangan lakukan ini.

Skrip dan data Anda tidak boleh dihaluskan menjadi satu direktori besar. Menaruh kode Anda di beberapa lokasi yang diketahui ( site-packagesatau /var/opt/udiatau sesuatu) terpisah dari data Anda. Gunakan kontrol versi yang baik pada kode Anda untuk memastikan bahwa Anda memiliki versi saat ini dan sebelumnya terpisah satu sama lain sehingga Anda dapat kembali ke versi sebelumnya dan menguji versi yang akan datang.

Intinya: Jangan menyatukan kode dan data.

Data sangat berharga. Kode datang dan pergi.

Berikan direktori yang berfungsi sebagai nilai argumen baris perintah. Anda dapat memberikan default sebagai variabel lingkungan. Jangan menyimpulkannya (atau menebaknya)

Jadikan nilai argumen yang diperlukan dan lakukan ini.

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

Jangan "menganggap" direktori berdasarkan lokasi perangkat lunak Anda. Ini tidak akan berhasil dengan baik dalam jangka panjang.

Ubah perintah crontab Anda ke

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

The (...)dimulai sub-shell yang mengeksekusi crond Anda sebagai satu perintah. The || exit 1menyebabkan cronjob Anda gagal dalam hal ini direktori tidak tersedia.

Meskipun solusi lain mungkin lebih elegan dalam jangka panjang untuk skrip spesifik Anda, contoh saya masih bisa berguna dalam kasus di mana Anda tidak dapat memodifikasi program atau perintah yang ingin Anda jalankan.