Fix shebang regex in Python lexer

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. This fixes rouge-ruby#945.
pyrmont · Jun 12, 2019 · 87f302b · 87f302b
1 parent 2151912
commit 87f302b
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