Merge pull request #24 from streambinder/master

Prevent specific while loop to be recognized as heredoc
lovesegfault · Feb 20, 2018 · c1ff0d4 · c1ff0d4
2 parents c66c2f8 + 9e98979
commit c1ff0d4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/beautysh/beautysh.py b/beautysh/beautysh.py
@@ -77,7 +77,7 @@ def beautify_string(self, data, path=''):
                    re.search(r'<<', test_record)):
                     in_here_doc = False
             else:  # not in here doc
-                if(re.search(r'<<-?', test_record)):
+                if(re.search(r'<<-?', test_record)) and not (re.search(r'done.*<<<', test_record)):
                     here_string = re.sub(
                         r'.*<<-?\s*[\'|"]?([_|\w]+)[\'|"]?.*', r'\1',
                         stripped_record, 1)