Merged in hroncok/pygments-main/doscon_prompt (pull request #734)

Recognize single > as a prompt in doscon

2 files changed, 19 insertions, 2 deletions

diff --git a/pygments/lexers/shell.py b/pygments/lexers/shell.py
index 31bc7e94..9fd6bf20 100644
--- a/pygments/lexers/shell.py
+++ b/pygments/lexers/shell.py
@@ -540,7 +540,7 @@ class MSDOSSessionLexer(ShellSessionBaseLexer):
     mimetypes = []
 
     _innerLexerCls = BatchLexer
-    _ps1rgx = r'^([^>]+>)(.*\n?)'
+    _ps1rgx = r'^([^>]*>)(.*\n?)'
     _ps2 = 'More? '
 
 
diff --git a/tests/test_shell.py b/tests/test_shell.py
index e283793e..1121240a 100644
--- a/tests/test_shell.py
+++ b/tests/test_shell.py
@@ -10,7 +10,7 @@
 import unittest
 
 from pygments.token import Token
-from pygments.lexers import BashLexer, BashSessionLexer
+from pygments.lexers import BashLexer, BashSessionLexer, MSDOSSessionLexer
 
 
 class BashTest(unittest.TestCase):
@@ -140,3 +140,20 @@ class BashSessionTest(unittest.TestCase):
         ]
         self.assertEqual(tokens, list(self.lexer.get_tokens(fragment)))
 
+class MSDOSSessionTest(unittest.TestCase):
+
+    def setUp(self):
+        self.lexer = MSDOSSessionLexer()
+
+    def testGtOnlyPrompt(self):
+        fragment = u'> py\nhi\n'
+        tokens = [
+            (Token.Text, u''),
+            (Token.Generic.Prompt, u'>'),
+            (Token.Text, u' '),
+            (Token.Text, u'py'),
+            (Token.Text, u''),
+            (Token.Text, u'\n'),
+            (Token.Generic.Output, u'hi\n'),
+        ]
+        self.assertEqual(tokens, list(self.lexer.get_tokens(fragment)))