Ich wollte die Verarbeitung zum Zeitpunkt des Tests unbedingt trennen. Lassen Sie uns anhand des Dateinamens des Laufzeitarguments mit sys.argv beurteilen.
import sys
if 'test_loader.py' in sys.argv:
db = load_database('test')
else:
db = load_database('production')
Laufzeit
python test_loader.py
Wenn Sie jedoch test_loader von einem anderen Skript aus aufrufen müssen, ist dies nicht der richtige Weg. ..
test_loader.py:
import os
if not os.getenv('DB'):
os.environ['DB'] = 'test'
Recommended Posts