142 lines
6.2 KiB
Python
142 lines
6.2 KiB
Python
import json
|
|
from django.conf import settings
|
|
from django.conf.urls.defaults import patterns, url
|
|
from django.contrib.admin import ModelAdmin, TabularInline
|
|
from django.contrib.contenttypes.models import ContentType
|
|
from django.http import HttpResponse
|
|
from django.shortcuts import render
|
|
from django.template.defaultfilters import capfirst
|
|
from django.views.decorators.csrf import csrf_exempt
|
|
|
|
from adminsortable.models import Sortable
|
|
|
|
STATIC_URL = settings.STATIC_URL
|
|
|
|
|
|
class SortableAdmin(ModelAdmin):
|
|
ordering = ('order', 'id')
|
|
|
|
class Meta:
|
|
abstract = True
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
super(SortableAdmin, self).__init__(*args, **kwargs)
|
|
self.has_sortable_tabular_inlines = False
|
|
for klass in self.inlines:
|
|
if issubclass(klass, SortableTabularInline):
|
|
self.has_sortable_tabular_inlines = True
|
|
|
|
def get_urls(self):
|
|
urls = super(SortableAdmin, self).get_urls()
|
|
admin_urls = patterns('',
|
|
url(r'^sorting/do-sorting/(?P<model_type_id>\d+)/$',
|
|
self.admin_site.admin_view(self.do_sorting_view),
|
|
name='admin_do_sorting'), #this view changes the order
|
|
url(r'^sort/$', self.admin_site.admin_view(self.sort_view),
|
|
name='admin_sort'), #this view shows a link to the drag-and-drop view
|
|
)
|
|
return admin_urls + urls
|
|
|
|
def sort_view(self, request):
|
|
"""
|
|
Custom admin view that displays the objects as a list whose sort order can be
|
|
changed via drag-and-drop.
|
|
"""
|
|
opts = self.model._meta
|
|
admin_site = self.admin_site
|
|
has_perm = request.user.has_perm(opts.app_label + '.' + opts.get_change_permission())
|
|
objects = self.model.objects.all()
|
|
|
|
"""
|
|
Determine if we need to regroup objects relative to a foreign key specified on the
|
|
model class that is extending Sortable.
|
|
"""
|
|
sortable_by = getattr(self.model, 'sortable_by', None)
|
|
if sortable_by:
|
|
sortable_by_class, sortable_by_expression = sortable_by()
|
|
sortable_by_class_display_name = sortable_by_class._meta.verbose_name_plural
|
|
sortable_by_class_is_sortable = sortable_by_class.is_sortable()
|
|
else:
|
|
sortable_by_class = sortable_by_expression = sortable_by_class_display_name = \
|
|
sortable_by_class_is_sortable = None
|
|
|
|
try:
|
|
verbose_name_plural = opts.verbose_name_plural.__unicode__()
|
|
except AttributeError:
|
|
verbose_name_plural = opts.verbose_name_plural
|
|
|
|
context = {
|
|
'title' : 'Drag and drop %s to change display order' % capfirst(verbose_name_plural),
|
|
'opts' : opts,
|
|
'root_path' : '/%s' % admin_site.root_path,
|
|
'app_label' : opts.app_label,
|
|
'has_perm' : has_perm,
|
|
'objects' : objects,
|
|
'group_expression' : sortable_by_expression,
|
|
'sortable_by_class' : sortable_by_class,
|
|
'sortable_by_class_is_sortable' : sortable_by_class_is_sortable,
|
|
'sortable_by_class_display_name' : sortable_by_class_display_name
|
|
}
|
|
return render(request, 'adminsortable/change_list.html', context)
|
|
|
|
def changelist_view(self, request, extra_context=None):
|
|
"""
|
|
If the model that inherits Sortable has more than one object,
|
|
its sort order can be changed. This view adds a link to the object_tools
|
|
block to take people to the view to change the sorting.
|
|
"""
|
|
if self.model.is_sortable():
|
|
self.change_list_template = 'adminsortable/change_list_with_sort_link.html'
|
|
return super(SortableAdmin, self).changelist_view(request, extra_context=extra_context)
|
|
|
|
def change_view(self, request, object_id, extra_context=None):
|
|
has_sortable_tabular_inlines = self.has_sortable_tabular_inlines
|
|
if has_sortable_tabular_inlines:
|
|
self.change_form_template = 'adminsortable/change_form.html'
|
|
extra_context = {'has_sortable_tabular_inlines' : self.has_sortable_tabular_inlines}
|
|
return super(SortableAdmin, self).change_view(request, object_id, extra_context=extra_context)
|
|
|
|
@csrf_exempt
|
|
def do_sorting_view(self, request, model_type_id=None):
|
|
"""
|
|
This view sets the ordering of the objects for the model type and primary keys
|
|
passed in. It must be an Ajax POST.
|
|
"""
|
|
|
|
if request.is_ajax() and request.method == 'POST':
|
|
try:
|
|
indexes = request.POST.get('indexes', []).split(',')
|
|
klass = ContentType.objects.get(id=model_type_id).model_class()
|
|
objects_dict = dict([(obj.pk, obj) for obj in klass.objects.filter(pk__in=indexes)])
|
|
min_index = min(objects_dict.values(), key=lambda x: getattr(x, 'order'))
|
|
min_index = getattr(min_index, 'order') or 0
|
|
for index in indexes:
|
|
obj = objects_dict[int(index)]
|
|
setattr(obj, 'order', min_index)
|
|
obj.save()
|
|
min_index += 1
|
|
response = {'objects_sorted' : True}
|
|
except (IndexError, klass.DoesNotExist, AttributeError):
|
|
pass
|
|
else:
|
|
response = {'objects_sorted' : False}
|
|
return HttpResponse(json.dumps(response, ensure_ascii=False),
|
|
mimetype='application/json')
|
|
|
|
|
|
class SortableTabularInline(TabularInline):
|
|
"""Custom template that enables sorting for tabular inlines"""
|
|
def __init__(self, *args, **kwargs):
|
|
super(SortableTabularInline, self).__init__(*args, **kwargs)
|
|
|
|
if not issubclass(self.model, Sortable):
|
|
raise Warning(u'Models that inherit SortableTabluarInline must inherit from Sortable')
|
|
|
|
"""
|
|
This property is referenced by tabular.html's <h2> to show a message on whether or
|
|
not the inlines are sortable. It is exposed in: inline_admin_formset.opts
|
|
"""
|
|
self.is_sortable = self.model.is_sortable()
|
|
|
|
template = 'adminsortable/edit_inline/tabular.html'
|