119 lines
3.2 KiB
Python
119 lines
3.2 KiB
Python
from django.contrib.contenttypes import generic
|
|
from django.contrib.contenttypes.models import ContentType
|
|
from django.db import models
|
|
|
|
from adminsortable.fields import SortableForeignKey
|
|
from adminsortable.models import Sortable
|
|
|
|
|
|
class SimpleModel(models.Model):
|
|
class Meta:
|
|
abstract = True
|
|
|
|
title = models.CharField(max_length=50)
|
|
|
|
def __unicode__(self):
|
|
return self.title
|
|
|
|
|
|
# A model that is sortable
|
|
class Category(SimpleModel, Sortable):
|
|
class Meta(Sortable.Meta):
|
|
"""
|
|
Classes that inherit from Sortable must define an inner
|
|
Meta class that inherits from Sortable.Meta or ordering
|
|
won't work as expected
|
|
"""
|
|
verbose_name_plural = 'Categories'
|
|
|
|
|
|
# A model with an override of its queryset for admin
|
|
class Widget(SimpleModel, Sortable):
|
|
class Meta(Sortable.Meta):
|
|
pass
|
|
|
|
def __unicode__(self):
|
|
return self.title
|
|
|
|
|
|
# A model that is sortable relative to a foreign key that is also sortable
|
|
# uses SortableForeignKey field. Works with versions 1.3+
|
|
class Project(SimpleModel, Sortable):
|
|
class Meta(Sortable.Meta):
|
|
pass
|
|
|
|
category = SortableForeignKey(Category)
|
|
description = models.TextField()
|
|
|
|
|
|
# Registered as a tabular inline on `Project`
|
|
class Credit(Sortable):
|
|
class Meta(Sortable.Meta):
|
|
pass
|
|
|
|
project = models.ForeignKey(Project)
|
|
first_name = models.CharField(max_length=30)
|
|
last_name = models.CharField(max_length=30)
|
|
|
|
def __unicode__(self):
|
|
return '{0} {1}'.format(self.first_name, self.last_name)
|
|
|
|
|
|
# Registered as a stacked inline on `Project`
|
|
class Note(Sortable):
|
|
class Meta(Sortable.Meta):
|
|
pass
|
|
|
|
project = models.ForeignKey(Project)
|
|
text = models.CharField(max_length=100)
|
|
|
|
def __unicode__(self):
|
|
return self.text
|
|
|
|
|
|
# A generic bound model
|
|
class GenericNote(SimpleModel, Sortable):
|
|
content_type = models.ForeignKey(ContentType,
|
|
verbose_name=u"Content type", related_name="generic_notes")
|
|
object_id = models.PositiveIntegerField(u"Content id")
|
|
content_object = generic.GenericForeignKey(ct_field='content_type',
|
|
fk_field='object_id')
|
|
|
|
class Meta(Sortable.Meta):
|
|
pass
|
|
|
|
def __unicode__(self):
|
|
return u'{}: {}'.format(self.title, self.content_object)
|
|
|
|
|
|
# An model registered as an inline that has a custom queryset
|
|
class Component(SimpleModel, Sortable):
|
|
class Meta(Sortable.Meta):
|
|
pass
|
|
|
|
widget = SortableForeignKey(Widget)
|
|
|
|
def __unicode__(self):
|
|
return self.title
|
|
|
|
|
|
class Person(Sortable):
|
|
class Meta(Sortable.Meta):
|
|
verbose_name_plural = 'People'
|
|
|
|
first_name = models.CharField(max_length=50)
|
|
last_name = models.CharField(max_length=50)
|
|
is_board_member = models.BooleanField('Board Member', default=False)
|
|
|
|
# Sorting Filters allow you to set up sub-sets of objects that need
|
|
# to have independent sorting. They are listed in order, from left
|
|
# to right in the sorting change list. You can use any standard
|
|
# Django ORM filter method.
|
|
sorting_filters = (
|
|
('Board Members', {'is_board_member': True}),
|
|
('Non-Board Members', {'is_board_member': False}),
|
|
)
|
|
|
|
def __unicode__(self):
|
|
return '{} {}'.format(self.first_name, self.last_name)
|