thijso/cdist
1
0
Fork 0
forked from ungleich-public/cdist
Code Issues Pull requests Projects Releases Wiki Activity

add parent dir to module search path only when importing fails

Browse source
This commit is contained in:
ander 2020-10-16 13:42:16 +03:00
parent 86057cef19
commit fd04c03613
1 changed files with 12 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

18
bin/cdist
View file

@ -25,13 +25,19 @@ import logging
import os
import sys
cdist_bin = os.path.abspath(__file__)
if os.path.islink(cdist_bin):
cdist_bin = os.readlink(cdist_bin)
cdist_dir = os.path.abspath(os.path.join(os.path.dirname(cdist_bin), os.pardir))
sys.path.insert(0, cdist_dir)
# try to import cdist and if that fails, then add this file's parent dir to
# module search path and try again. additionally check if this file is
# symlinked, so user can symlink this file to, for example, ~/.local/bin.
try:
import cdist
except ModuleNotFoundError:
cdist_bin = os.path.abspath(__file__)
if os.path.islink(cdist_bin):
cdist_bin = os.readlink(cdist_bin)
cdist_dir = os.path.abspath(os.path.join(os.path.dirname(cdist_bin), os.pardir))
sys.path.insert(0, cdist_dir)
import cdist
import cdist
import cdist.argparse
import cdist.banner
import cdist.config