mirror of
https://github.com/ARM-software/workload-automation.git
synced 2025-09-03 03:42:35 +01:00
utils: pep8 fixes
This commit is contained in:
@@ -100,6 +100,7 @@ POD_TYPES = [
|
||||
cpu_mask,
|
||||
]
|
||||
|
||||
|
||||
class WAJSONEncoder(_json.JSONEncoder):
|
||||
|
||||
def default(self, obj): # pylint: disable=method-hidden
|
||||
@@ -190,13 +191,16 @@ def _wa_regex_representer(dumper, data):
|
||||
text = '{}:{}'.format(data.flags, data.pattern)
|
||||
return dumper.represent_scalar(_regex_tag, text)
|
||||
|
||||
|
||||
def _wa_level_representer(dumper, data):
|
||||
text = '{}:{}'.format(data.name, data.level)
|
||||
return dumper.represent_scalar(_level_tag, text)
|
||||
|
||||
|
||||
def _wa_cpu_mask_representer(dumper, data):
|
||||
return dumper.represent_scalar(_cpu_mask_tag, data.mask())
|
||||
|
||||
|
||||
def _wa_dict_constructor(loader, node):
|
||||
pairs = loader.construct_pairs(node)
|
||||
seen_keys = set()
|
||||
@@ -212,11 +216,13 @@ def _wa_regex_constructor(loader, node):
|
||||
flags, pattern = value.split(':', 1)
|
||||
return re.compile(pattern, int(flags or 0))
|
||||
|
||||
|
||||
def _wa_level_constructor(loader, node):
|
||||
value = loader.construct_scalar(node)
|
||||
name, value = value.split(':', 1)
|
||||
return level(name, value)
|
||||
|
||||
|
||||
def _wa_cpu_mask_constructor(loader, node):
|
||||
value = loader.construct_scalar(node)
|
||||
return cpu_mask(value)
|
||||
@@ -317,9 +323,9 @@ def _read_pod(fh, fmt=None):
|
||||
if fmt == '':
|
||||
# Special case of no given file extension
|
||||
message = ("Could not determine format " +
|
||||
"from file extension for \"{}\". ".format(getattr(fh, 'name', '<none>')) +
|
||||
"Please specify it or modify the fmt parameter.")
|
||||
raise ValueError(message)
|
||||
"from file extension for \"{}\". "
|
||||
"Please specify it or modify the fmt parameter.")
|
||||
raise ValueError(message.format(getattr(fh, 'name', '<none>')))
|
||||
if fmt == 'yaml':
|
||||
return yaml.load(fh)
|
||||
elif fmt == 'json':
|
||||
@@ -344,7 +350,7 @@ def _write_pod(pod, wfh, fmt=None):
|
||||
|
||||
|
||||
def is_pod(obj):
|
||||
if type(obj) not in POD_TYPES:
|
||||
if type(obj) not in POD_TYPES:
|
||||
return False
|
||||
if hasattr(obj, 'items'):
|
||||
for k, v in obj.items():
|
||||
|
Reference in New Issue
Block a user