Merge pull request lovesegfault#26 from oliverhulett/oliver/support-h…

…ere-strings Here strings not just after loops
bollwyvl · Aug 16, 2018 · 634530f · 634530f
2 parents 991569c + 8362d56
commit 634530f
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)) and not (re.search(r'done.*<<<', test_record)):
+                if(re.search(r'<<-?', test_record)) and not (re.search(r'.*<<<', test_record)):
                     here_string = re.sub(
                         r'.*<<-?\s*[\'|"]?([_|\w]+)[\'|"]?.*', r'\1',
                         stripped_record, 1)