python/jc
1
0
Fork 0
mirror of https://github.com/kellyjonbrazil/jc.git synced 2025-06-17 00:07:37 +02:00
Code Issues Releases Activity

add input type checking

Browse Source
This commit is contained in:
Kelly Brazil
2021-11-30 09:51:27 -08:00
parent cfb58b1cf3
commit caf0a5c871
2 changed files with 7 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
jc/parsers/vmstat_s.py
View File

@ -145,10 +145,10 @@ def parse(data, raw=False, quiet=False, ignore_exceptions=False):
Iterator object
"""
if not quiet:
jc.utils.compatibility(__name__, info.compatible)
if not quiet: jc.utils.compatibility(__name__, info.compatible)
if not hasattr(data, '__iter__') or isinstance(data, (str, bytes)):
raise TypeError("Input data must be a non-string iterable object.")
output_line = {}
procs = None
buff_cache = None
disk = None
@ -156,7 +156,10 @@ def parse(data, raw=False, quiet=False, ignore_exceptions=False):
tz = None
for line in data:
output_line = {}
try:
if not isinstance(line, str): raise TypeError("Input line must be a 'str' object.")
# skip blank lines
if line.strip() == '':
continue

2
tests/test_vmstat_s.py
View File

@ -75,7 +75,7 @@ class MyTests(unittest.TestCase):
"""
Test 'vmstat' with no data
"""
self.assertEqual(list(jc.parsers.vmstat_s.parse('', quiet=True)), [])
self.assertEqual(list(jc.parsers.vmstat_s.parse([], quiet=True)), [])
def test_vmstat_s_unparsable(self):
data = 'unparsable data'