def init_logger(args):
"""
init the logger
:param args:
:return:
"""
log_title = '{project}'.format(project=args.project)
log_dir = os.path.join(CUR_DIR, 'log', args.project)
if args.suite and args.project != args.suite:
log_title += '-{suite}'.format(suite=args.suite)
log_dir = os.path.join(log_dir, args.suite)
if args.case_list:
log_title += '-{str_case}'.format(str_case='_'.join(args.case_list))
if args.output:
log_dir = os.path.join(CUR_DIR, args.output)
log_name = log_title + '-' + STR_TIME
log_path = os.path.join(log_dir, log_name + '.log')
logger = log.get_logger(log_path, output_logfile=True, debug=args.debug, reset_logger=True)
return logger
@Email : [email protected] """ import os import random import json import xlrd from prettytable import PrettyTable from concurrent.futures import ThreadPoolExecutor, as_completed from storagetest.pkgs.fileops import FileOps from storagetest.pkgs.dd import DD from storagetest.libs import utils, log from storagetest.libs.exceptions import PlatformError, NoSuchDir logger = log.get_logger() class RawUT(object): """RAW unit test""" def __init__(self, raw_device): self.raw_device = raw_device self.phase_list = [] def verify(self): if os.name != "posix": raise PlatformError("Just support for linux machine!") if not os.path.isdir(self.raw_device): raise NoSuchDir(self.raw_device) def print_phase(self):
try: # Python 2.7+
from logging import NullHandler
except ImportError:
class NullHandler(logging.Handler):
def emit(self, record):
pass
from storagetest.libs import log
# ======================
# --- Global
# ======================
# logging_logger = logging.getLogger(__name__)
logging_logger = log.get_logger()
def _sleep_progressbar(sleep_time):
"""
Print a progress bar, total value: sleep_time(seconds)
:param sleep_time:
:return:
"""
widgets = [
'Progress: ',
Percentage(), ' ',
Bar(marker=RotatingMarker('-=>')), ' ',
ETA()
]
