parser: allow script to be empty

idank · Jan 17, 2024 · ea0c135 · ea0c135
1 parent 81a0580
commit ea0c135
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 0 deletions.
diff --git a/bashlex/parser.py b/bashlex/parser.py
@@ -647,6 +647,8 @@ def parse(s, strictmode=True, expansionlimit=None, convertpos=False, proceedoner
     '''
     p = _parser(s, strictmode=strictmode, expansionlimit=expansionlimit, proceedonerror=proceedonerror)
     parts = [p.parse()]
+    if parts[0] is None or isinstance(parts[0], str):
+        return []
 
     class endfinder(ast.nodevisitor):
         def __init__(self):

diff --git a/tests/test_parser.py b/tests/test_parser.py
@@ -125,6 +125,22 @@ def assertASTsEquals(self, s, expectedlist, **parserargs):
             msg = 'ASTs not equal for %r\n\nresult:\n\n%s\n\n!=\n\nexpected:\n\n%s' % (s, result.dump(), expected.dump())
             self.assertEqual(result, expected, msg)
 
+    def test_empty(self):
+        s = ''
+        self.assertASTsEquals(s, [])
+
+        s = '\n'
+        self.assertASTsEquals(s, [])
+
+        s = ' '
+        self.assertASTsEquals(s, [])
+
+        s = '# Comment'
+        self.assertASTsEquals(s, [])
+
+        s = '# Comment\n# Another comment'
+        self.assertASTsEquals(s, [])
+
     def test_command(self):
         s = 'a b c'
         self.assertASTEquals(s,