| __author__ = 'ktisha' | |
| import os | |
| import sys | |
| PYTHON_VERSION_MAJOR = sys.version_info[0] | |
| PYTHON_VERSION_MINOR = sys.version_info[1] | |
| ENABLE_DEBUG_LOGGING = False | |
| if os.getenv("UTRUNNER_ENABLE_DEBUG_LOGGING"): | |
| ENABLE_DEBUG_LOGGING = True | |
| def debug(what): | |
| if ENABLE_DEBUG_LOGGING: | |
| sys.stdout.writelines(str(what) + '\n') | |
| def adjust_sys_path(add_script_parent=True, script_index=1): | |
| sys.path.pop(0) | |
| if add_script_parent: | |
| script_path = os.path.dirname(sys.argv[script_index]) | |
| insert_to_sys_path(script_path) | |
| def adjust_django_sys_path(): | |
| pycharm_path = sys.path.pop(0) | |
| script_path = sys.argv[-1] | |
| insert_to_sys_path(script_path) | |
| sys.path.append(pycharm_path) | |
| def import_system_module(name): | |
| if sys.platform == "cli": # hack for the ironpython | |
| return __import__(name) | |
| if sys.version_info >= (2, 7): | |
| import importlib | |
| module = importlib.import_module(name) | |
| module.__name__ = "pycharm_" + name | |
| return module | |
| else: | |
| import imp | |
| f, filename, desc = imp.find_module(name) | |
| return imp.load_module('pycharm_' + name, f, filename, desc) | |
| def getModuleName(prefix, cnt): | |
| return prefix + "%" + str(cnt) | |
| def insert_to_sys_path(script_path): | |
| while script_path in sys.path: | |
| sys.path.remove(script_path) | |
| sys.path.insert(0, script_path) |