import inspect import sys from integration_scripts import common_config from integration_scripts.constants.filenames import DEV_MAIN from integration_scripts.db_utils.db_utils import * # noqa frame = inspect.stack() if 'pytest' in sys.modules: common_config.ENVIRONMENT = common_config.TEST_ENVIRONMENT is_dev = True in [DEV_MAIN in f[1] for f in frame] is_test = common_config.ENVIRONMENT == common_config.TEST_ENVIRONMENT if common_config.AR_MYSQL_HOST or is_test or is_dev: from integration_scripts.db_utils.mysql_utils import * # noqa if common_config.SNOWFLAKE_USER or is_test or is_dev: from integration_scripts.db_utils.snowflake_utils import * # noqa