This commit is contained in:
Reinout van Rees 2013-01-07 13:39:05 -08:00
commit 1f8b2b4a6f
3 changed files with 26 additions and 1 deletions

View File

@ -383,6 +383,20 @@ try:
except ImportError:
apply_markdown = None
# Restructured text (docutils) is optional
try:
import docutils.core
def apply_restructuredtext(text):
parts = docutils.core.publish_parts(
text,
writer_name='html',
settings_overrides={'initial_header_level': 2,
'doctitle_xform': False})
return parts['html_body']
except ImportError:
apply_restructuredtext = None
# Yaml is optional
try:

View File

@ -1,6 +1,7 @@
from django.test import TestCase
from rest_framework.views import APIView
from rest_framework.compat import apply_markdown
from rest_framework.compat import apply_restructuredtext
# We check that docstrings get nicely un-indented.
DESCRIPTION = """an example docstring
@ -111,3 +112,11 @@ class TestViewNamesAndDescriptions(TestCase):
gte_21_match = apply_markdown(DESCRIPTION) == MARKED_DOWN_gte_21
lt_21_match = apply_markdown(DESCRIPTION) == MARKED_DOWN_lt_21
self.assertTrue(gte_21_match or lt_21_match)
def test_restructuredtext(self):
"""Ensure restructuredtext to HTML works as expected."""
if apply_restructuredtext:
# The output isn't tested verbatim because of small rendering changes
# between docutils versions.
self.assertTrue('<h3>another header</h3>'
in apply_restructuredtext(DESCRIPTION))

View File

@ -9,7 +9,7 @@ from django.utils.html import escape
from django.utils.safestring import mark_safe
from django.views.decorators.csrf import csrf_exempt
from rest_framework import status, exceptions
from rest_framework.compat import View, apply_markdown
from rest_framework.compat import View, apply_markdown, apply_restructuredtext
from rest_framework.response import Response
from rest_framework.request import Request
from rest_framework.settings import api_settings
@ -119,6 +119,8 @@ class APIView(View):
# TODO: deprecate?
if apply_markdown:
description = apply_markdown(description)
elif apply_restructuredtext:
description = apply_restructuredtext(description)
else:
description = escape(description).replace('\n', '<br />')
return mark_safe(description)