d3e0301aee
we should only load one module once. if we load one module twice, python will regard the same object loaded in the first time and second time as different objects. it will lead to strange errors like `isinstance(object, type_of_this_object)` return False
60 lines
1.7 KiB
Python
60 lines
1.7 KiB
Python
from __future__ import print_function
|
|
import sys
|
|
|
|
|
|
_COLOR_CODES = {
|
|
"white": u'\033[0m',
|
|
"red": u'\033[31m',
|
|
"green": u'\033[32m',
|
|
"orange": u'\033[33m',
|
|
"blue": u'\033[34m',
|
|
"purple": u'\033[35m',
|
|
"W": u'\033[0m',
|
|
"R": u'\033[31m',
|
|
"G": u'\033[32m',
|
|
"O": u'\033[33m',
|
|
"B": u'\033[34m',
|
|
"P": u'\033[35m'
|
|
}
|
|
|
|
|
|
def console_log(data, color="white", end="\n"):
|
|
"""
|
|
log data to console.
|
|
(if not flush console log, Gitlab-CI won't update logs during job execution)
|
|
|
|
:param data: data content
|
|
:param color: color
|
|
"""
|
|
if color not in _COLOR_CODES:
|
|
color = "white"
|
|
color_codes = _COLOR_CODES[color]
|
|
if isinstance(data, type(b'')):
|
|
data = data.decode('utf-8', 'replace')
|
|
print(color_codes + data, end=end)
|
|
if color not in ["white", "W"]:
|
|
# reset color to white for later logs
|
|
print(_COLOR_CODES["white"] + u"\r")
|
|
sys.stdout.flush()
|
|
|
|
|
|
__LOADED_MODULES = dict()
|
|
# we should only load one module once.
|
|
# if we load one module twice,
|
|
# python will regard the same object loaded in the first time and second time as different objects.
|
|
# it will lead to strange errors like `isinstance(object, type_of_this_object)` return False
|
|
|
|
|
|
def load_source(name, path):
|
|
try:
|
|
return __LOADED_MODULES[name]
|
|
except KeyError:
|
|
try:
|
|
from importlib.machinery import SourceFileLoader
|
|
ret = SourceFileLoader(name, path).load_module()
|
|
except ImportError:
|
|
# importlib.machinery doesn't exists in Python 2 so we will use imp (deprecated in Python 3)
|
|
import imp
|
|
ret = imp.load_source(name, path)
|
|
__LOADED_MODULES[name] = ret
|
|
return ret
|