I've been trying to add syntax highlighting to my django site. The problem is I'm getting the
and <br />
characters formatted as well. Is there a way to preserve these characterss? Here is the code I'm using:
from BeautifulSoup import BeautifulSoup
from django import template
from django.template.defaultfilters import stringfilter
import pygments
import pygments.formatters
import pygments.lexers
register = template.Library()
@register.filter
@stringfilter
def pygmentized(html):
soup = BeautifulSoup(html)
codeblocks = soup.findAll('code')
for block in codeblocks:
if block.has_key('class'):
try:
code = ''.join([unicode(item) for item in block.contents])
lexer = pygments.lexers.get_lexer_by_name(block['class'], stripall=True)
formatter = pygments.formatters.Html开发者_如何学JAVAFormatter()
code_hl = pygments.highlight(code, lexer, formatter)
block.contents = [BeautifulSoup(code_hl)]
block.name = 'code'
except:
raise
return unicode(soup)
Well, Petri is right, pre is for code blocks. Before he pointed that out I just wrote a function to clean the first output, it's messy but maybe someone who just needs to remove certain things from the final output might find it ok:
from BeautifulSoup import BeautifulSoup
from django import template
from django.template.defaultfilters import stringfilter
import pygments
import pygments.formatters
import pygments.lexers
register = template.Library()
wanted = {'br': '<br />', 'BR': '<BR />', 'nbsp': ' ', 'NBSP': '&NBSP;', '/>': ''}
def uglyfilter(html):
content = BeautifulSoup(html)
for node in content.findAll('span'):
data = ''.join(node.findAll(text=True))
if wanted.has_key(data):
node.replaceWith(wanted.get(data))
return unicode(content)
@register.filter
@stringfilter
def pygmentized(html):
soup = BeautifulSoup(html)
codeblocks = soup.findAll('pre')
for block in codeblocks:
if block.has_key('class'):
try:
code = ''.join([unicode(item) for item in block.contents])
lexer = pygments.lexers.get_lexer_by_name(block['class'], stripall=True)
formatter = pygments.formatters.HtmlFormatter()
code_hl = pygments.highlight(code, lexer, formatter)
clean = uglyfilter(code_hl)
block.contents = [BeautifulSoup(clean)]
block.name = 'pre'
except:
raise
return unicode(soup)
精彩评论