Skip to content

Commit a8053f8

Browse files
committed
update logging level logic
1 parent c53d138 commit a8053f8

File tree

2 files changed

+36
-51
lines changed

2 files changed

+36
-51
lines changed

automation/script/module.py

+34-51
Original file line numberDiff line numberDiff line change
@@ -202,7 +202,7 @@ def run(self, i):
202202
to let scripts deal with that
203203
204204
(silent) (bool): if True, attempt to suppress all info if supported
205-
(sets MLC_SILENT=yes)
205+
(sets MLC_TMP_SILENT=yes)
206206
(s) (bool): the same as 'silent'
207207
...
208208
@@ -373,42 +373,40 @@ def _run(self, i):
373373

374374
# Check verbose and silent
375375

376-
# get the current log level so that the log levels could be reverted
377-
# back when calling the dependency scripts
378-
current_logging_level = logging.getLevelName(logger.level)
379-
tmp_logging_level = logging.getLevelName(logger.level)
376+
# Get the current log level so that the log levels could be reverted
377+
# after execution of the script and corresponding dependencies
378+
original_logging_level = logger.level
380379

381380
verbose = False
382-
383-
silent = True if is_true(i.get('silent', '')) else False
384-
385-
if not silent:
381+
silent = False
382+
383+
if i.get('verbose', '') != '':
384+
verbose = True if is_true(i.get('verbose', '')) else False
385+
elif i.get('v', '') != '':
386+
verbose = True if is_true(i.get('v', '')) else False
387+
elif env.get('MLC_VERBOSE', '') != '':
388+
verbose = True if is_true(env.get('MLC_VERBOSE', '')) else False
389+
390+
if i.get('silent', '') != '':
391+
silent = True if is_true(i.get('silent', '')) else False
392+
elif i.get('s', '') != '':
386393
silent = True if is_true(i.get('s', '')) else False
394+
elif env.get('MLC_SILENT', '') != '':
395+
silent = True if is_true(env.get('MLC_SILENT', '')) else False
396+
397+
if verbose and silent:
398+
logger.warning("Both verbose and silent is set to True. Verbose will take precedence.")
399+
silent = False
387400

388401
if silent:
389-
if 'verbose' in i:
390-
del (i['verbose'])
391-
if 'v' in i:
392-
del (i['v'])
393402
env['MLC_TMP_SILENT'] = 'yes'
394403
logger.setLevel(logging.ERROR)
395-
tmp_logging_level = logging.ERROR
396404
run_state['tmp_silent'] = True
397405

398-
if 'verbose' in i:
399-
verbose = i['verbose']
400-
elif 'v' in i:
401-
verbose = i['v']
402-
403406
if verbose:
404407
env['MLC_TMP_VERBOSE'] = 'yes'
405408
run_state['tmp_verbose'] = True
406409
logger.setLevel(logging.DEBUG)
407-
tmp_logging_level = logging.DEBUG
408-
409-
# populate the logging levels in run state
410-
run_state['tmp_logging_level'] = tmp_logging_level
411-
run_state['current_logging_level'] = current_logging_level
412410

413411
print_deps = i.get('print_deps', False)
414412
print_versions = i.get('print_versions', False)
@@ -1242,13 +1240,11 @@ def _run(self, i):
12421240
logger.debug(
12431241
recursion_spaces +
12441242
' - Checking dynamic dependencies on other MLC scripts:')
1245-
logger.setLevel(current_logging_level)
12461243
r = self._call_run_deps(deps, self.local_env_keys, local_env_keys_from_meta, env, state, const, const_state, add_deps_recursive,
12471244
recursion_spaces + extra_recursion_spaces,
1248-
remembered_selections, variation_tags_string, True, debug_script_tags, False, show_time, extra_recursion_spaces, run_state)
1245+
remembered_selections, variation_tags_string, True, debug_script_tags, verbose, show_time, extra_recursion_spaces, run_state)
12491246
if r['return'] > 0:
12501247
return r
1251-
logger.setLevel(tmp_logging_level)
12521248

12531249
logger.debug(
12541250
recursion_spaces +
@@ -1264,13 +1260,11 @@ def _run(self, i):
12641260
recursion_spaces +
12651261
' - Checking prehook dependencies on other MLC scripts:')
12661262

1267-
logger.setLevel(current_logging_level)
12681263
r = self._call_run_deps(prehook_deps, self.local_env_keys, local_env_keys_from_meta, env, state, const, const_state, add_deps_recursive,
12691264
recursion_spaces + extra_recursion_spaces,
1270-
remembered_selections, variation_tags_string, True, debug_script_tags, False, show_time, extra_recursion_spaces, run_state)
1265+
remembered_selections, variation_tags_string, True, debug_script_tags, verbose, show_time, extra_recursion_spaces, run_state)
12711266
if r['return'] > 0:
12721267
return r
1273-
logger.setLevel(tmp_logging_level)
12741268

12751269
# Continue with the selected cached script
12761270
cached_script = found_cached_scripts[selection]
@@ -1329,26 +1323,22 @@ def _run(self, i):
13291323
clean_env_keys_post_deps = meta.get(
13301324
'clean_env_keys_post_deps', [])
13311325

1332-
logger.setLevel(current_logging_level)
13331326
r = self._call_run_deps(posthook_deps, self.local_env_keys, clean_env_keys_post_deps, env, state, const, const_state, add_deps_recursive,
13341327
recursion_spaces + extra_recursion_spaces,
1335-
remembered_selections, variation_tags_string, True, debug_script_tags, False, show_time, extra_recursion_spaces, run_state)
1328+
remembered_selections, variation_tags_string, True, debug_script_tags, verbose, show_time, extra_recursion_spaces, run_state)
13361329
if r['return'] > 0:
13371330
return r
1338-
logger.setLevel(tmp_logging_level)
13391331

13401332
logger.debug(
13411333
recursion_spaces +
13421334
' - Checking post dependencies on other MLC scripts:')
13431335

13441336
# Check chain of post dependencies on other MLC scripts
1345-
logger.setLevel(current_logging_level)
13461337
r = self._call_run_deps(post_deps, self.local_env_keys, clean_env_keys_post_deps, env, state, const, const_state, add_deps_recursive,
13471338
recursion_spaces + extra_recursion_spaces,
1348-
remembered_selections, variation_tags_string, True, debug_script_tags, False, show_time, extra_recursion_spaces, run_state)
1339+
remembered_selections, variation_tags_string, True, debug_script_tags, verbose, show_time, extra_recursion_spaces, run_state)
13491340
if r['return'] > 0:
13501341
return r
1351-
logger.setLevel(tmp_logging_level)
13521342

13531343
if renew or (not found_cached and num_found_cached_scripts == 0):
13541344
# Add more tags to cached tags
@@ -1551,13 +1541,11 @@ def _run(self, i):
15511541
recursion_spaces +
15521542
' - Checking docker run dependencies on other MLC scripts:')
15531543

1554-
logger.setLevel(current_logging_level)
15551544
r = self._call_run_deps(docker_deps, self.local_env_keys, local_env_keys_from_meta, env, state, const, const_state, add_deps_recursive,
15561545
recursion_spaces + extra_recursion_spaces,
1557-
remembered_selections, variation_tags_string, False, debug_script_tags, False, show_time, extra_recursion_spaces, run_state)
1546+
remembered_selections, variation_tags_string, False, debug_script_tags, verbose, show_time, extra_recursion_spaces, run_state)
15581547
if r['return'] > 0:
15591548
return r
1560-
logger.setLevel(tmp_logging_level)
15611549

15621550
logger.debug(
15631551
recursion_spaces +
@@ -1592,7 +1580,7 @@ def _run(self, i):
15921580
'variation_tags_string': variation_tags_string,
15931581
'found_cached': False,
15941582
'debug_script_tags': debug_script_tags,
1595-
'verbose': False,
1583+
'verbose': verbose,
15961584
'meta': meta,
15971585
'self': self
15981586
}
@@ -1651,13 +1639,11 @@ def _run(self, i):
16511639
logger.debug(recursion_spaces +
16521640
' - Checking dependencies on other MLC scripts:')
16531641

1654-
logger.setLevel(current_logging_level)
16551642
r = self._call_run_deps(deps, self.local_env_keys, local_env_keys_from_meta, env, state, const, const_state, add_deps_recursive,
16561643
recursion_spaces + extra_recursion_spaces,
1657-
remembered_selections, variation_tags_string, False, debug_script_tags, False, show_time, extra_recursion_spaces, run_state)
1644+
remembered_selections, variation_tags_string, False, debug_script_tags, verbose, show_time, extra_recursion_spaces, run_state)
16581645
if r['return'] > 0:
16591646
return r
1660-
logger.setLevel(tmp_logging_level)
16611647

16621648
logger.debug(recursion_spaces +
16631649
' - Processing env after dependencies ...')
@@ -1869,13 +1855,11 @@ def _run(self, i):
18691855
recursion_spaces +
18701856
' - Checking prehook dependencies on other MLC scripts:')
18711857

1872-
logger.setLevel(current_logging_level)
18731858
r = self._call_run_deps(prehook_deps, self.local_env_keys, local_env_keys_from_meta, env, state, const, const_state, add_deps_recursive,
18741859
recursion_spaces + extra_recursion_spaces,
1875-
remembered_selections, variation_tags_string, found_cached, debug_script_tags, False, show_time, extra_recursion_spaces, run_state)
1860+
remembered_selections, variation_tags_string, found_cached, debug_script_tags, verbose, show_time, extra_recursion_spaces, run_state)
18761861
if r['return'] > 0:
18771862
return r
1878-
logger.setLevel(tmp_logging_level)
18791863

18801864
if not fake_run:
18811865
env_key_mappings = meta.get("env_key_mappings", {})
@@ -1915,12 +1899,10 @@ def _run(self, i):
19151899
clean_env_keys_post_deps = meta.get(
19161900
'clean_env_keys_post_deps', [])
19171901

1918-
logger.setLevel(current_logging_level)
19191902
r = self._run_deps(post_deps, clean_env_keys_post_deps, env, state, const, const_state, add_deps_recursive, recursion_spaces,
1920-
remembered_selections, variation_tags_string, found_cached, debug_script_tags, False, show_time, extra_recursion_spaces, run_state)
1903+
remembered_selections, variation_tags_string, found_cached, debug_script_tags, verbose, show_time, extra_recursion_spaces, run_state)
19211904
if r['return'] > 0:
19221905
return r
1923-
logger.setLevel(tmp_logging_level)
19241906

19251907
# Add extra tags from env updated by deps (such as python version
19261908
# and compiler version, etc)
@@ -2268,6 +2250,9 @@ def _run(self, i):
22682250
if i.get('pause', False):
22692251
input('Press Enter to continue ...')
22702252

2253+
# set the logger level back to the original
2254+
logger.setLevel(original_logging_level)
2255+
22712256
return rr
22722257

22732258
##########################################################################
@@ -5286,13 +5271,11 @@ def prepare_and_run_script_with_postprocessing(i, postprocess="postprocess"):
52865271
postprocess,
52875272
customize_code.__file__))
52885273

5289-
logger.setLevel(run_state['current_logging_level'])
52905274
if len(posthook_deps) > 0 and (postprocess == "postprocess"):
52915275
r = script_automation._call_run_deps(posthook_deps, local_env_keys, local_env_keys_from_meta, env, state, const, const_state,
52925276
add_deps_recursive, recursion_spaces, remembered_selections, variation_tags_string, found_cached, debug_script_tags, verbose, show_time, ' ', run_state)
52935277
if r['return'] > 0:
52945278
return r
5295-
logger.setLevel(run_state['tmp_logging_level'])
52965279

52975280
if (postprocess == "postprocess") and customize_code is not None and 'postprocess' in dir(
52985281
customize_code):

script/detect-cpu/meta.yaml

+2
Original file line numberDiff line numberDiff line change
@@ -8,6 +8,8 @@ clean_files:
88
- tmp-wmic-cpu.csv
99
deps:
1010
- tags: detect,os
11+
names:
12+
- detect-os
1113
new_env_keys:
1214
- MLC_HOST_CPU_*
1315
- MLC_HOST_MEMORY_CAPACITY

0 commit comments

Comments
 (0)