Fix importing cdist module

Resolve #845.
This commit is contained in:
Darko Poljak 2020-11-14 10:13:58 +01:00
parent 5092752786
commit 76aa00b12e
1 changed files with 15 additions and 17 deletions

View File

@ -25,25 +25,23 @@ import logging
import os import os
import sys import sys
# try to import cdist and if that fails, # See if this file's parent is cdist module
# then add this file's parent dir to # and if so add it to module search path.
# module search path and try again. cdist_dir = os.path.realpath(
try: os.path.join(
import cdist os.path.dirname(os.path.realpath(__file__)),
except ModuleNotFoundError: os.pardir))
cdist_dir = os.path.realpath( cdist_init_dir = os.path.join(cdist_dir, 'cdist', '__init__.py')
os.path.join( if os.path.exists(cdist_init_dir):
os.path.dirname(os.path.realpath(__file__)),
os.pardir))
sys.path.insert(0, cdist_dir) sys.path.insert(0, cdist_dir)
import cdist
import cdist.argparse import cdist # noqa 402
import cdist.banner import cdist.argparse # noqa 402
import cdist.config import cdist.banner # noqa 402
import cdist.install import cdist.config # noqa 402
import cdist.shell import cdist.install # noqa 402
import cdist.inventory import cdist.shell # noqa 402
import cdist.inventory # noqa 402
def commandline(): def commandline():