Fix shebang regex in Python lexer (#1172)

The Python lexer only supported minor version numbers for shebangs involving `python2`. Minor versions of Python 3 should also be included. This commit corrects that problem and supports minor version numbers higher than 9.
rouge-ruby · Jun 17, 2019 · 5fb0ea6 · 5fb0ea6
1 parent 48daca1
commit 5fb0ea6
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/lib/rouge/lexers/python.rb b/lib/rouge/lexers/python.rb
@@ -12,7 +12,7 @@ class Python < RegexLexer
       mimetypes 'text/x-python', 'application/x-python'
 
       def self.detect?(text)
-        return true if text.shebang?(/pythonw?(3|2(\.\d)?)?/)
+        return true if text.shebang?(/pythonw?(?:[23](?:\.\d+)?)?/)
       end
 
       def self.keywords

diff --git a/spec/lexers/python_spec.rb b/spec/lexers/python_spec.rb
@@ -23,9 +23,11 @@
 
     it 'guesses by source' do
       assert_guess :source => '#!/usr/bin/env python'
-      assert_guess :source => '#!/usr/local/bin/python3'
       assert_guess :source => '#!/usr/bin/python2'
       assert_guess :source => '#!/usr/bin/python2.7'
+      assert_guess :source => '#!/usr/local/bin/python3'
+      assert_guess :source => '#!/usr/local/bin/python3.5'
+      assert_guess :source => '#!/usr/local/bin/python3.14'
       deny_guess   :source => '#!/usr/bin/env python4'
     end
   end