Wie kann man das aktuelle Skriptverzeichnis richtig bestimmen?

Ich würde gerne sehen, wie man das aktuelle Skriptverzeichnis in Python am besten ermittelt.

Ich habe festgestellt, dass es aufgrund der vielen Möglichkeiten, Python-Code aufzurufen, schwierig ist, eine gute Lösung zu finden.

Hier sind einige Probleme:

• __file__ist nicht definiert, wenn das Skript ausgeführt wird mit exec,execfile
• __module__ wird nur in Modulen definiert

Anwendungsfälle:

• ./myfile.py
• python myfile.py
• ./somedir/myfile.py
• python somedir/myfile.py
• execfile('myfile.py') (von einem anderen Skript, das sich in einem anderen Verzeichnis befinden kann und das ein anderes aktuelles Verzeichnis haben kann.

Ich weiß, dass es keine perfekte Lösung gibt, aber ich suche nach dem besten Ansatz, der die meisten Fälle löst.

Der am häufigsten verwendete Ansatz ist, os.path.dirname(os.path.abspath(__file__))aber dies funktioniert wirklich nicht, wenn Sie das Skript von einem anderen mit ausführen exec().

Warnung

Jede Lösung, die das aktuelle Verzeichnis verwendet, schlägt fehl. Dies kann je nach Aufruf des Skripts unterschiedlich sein oder innerhalb des laufenden Skripts geändert werden.

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

ist in der Tat das Beste, was Sie bekommen werden.

Es ist ungewöhnlich, ein Skript mit exec/ auszuführen execfile. Normalerweise sollten Sie die Modulinfrastruktur zum Laden von Skripten verwenden. Wenn Sie diese Methoden verwenden müssen, empfehle ich die Einstellung __file__in der globalsÜbergabe an das Skript, damit es diesen Dateinamen lesen kann.

Es gibt keine andere Möglichkeit, den Dateinamen im ausgeführten Code abzurufen: Wie Sie bemerken, befindet sich das CWD möglicherweise an einer völlig anderen Stelle.

Wenn Sie wirklich den Fall abdecken möchten, über den ein Skript aufgerufen wird execfile(...), können Sie mithilfe des inspectModuls den Dateinamen (einschließlich des Pfads) ableiten. Soweit mir bekannt ist, funktioniert dies in allen von Ihnen aufgeführten Fällen:

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())

Es funktioniert auf CPython, Jython, Pypy. Es funktioniert, wenn das Skript mit ausgeführt wird execfile()( sys.argv[0]und __file__-basierte Lösungen hier fehlschlagen würden). Es funktioniert, wenn sich das Skript in einer ausführbaren Zip-Datei (/ einem Ei) befindet . Es funktioniert, wenn das Skript PYTHONPATH=/path/to/library.zip python -mscript_to_runaus einer Zip-Datei "importiert" ( ) wird. In diesem Fall wird der Archivpfad zurückgegeben. Es funktioniert, wenn das Skript in eine eigenständige ausführbare Datei ( sys.frozen) kompiliert wird . Es funktioniert für Symlinks ( realpatheliminiert symbolische Links). Es funktioniert in einem interaktiven Interpreter. In diesem Fall wird das aktuelle Arbeitsverzeichnis zurückgegeben.

In Python 3.4+ können Sie das einfachere pathlibModul verwenden:

from inspect import currentframe, getframeinfo
from pathlib import Path

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

Der os.path...Ansatz war das "erledigte Ding" in Python 2.

In Python 3 finden Sie das Skriptverzeichnis wie folgt:

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

Verwenden Sie einfach os.path.dirname(os.path.abspath(__file__))und prüfen Sie sehr sorgfältig, ob für den Fall, in dem execes verwendet wird, ein wirklicher Bedarf besteht. Es könnte ein Zeichen für ein problematisches Design sein, wenn Sie Ihr Skript nicht als Modul verwenden können.

Denken Sie an Zen of Python # 8 , und wenn Sie der Meinung sind, dass es ein gutes Argument für einen Anwendungsfall gibt, für den es funktionieren muss exec, teilen Sie uns bitte einige Details zum Hintergrund des Problems mit.

Würde

import os
cwd = os.getcwd()

Tun Sie, was Sie wollen? Ich bin mir nicht sicher, was genau Sie mit dem "aktuellen Skriptverzeichnis" meinen. Was wäre die erwartete Ausgabe für die von Ihnen angegebenen Anwendungsfälle?

Erstens ... ein paar fehlende Anwendungsfälle hier, wenn wir über Möglichkeiten sprechen, anonymen Code einzufügen.

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

Die eigentliche Frage ist jedoch, was ist Ihr Ziel - versuchen Sie, irgendeine Art von Sicherheit durchzusetzen? Oder sind Sie nur daran interessiert, was geladen wird?

Wenn Sie an Sicherheit interessiert sind , spielt der Dateiname, der über exec / execfile importiert wird, keine Rolle. Sie sollten rexec verwenden , das Folgendes bietet:

Dieses Modul enthält die RExec-Klasse, die die Methoden r_eval (), r_execfile (), r_exec () und r_import () unterstützt. Hierbei handelt es sich um eingeschränkte Versionen der Standard-Python-Funktionen eval (), execfile () sowie der Anweisungen exec und import. Code, der in dieser eingeschränkten Umgebung ausgeführt wird, hat nur Zugriff auf Module und Funktionen, die als sicher gelten. Sie können RExec-Funktionen nach Bedarf hinzufügen oder entfernen.

Wenn dies jedoch eher eine akademische Angelegenheit ist, finden Sie hier einige doofe Ansätze, in die Sie möglicherweise etwas tiefer eintauchen können.

Beispielskripte:

./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)

Ausgabe

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

Dies ist natürlich eine ressourcenintensive Methode. Sie würden Ihren gesamten Code nachverfolgen. Nicht sehr effizient. Aber ich denke, es ist ein neuartiger Ansatz, da er auch dann weiter funktioniert, wenn Sie tiefer in das Nest eindringen. Sie können 'eval' nicht überschreiben. Obwohl Sie können execfile außer Kraft setzen ().

Beachten Sie, dass dieser Ansatz nur exec / execfile abdeckt, nicht 'import'. Für das Laden von Modulen auf höherer Ebene können Sie möglicherweise sys.path_hooks verwenden (mit freundlicher Genehmigung von PyMOTW).

Das ist alles, was ich auf dem Kopf habe.

Hier ist eine Teillösung, die immer noch besser ist als alle bisher veröffentlichten.

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__))

Jetzt funktioniert dies bei allen Aufrufen, aber wenn jemand chdir()das aktuelle Verzeichnis ändert, schlägt dies ebenfalls fehl.

Anmerkungen:

• sys.argv[0]wird nicht funktionieren, wird zurückkehren, -cwenn Sie das Skript mit ausführenpython -c "execfile('path-tester.py')"
• Ich habe einen vollständigen Test unter https://gist.github.com/1385555 veröffentlicht, und Sie können ihn gerne verbessern.

Dies sollte in den meisten Fällen funktionieren:

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

Hoffentlich hilft dies: - Wenn Sie ein Skript / Modul von einem beliebigen Ort aus ausführen, können Sie auf die __file__Variable zugreifen , bei der es sich um eine Modulvariable handelt, die den Speicherort des Skripts darstellt.

Wenn Sie dagegen den Interpreter verwenden, haben Sie keinen Zugriff auf diese Variable. Dort erhalten Sie einen Namen NameErrorund erhalten os.getcwd()das falsche Verzeichnis, wenn Sie die Datei von einem anderen Ort aus ausführen.

Diese Lösung sollte Ihnen in allen Fällen das bieten, wonach Sie suchen:

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

Ich habe es nicht gründlich getestet, aber es hat mein Problem gelöst.