python: Thay đổi thư mục làm việc của tập lệnh thành thư mục riêng của tập lệnh

Tôi chạy một vỏ trăn từ crontab mỗi phút:

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

/home/udi/foocó một số thư mục con cần thiết, như /home/udi/foo/logvà /home/udi/foo/config, trong đó /home/udi/foo/bar.pyđề cập đến.

Vấn đề là crontabchạy tập lệnh từ một thư mục làm việc khác, vì vậy cố gắng mở ./log/bar.logthất bại.

Có cách nào hay để nói kịch bản thay đổi thư mục làm việc thành thư mục riêng của tập lệnh không? Tôi sẽ thích một giải pháp sẽ hoạt động cho bất kỳ vị trí tập lệnh nào, thay vì nói rõ ràng kịch bản đó ở đâu.

BIÊN TẬP:

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

Là giải pháp thanh lịch nhỏ gọn nhất. Cảm ơn câu trả lời và giải thích của bạn!

Điều này sẽ thay đổi thư mục làm việc hiện tại của bạn để mở các đường dẫn tương đối sẽ hoạt động:

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

Tuy nhiên, bạn đã hỏi làm thế nào để thay đổi vào bất kỳ thư mục nào mà tập lệnh Python của bạn được đặt, ngay cả khi bạn không biết thư mục đó sẽ là gì khi bạn viết tập lệnh của mình. Để làm điều này, bạn có thể sử dụng các os.pathchức năng:

import os

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

Điều này lấy tên tệp của tập lệnh của bạn, chuyển đổi nó thành một đường dẫn tuyệt đối, sau đó trích xuất thư mục của đường dẫn đó, sau đó thay đổi vào thư mục đó.

Bạn có thể có được một phiên bản ngắn hơn bằng cách sử dụng sys.path[0].

os.chdir(sys.path[0])

Từ http://docs.python.org/l Library / sys.html # sys.path

Như được khởi tạo khi khởi động chương trình, mục đầu tiên của danh sách này path[0], là thư mục chứa tập lệnh được sử dụng để gọi trình thông dịch Python

Đừng làm điều này.

Tập lệnh và dữ liệu của bạn không nên được trộn vào một thư mục lớn. Đặt mã của bạn trong một số vị trí đã biết ( site-packageshoặc /var/opt/udihoặc một cái gì đó) tách rời khỏi dữ liệu của bạn. Sử dụng kiểm soát phiên bản tốt trên mã của bạn để đảm bảo rằng bạn có các phiên bản hiện tại và trước đó tách biệt với nhau để bạn có thể quay lại các phiên bản trước và thử nghiệm các phiên bản trong tương lai.

Dòng dưới cùng: Không trộn lẫn mã và dữ liệu.

Dữ liệu là quý giá. Mã đến rồi đi.

Cung cấp thư mục làm việc như một giá trị đối số dòng lệnh. Bạn có thể cung cấp một mặc định như một biến môi trường. Đừng suy diễn nó (hoặc đoán nó)

Làm cho nó một giá trị đối số cần thiết và làm điều này.

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

Không "giả sử" một thư mục dựa trên vị trí của phần mềm của bạn. Nó sẽ không hoạt động tốt trong thời gian dài.

Thay đổi lệnh crontab của bạn thành

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

Khởi động (...)một lớp vỏ con mà crond của bạn thực thi như một lệnh đơn. Các || exit 1nguyên nhân cronjob của bạn không thành công trong trường hợp thư mục không có sẵn.

Mặc dù về lâu dài các giải pháp khác có thể thanh lịch hơn đối với các tập lệnh cụ thể của bạn, ví dụ của tôi vẫn có thể hữu ích trong trường hợp bạn không thể sửa đổi chương trình hoặc lệnh mà bạn muốn thực thi.