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 checks

Browse Source
This commit is contained in:
Kelly Brazil
2021-11-30 09:56:33 -08:00
parent caf0a5c871
commit 3a9f0934c4
2 changed files with 5 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
jc/parsers/ls_s.py
View File

@ -123,13 +123,15 @@ def parse(data, raw=False, quiet=False, ignore_exceptions=False):
Iterator object Iterator object
""" """
if not quiet: if not quiet: jc.utils.compatibility(__name__, info.compatible)
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.")
parent = '' parent = ''
for line in data: for line in data:
try: try:
if not isinstance(line, str): raise TypeError("Input line must be a 'str' object.")
# skip line if it starts with 'total 1234' # skip line if it starts with 'total 1234'
if re.match(r'total [0-9]+', line): if re.match(r'total [0-9]+', line):

2
tests/test_ls_s.py
View File

@ -103,7 +103,7 @@ class MyTests(unittest.TestCase):
""" """
Test plain 'ls' on an empty directory Test plain 'ls' on an empty directory
""" """
self.assertEqual(list(jc.parsers.ls_s.parse('', quiet=True)), []) self.assertEqual(list(jc.parsers.ls_s.parse([], quiet=True)), [])
def test_ls_s_centos_7_7_raise_exception(self): def test_ls_s_centos_7_7_raise_exception(self):
""" """