160 lines
4.8 KiB
Python
160 lines
4.8 KiB
Python
from datetime import datetime
|
|
import os.path
|
|
import random
|
|
from BeautifulSoup import BeautifulSoup
|
|
|
|
from django.shortcuts import render_to_response
|
|
from django.template import RequestContext
|
|
from django.http import HttpResponse
|
|
from django.utils.functional import Promise
|
|
from django.utils.translation import force_unicode
|
|
from django.utils.simplejson import JSONEncoder
|
|
from django import forms
|
|
from django.template.defaultfilters import urlize as django_urlize
|
|
|
|
|
|
def render_to(template_path):
|
|
"""
|
|
Expect the dict from view. Render returned dict with
|
|
RequestContext.
|
|
"""
|
|
|
|
def decorator(func):
|
|
def wrapper(request, *args, **kwargs):
|
|
import pdb
|
|
#output = pdb.runcall(func, request, *args, **kwargs)
|
|
output = func(request, *args, **kwargs)
|
|
if not isinstance(output, dict):
|
|
return output
|
|
kwargs = {'context_instance': RequestContext(request)}
|
|
if 'MIME_TYPE' in output:
|
|
kwargs['mimetype'] = output.pop('MIME_TYPE')
|
|
if 'TEMPLATE' in output:
|
|
template = output.pop('TEMPLATE')
|
|
else:
|
|
template = template_path
|
|
return render_to_response(template, output, **kwargs)
|
|
return wrapper
|
|
|
|
return decorator
|
|
|
|
|
|
def paged(paged_list_name, per_page):#, per_page_var='per_page'):
|
|
"""
|
|
Parse page from GET data and pass it to view. Split the
|
|
query set returned from view.
|
|
"""
|
|
|
|
def decorator(func):
|
|
def wrapper(request, *args, **kwargs):
|
|
result = func(request, *args, **kwargs)
|
|
if not isinstance(result, dict):
|
|
return result
|
|
try:
|
|
page = int(request.GET.get('page', 1))
|
|
except ValueError:
|
|
page = 1
|
|
|
|
real_per_page = per_page
|
|
|
|
#if per_page_var:
|
|
#try:
|
|
#value = int(request.GET[per_page_var])
|
|
#except (ValueError, KeyError):
|
|
#pass
|
|
#else:
|
|
#if value > 0:
|
|
#real_per_page = value
|
|
|
|
from django.core.paginator import Paginator
|
|
paginator = Paginator(result['paged_qs'], real_per_page)
|
|
result[paged_list_name] = paginator.page(page).object_list
|
|
result['page'] = page
|
|
result['page_list'] = range(1, paginator.num_pages + 1)
|
|
result['pages'] = paginator.num_pages
|
|
result['per_page'] = real_per_page
|
|
result['request'] = request
|
|
return result
|
|
return wrapper
|
|
|
|
return decorator
|
|
|
|
|
|
def ajax(func):
|
|
"""
|
|
Checks request.method is POST. Return error in JSON in other case.
|
|
|
|
If view returned dict, returns JsonResponse with this dict as content.
|
|
"""
|
|
def wrapper(request, *args, **kwargs):
|
|
if request.method == 'POST':
|
|
try:
|
|
response = func(request, *args, **kwargs)
|
|
except Exception, ex:
|
|
response = {'error': traceback.format_exc()}
|
|
else:
|
|
response = {'error': {'type': 403, 'message': 'Accepts only POST request'}}
|
|
if isinstance(response, dict):
|
|
return JsonResponse(response)
|
|
else:
|
|
return response
|
|
return wrapper
|
|
|
|
|
|
class LazyJSONEncoder(JSONEncoder):
|
|
"""
|
|
This fing need to save django from crashing.
|
|
"""
|
|
|
|
def default(self, o):
|
|
if isinstance(o, Promise):
|
|
return force_unicode(o)
|
|
else:
|
|
return super(LazyJSONEncoder, self).default(o)
|
|
|
|
|
|
class JsonResponse(HttpResponse):
|
|
"""
|
|
HttpResponse subclass that serialize data into JSON format.
|
|
"""
|
|
|
|
def __init__(self, data, mimetype='application/json'):
|
|
json_data = LazyJSONEncoder().encode(data)
|
|
super(JsonResponse, self).__init__(
|
|
content=json_data, mimetype=mimetype)
|
|
|
|
|
|
def build_form(Form, _request, GET=False, *args, **kwargs):
|
|
"""
|
|
Shorcut for building the form instance of given form class.
|
|
"""
|
|
|
|
if not GET and 'POST' == _request.method:
|
|
form = Form(_request.POST, _request.FILES, *args, **kwargs)
|
|
elif GET and 'GET' == _request.method:
|
|
form = Form(_request.GET, _request.FILES, *args, **kwargs)
|
|
else:
|
|
form = Form(*args, **kwargs)
|
|
return form
|
|
|
|
def urlize(data):
|
|
"""
|
|
Urlize plain text links in the HTML contents.
|
|
|
|
Do not urlize content of A and CODE tags.
|
|
"""
|
|
|
|
soup = BeautifulSoup(data)
|
|
for chunk in soup.findAll(text=True):
|
|
islink = False
|
|
ptr = chunk.parent
|
|
while ptr.parent:
|
|
if ptr.name == 'a' or ptr.name == 'code':
|
|
islink = True
|
|
break
|
|
ptr = ptr.parent
|
|
|
|
if not islink:
|
|
chunk = chunk.replaceWith(django_urlize(unicode(chunk)))
|
|
|
|
return unicode(soup)
|