Use Pygments API

ksato9700 · Oct 4, 2014 · 5a44ed9 · 5a44ed9
1 parent f0c5b71
commit 5a44ed9
Showing 1 changed file with 10 additions and 10 deletions.
diff --git a/markdown/extensions/codehilite.py b/markdown/extensions/codehilite.py
@@ -22,8 +22,8 @@
 import warnings
 try:
     from pygments import highlight
-    from pygments.lexers import get_lexer_by_name, guess_lexer, TextLexer
-    from pygments.formatters import HtmlFormatter
+    from pygments.lexers import get_lexer_by_name, guess_lexer
+    from pygments.formatters import get_formatter_by_name
     pygments = True
 except ImportError:
     pygments = False
@@ -109,14 +109,15 @@ def hilite(self):
                     if self.guess_lang:
                         lexer = guess_lexer(self.src)
                     else:
-                        lexer = TextLexer()
+                        lexer = get_lexer_by_name('text')
                 except ValueError:
-                    lexer = TextLexer()
-            formatter = HtmlFormatter(linenos=self.linenums,
-                                      cssclass=self.css_class,
-                                      style=self.style,
-                                      noclasses=self.noclasses,
-                                      hl_lines=self.hl_lines)
+                    lexer = get_lexer_by_name('text')
+            formatter = get_formatter_by_name('html',
+                                              linenos=self.linenums,
+                                              cssclass=self.css_class,
+                                              style=self.style,
+                                              noclasses=self.noclasses,
+                                              hl_lines=self.hl_lines)
             return highlight(self.src, lexer, formatter)
         else:
             # just escape and build markup usable by JS highlighting libs
@@ -247,4 +248,3 @@ def extendMarkdown(self, md, md_globals):
 
 def makeExtension(*args, **kwargs):
   return CodeHiliteExtension(*args, **kwargs)
-