1
0
mirror of https://github.com/kellyjonbrazil/jc.git synced 2025-08-08 22:36:48 +02:00

fix zpool-status for multi-line fields

This commit is contained in:
Kelly Brazil
2023-02-04 17:37:48 -08:00
parent adf5f403ae
commit 96cb01f57a

View File

@@ -36,6 +36,7 @@ Examples:
from typing import List, Dict from typing import List, Dict
from jc.jc_types import JSONDictType from jc.jc_types import JSONDictType
import jc.utils import jc.utils
from jc.parsers.kv import parse as kv_parse
class info(): class info():
@@ -67,6 +68,25 @@ def _process(proc_data: List[JSONDictType]) -> List[JSONDictType]:
return proc_data return proc_data
def _build_config_list(string: str) -> List[Dict]:
config_list: List = []
for line in filter(None, string.splitlines()):
if line.strip().endswith('READ WRITE CKSUM'):
continue
line_list = line.strip().split(maxsplit=5)
config_obj: Dict = {}
config_obj['name'] = line_list[0]
config_obj['state'] = line_list[1]
config_obj['read'] = line_list[2]
config_obj['write'] = line_list[3]
config_obj['checksum'] = line_list[4]
if len(line_list) == 6:
config_obj['errors'] = line_list[5]
config_list.append(config_obj)
return config_list
def parse( def parse(
data: str, data: str,
raw: bool = False, raw: bool = False,
@@ -89,66 +109,33 @@ def parse(
jc.utils.input_type_check(data) jc.utils.input_type_check(data)
raw_output: List[Dict] = [] raw_output: List[Dict] = []
pool_str: str = ''
pool_obj: Dict = {} pool_obj: Dict = {}
in_config: bool = False
parent: str = ''
config: List[Dict] = []
config_obj: Dict = {}
if jc.utils.has_data(data): if jc.utils.has_data(data):
for line in filter(None, data.splitlines()): for line in filter(None, data.splitlines()):
line_list = line.strip().split(maxsplit=1)
if line.startswith(' pool: '): if line.lstrip().startswith('pool: '):
if pool_obj: if pool_str:
if config: pool_obj = kv_parse(pool_str)
pool_obj['config'] = config if 'config' in pool_obj:
pool_obj['config'] = _build_config_list(pool_obj['config'])
raw_output.append(pool_obj) raw_output.append(pool_obj)
pool_str = ''
config_obj = {} pool_str += line + '\n'
config = []
parent = ''
in_config = False
pool_obj = {
"pool": line_list[1]
}
continue continue
if line.startswith(' state: ') \ if line.startswith(' '):
or line.startswith(' scan: ') \ pool_str += line + '\n'
or line.startswith('errors: '):
pool_obj[line_list[0][:-1]] = line_list[1]
in_config = False
continue continue
if line.startswith('config:'): pool_str += line.strip() + '\n'
in_config = True
continue
if in_config and line.strip().endswith('READ WRITE CKSUM'): if pool_str:
continue pool_obj = kv_parse(pool_str)
if 'config' in pool_obj:
if in_config: pool_obj['config'] = _build_config_list(pool_obj['config'])
config_line = line.rstrip().split()
config_obj = {}
if line.startswith(' '):
config_obj['parent'] = parent
config_obj['name'] = config_line[0]
else:
parent = config_line[0]
config_obj['name'] = parent
config_obj['state'] = config_line[1]
config_obj['read'] = config_line[2]
config_obj['write'] = config_line[3]
config_obj['checksum'] = config_line[4]
config.append(config_obj)
if pool_obj:
if config:
pool_obj['config'] = config
raw_output.append(pool_obj) raw_output.append(pool_obj)
return raw_output if raw else _process(raw_output) return raw_output if raw else _process(raw_output)