django-rest-framework/examples/pygments_api/views.py

90 lines
3.8 KiB
Python
Raw Normal View History

from __future__ import with_statement # for python 2.5
2011-01-26 11:58:09 +03:00
from django.conf import settings
from django.core.urlresolvers import reverse
2011-01-26 11:58:09 +03:00
from djangorestframework.resource import Resource
2011-02-19 16:54:55 +03:00
from djangorestframework.response import Response
from djangorestframework.emitters import BaseEmitter
2011-02-19 16:54:55 +03:00
from djangorestframework import status
2011-01-26 11:58:09 +03:00
from pygments.formatters import HtmlFormatter
from pygments.lexers import get_lexer_by_name
from pygments import highlight
from forms import PygmentsForm
import os
import uuid
import operator
2011-01-26 11:58:09 +03:00
# We need somewhere to store the code that we highlight
HIGHLIGHTED_CODE_DIR = os.path.join(settings.MEDIA_ROOT, 'pygments')
2011-03-23 01:48:59 +03:00
MAX_FILES = 10
def list_dir_sorted_by_ctime(dir):
"""Return a list of files sorted by creation time"""
2011-03-23 01:48:59 +03:00
filepaths = [os.path.join(dir, file) for file in os.listdir(dir) if not file.startswith('.')]
return [item[0] for item in sorted([(path, os.path.getctime(path)) for path in filepaths],
key=operator.itemgetter(1), reverse=False)]
def remove_oldest_files(dir, max_files):
2011-01-30 19:51:06 +03:00
"""Remove the oldest files in a directory 'dir', leaving at most 'max_files' remaining.
We use this to limit the number of resources in the sandbox."""
[os.remove(path) for path in list_dir_sorted_by_ctime(dir)[max_files:]]
2011-01-26 11:58:09 +03:00
class HTMLEmitter(BaseEmitter):
"""Basic emitter which just returns the content without any further serialization."""
media_type = 'text/html'
class PygmentsRoot(Resource):
"""This example demonstrates a simple RESTful Web API aound the awesome pygments library.
This top level resource is used to create highlighted code snippets, and to list all the existing code snippets."""
2011-01-26 11:58:09 +03:00
form = PygmentsForm
allowed_methods = anon_allowed_methods = ('GET', 'POST',)
def get(self, request, auth):
"""Return a list of all currently existing snippets."""
unique_ids = [os.path.split(f)[1] for f in list_dir_sorted_by_ctime(HIGHLIGHTED_CODE_DIR)]
return [reverse('pygments-instance', args=[unique_id]) for unique_id in unique_ids]
2011-01-26 11:58:09 +03:00
def post(self, request, auth, content):
"""Create a new highlighed snippet and return it's location.
For the purposes of the sandbox example, also ensure we delete the oldest snippets if we have > MAX_FILES."""
unique_id = str(uuid.uuid1())
2011-01-26 11:58:09 +03:00
pathname = os.path.join(HIGHLIGHTED_CODE_DIR, unique_id)
lexer = get_lexer_by_name(content['lexer'])
linenos = 'table' if content['linenos'] else False
options = {'title': content['title']} if content['title'] else {}
formatter = HtmlFormatter(style=content['style'], linenos=linenos, full=True, **options)
with open(pathname, 'w') as outfile:
highlight(content['code'], lexer, formatter, outfile)
remove_oldest_files(HIGHLIGHTED_CODE_DIR, MAX_FILES)
return Response(status.HTTP_201_CREATED, headers={'Location': reverse('pygments-instance', args=[unique_id])})
2011-01-26 11:58:09 +03:00
class PygmentsInstance(Resource):
"""Simply return the stored highlighted HTML file with the correct mime type.
This Resource only emits HTML and uses a standard HTML emitter rather than the emitters.DocumentingHTMLEmitter class."""
2011-01-26 11:58:09 +03:00
allowed_methods = anon_allowed_methods = ('GET',)
emitters = (HTMLEmitter,)
def get(self, request, auth, unique_id):
"""Return the highlighted snippet."""
2011-01-26 11:58:09 +03:00
pathname = os.path.join(HIGHLIGHTED_CODE_DIR, unique_id)
if not os.path.exists(pathname):
return Resource(status.HTTP_404_NOT_FOUND)
return open(pathname, 'r').read()
2011-01-30 19:51:06 +03:00
def delete(self, request, auth, unique_id):
"""Delete the highlighted snippet."""
pathname = os.path.join(HIGHLIGHTED_CODE_DIR, unique_id)
if not os.path.exists(pathname):
return Resource(status.HTTP_404_NOT_FOUND)
return os.remove(pathname)
2011-01-26 11:58:09 +03:00