Bagaimana cara menentukan direktori skrip saat ini?

Saya ingin melihat apa cara terbaik untuk menentukan direktori skrip saat ini di python?

Saya menemukan bahwa, karena banyak cara memanggil kode python, sulit untuk menemukan solusi yang baik.

Berikut ini beberapa masalah:

• __file__tidak ditentukan jika skrip dieksekusi dengan exec,execfile
• __module__ didefinisikan hanya dalam modul

Gunakan kasing:

• ./myfile.py
• python myfile.py
• ./somedir/myfile.py
• python somedir/myfile.py
• execfile('myfile.py') (dari skrip lain, yang dapat ditemukan di direktori lain dan yang dapat memiliki direktori saat ini.

Saya tahu bahwa tidak ada solusi yang sempurna, tetapi saya sedang mencari pendekatan terbaik yang memecahkan sebagian besar kasus.

Pendekatan yang paling banyak digunakan adalah os.path.dirname(os.path.abspath(__file__))tetapi ini benar-benar tidak berhasil jika Anda menjalankan skrip dari yang lain dengan exec().

Peringatan

Solusi apa pun yang menggunakan direktori saat ini akan gagal, ini bisa berbeda berdasarkan pada cara skrip dipanggil atau dapat diubah di dalam skrip yang sedang berjalan.

os.path.dirname(os.path.abspath(__file__))

memang yang terbaik yang akan Anda dapatkan.

Tidak biasa mengeksekusi skrip dengan exec/ execfile; biasanya Anda harus menggunakan infrastruktur modul untuk memuat skrip. Jika Anda harus menggunakan metode ini, saya sarankan pengaturan __file__di globalsAnda meneruskan ke skrip sehingga dapat membaca nama file itu.

Tidak ada cara lain untuk mendapatkan nama file dalam kode sebelumnya: seperti yang Anda perhatikan, CWD mungkin berada di tempat yang sama sekali berbeda.

Jika Anda benar-benar ingin membahas kasus dimana sebuah skrip dipanggil execfile(...), Anda dapat menggunakan inspectmodul untuk menyimpulkan nama file (termasuk path). Sejauh yang saya ketahui, ini akan berfungsi untuk semua kasus yang Anda daftarkan:

filename = inspect.getframeinfo(inspect.currentframe()).filename
path = os.path.dirname(os.path.abspath(filename))

#!/usr/bin/env python
import inspect
import os
import sys

def get_script_dir(follow_symlinks=True):
    if getattr(sys, 'frozen', False): # py2exe, PyInstaller, cx_Freeze
        path = os.path.abspath(sys.executable)
    else:
        path = inspect.getabsfile(get_script_dir)
    if follow_symlinks:
        path = os.path.realpath(path)
    return os.path.dirname(path)

print(get_script_dir())

Ini bekerja pada CPython, Jython, Pypy. Ini berfungsi jika skrip dieksekusi menggunakan execfile()( sys.argv[0]dan __file__solusi berbasis akan gagal di sini). Ini berfungsi jika skrip berada di dalam file zip yang dapat dieksekusi (/ telur) . Ini berfungsi jika skrip "diimpor" ( PYTHONPATH=/path/to/library.zip python -mscript_to_run) dari file zip; mengembalikan jalur arsip dalam kasus ini. Ini berfungsi jika skrip dikompilasi menjadi executable mandiri ( sys.frozen). Ini berfungsi untuk symlink ( realpathmenghilangkan tautan simbolik). Ia bekerja dalam interpreter interaktif; mengembalikan direktori kerja saat ini dalam kasus ini.

Dalam Python 3.4+ Anda dapat menggunakan pathlibmodul yang lebih sederhana :

from inspect import currentframe, getframeinfo
from pathlib import Path

filename = getframeinfo(currentframe()).filename
parent = Path(filename).resolve().parent

The os.path...pendekatan adalah 'hal yang dilakukan' dengan Python 2.

Di Python 3, Anda dapat menemukan direktori skrip sebagai berikut:

from pathlib import Path
cwd = Path(__file__).parents[0]

Cukup gunakan os.path.dirname(os.path.abspath(__file__))dan periksa dengan sangat hati-hati apakah ada kebutuhan nyata untuk kasus di mana execdigunakan. Ini bisa menjadi tanda desain bermasalah jika Anda tidak dapat menggunakan skrip Anda sebagai modul.

Ingatlah Zen dari Python # 8 , dan jika Anda yakin ada argumen yang bagus untuk kasus penggunaan di mana harus digunakan exec, maka silakan beri tahu kami beberapa detail lebih lanjut tentang latar belakang masalah.

Akan

import os
cwd = os.getcwd()

lakukan apa yang kamu inginkan? Saya tidak yakin apa yang Anda maksud dengan "direktori skrip saat ini". Apa yang diharapkan dari output untuk kasus penggunaan yang Anda berikan?

Pertama .. beberapa kasus penggunaan hilang di sini jika kita berbicara tentang cara menyuntikkan kode anonim ..

code.compile_command()
code.interact()
imp.load_compiled()
imp.load_dynamic()
imp.load_module()
__builtin__.compile()
loading C compiled shared objects? example: _socket?)

Tetapi, pertanyaan sebenarnya adalah, apa tujuan Anda - apakah Anda mencoba untuk menegakkan semacam keamanan? Atau apakah Anda hanya tertarik pada apa yang dimuat.

Jika Anda tertarik pada keamanan , nama file yang diimpor melalui exec / execfile tidak penting - Anda harus menggunakan rexec , yang menawarkan hal berikut:

Modul ini berisi kelas RExec, yang mendukung r_eval (), r_execfile (), r_exec (), dan metode r_import (), yang merupakan versi terbatas dari fungsi standar Python eval (), execfile () dan pernyataan exec dan impor. Kode yang dijalankan di lingkungan terbatas ini hanya akan memiliki akses ke modul dan fungsi yang dianggap aman; Anda dapat subkelas RExec menambah atau menghapus kemampuan yang diinginkan.

Namun, jika ini lebih merupakan pengejaran akademis .. berikut adalah beberapa pendekatan konyol yang mungkin bisa Anda gali lebih dalam ..

Contoh skrip:

./deep.py

print ' >> level 1'
execfile('deeper.py')
print ' << level 1'

./deeper.py

print '\t >> level 2'
exec("import sys; sys.path.append('/tmp'); import deepest")
print '\t << level 2'

/tmp/deepest.py

print '\t\t >> level 3'
print '\t\t\t I can see the earths core.'
print '\t\t << level 3'

./codespy.py

import sys, os

def overseer(frame, event, arg):
    print "loaded(%s)" % os.path.abspath(frame.f_code.co_filename)

sys.settrace(overseer)
execfile("deep.py")
sys.exit(0)

Keluaran

loaded(/Users/synthesizerpatel/deep.py)
>> level 1
loaded(/Users/synthesizerpatel/deeper.py)
    >> level 2
loaded(/Users/synthesizerpatel/<string>)
loaded(/tmp/deepest.py)
        >> level 3
            I can see the earths core.
        << level 3
    << level 2
<< level 1

Tentu saja, ini adalah cara yang intensif sumber daya untuk melakukannya, Anda akan melacak semua kode Anda .. Tidak terlalu efisien. Tapi, saya pikir ini pendekatan baru karena terus bekerja bahkan ketika Anda masuk lebih dalam ke sarang. Anda tidak dapat mengganti 'eval'. Meskipun Anda bisa menimpa execfile ().

Catatan, pendekatan ini hanya mencakup exec / execfile, bukan 'impor'. Untuk pemuatan kait 'modul' tingkat tinggi, Anda mungkin dapat menggunakan sys.path_hooks (Write-up courtesy of PyMOTW).

Itulah semua yang ada di kepala saya.

Ini adalah solusi parsial, masih lebih baik dari semua yang dipublikasikan sejauh ini.

import sys, os, os.path, inspect

#os.chdir("..")

if '__file__' not in locals():
    __file__ = inspect.getframeinfo(inspect.currentframe())[0]

print os.path.dirname(os.path.abspath(__file__))

Sekarang ini berfungsi semua panggilan tetapi jika seseorang menggunakan chdir()untuk mengubah direktori saat ini, ini juga akan gagal.

Catatan:

• sys.argv[0]tidak akan berfungsi, akan kembali -cjika Anda menjalankan skrip denganpython -c "execfile('path-tester.py')"
• Saya menerbitkan tes lengkap di https://gist.github.com/1385555 dan Anda dipersilakan untuk memperbaikinya.

Ini harus bekerja dalam banyak kasus:

import os,sys
dirname=os.path.dirname(os.path.realpath(sys.argv[0]))

Semoga ini membantu: - Jika Anda menjalankan skrip / modul dari mana saja Anda akan dapat mengakses __file__variabel yang merupakan variabel modul yang mewakili lokasi skrip.

Di sisi lain, jika Anda menggunakan juru bahasa Anda tidak memiliki akses ke variabel itu, di mana Anda akan mendapatkan nama NameErrordan os.getcwd()akan memberi Anda direktori yang salah jika Anda menjalankan file dari tempat lain.

Solusi ini akan memberi Anda apa yang Anda cari dalam semua kasus:

from inspect import getsourcefile
from os.path import abspath
abspath(getsourcefile(lambda:0))

Saya belum benar-benar mengujinya tetapi itu menyelesaikan masalah saya.