def test_init(self):
loglevel = logging.ERROR
logfmt = log.LOG_FORMAT
mock_cfg_func = mock.Mock()
logging.basicConfig = mock_cfg_func
with tempfile.TemporaryDirectory() as tmp_dir:
logfile = Path(tmp_dir).joinpath("my_output.log")
with mock.patch(
"hed_utils.support.log.add_file_handler") as mock_add_hdlr:
log.init(loglevel, logfmt, True, logfile)
log.init(loglevel, logfmt, False)
mock_add_hdlr.assert_called_once_with(logging.getLogger(),
logfile, loglevel,
logfmt)
expected_cfg_calls = [
mock.call(level=loglevel, format=logfmt,
stream=sys.stdout),
mock.call(level=loglevel, format=logfmt)
]
actual_cfg_calls = mock_cfg_func.mock_calls
self.assertListEqual(expected_cfg_calls, actual_cfg_calls)
def main():
from hed_utils.selenium import chrome_driver, SharedDriver
from hed_utils.support import log
log.init(logging.DEBUG)
driver = chrome_driver.create_instance()
SharedDriver.set_instance(driver)
urls = [
"https://youtube.com", "https://facebook.com",
"https://stackoverflow.com"
]
for url in urls:
get_load_stats(SharedDriver(), url)
def main(args):
"""Main entry point allowing external calls
Args:
args ([str]): command line parameter list
"""
args = _parse_args(args)
print(
f"kill_all(name='{args.name}', pattern='{args.pattern}', pid='{args.pid}', ignorecase='{args.ignorecase}', "
f"dry='{args.dry}')")
if args.loglevel:
log.init(level=args.loglevel)
victims = [
victim._asdict() for victim in os_util.kill_all(
name=args.name, pattern=args.pattern, pid=args.pid, dry=args.dry)
]
print(f"kill_all victims:\n" + tabulate(victims, headers="keys"))
def init_logging(level):
logfile = get_log_filepath()
log.init(level=level or logging.INFO, file=logfile, log_format=LOG_FORMAT)
_log.info("initialized log-file at: %s", logfile)
def _init_logging(level):
from hed_utils.support import log
log.init(level=level, log_format=LOG_FORMAT)