143 lines
4.9 KiB
Python
143 lines
4.9 KiB
Python
from django.template.defaultfilters import slugify
|
|
from django.utils.safestring import mark_safe
|
|
from django.contrib import admin
|
|
from django import forms
|
|
|
|
from models import *
|
|
from ietf.person.models import *
|
|
from ietf.doc.utils import get_state_types
|
|
from ietf.utils.admin import admin_link
|
|
|
|
class StateTypeAdmin(admin.ModelAdmin):
|
|
list_display = ["slug", "label"]
|
|
admin.site.register(StateType, StateTypeAdmin)
|
|
|
|
class StateAdmin(admin.ModelAdmin):
|
|
list_display = ["slug", "type", 'name', 'order', 'desc']
|
|
filter_horizontal = ["next_states"]
|
|
admin.site.register(State, StateAdmin)
|
|
|
|
class DocAliasInline(admin.TabularInline):
|
|
model = DocAlias
|
|
extra = 1
|
|
|
|
class DocAuthorInline(admin.TabularInline):
|
|
model = DocumentAuthor
|
|
raw_id_fields = ['author', ]
|
|
extra = 1
|
|
|
|
# document form for managing states in a less confusing way
|
|
|
|
class StatesWidget(forms.SelectMultiple):
|
|
"""Display all applicable states as separate select boxes,
|
|
requires 'instance' have been set on the widget."""
|
|
def render(self, name, value, attrs=None, choices=()):
|
|
|
|
types = StateType.objects.filter(slug__in=get_state_types(self.instance)).order_by("slug")
|
|
|
|
categorized_choices = []
|
|
for t in types:
|
|
states = State.objects.filter(type=t).select_related()
|
|
if states:
|
|
categorized_choices.append((t.label, states))
|
|
|
|
html = []
|
|
first = True
|
|
for label, states in categorized_choices:
|
|
htmlid = "id_%s_%s" % (name, slugify(label))
|
|
|
|
html.append('<div style="clear:both;padding-top:%s">' % ("1em" if first else "0.5em"))
|
|
html.append(u'<label for="%s">%s:</label>' % (htmlid, label))
|
|
html.append(u'<select name="%s" id="%s">' % (name, htmlid))
|
|
html.append(u'<option value="">-----------</option>')
|
|
for s in states:
|
|
html.append('<option %s value="%s">%s</option>' % ("selected" if s.pk in value else "", s.pk, s.name))
|
|
html.append(u'</select>')
|
|
html.append("</div>")
|
|
|
|
first = False
|
|
|
|
return mark_safe(u"".join(html))
|
|
|
|
class StatesField(forms.ModelMultipleChoiceField):
|
|
def __init__(self, *args, **kwargs):
|
|
# use widget with multiple select boxes
|
|
kwargs['widget'] = StatesWidget
|
|
super(StatesField, self).__init__(*args, **kwargs)
|
|
|
|
def clean(self, value):
|
|
if value and isinstance(value, (list, tuple)):
|
|
# remove "", in case a state is reset
|
|
value = [x for x in value if x]
|
|
return super(StatesField, self).clean(value)
|
|
|
|
class DocumentForm(forms.ModelForm):
|
|
states = StatesField(queryset=State.objects.all(), required=False)
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
super(DocumentForm, self).__init__(*args, **kwargs)
|
|
|
|
# we don't normally have access to the instance in the widget
|
|
# so set it here
|
|
self.fields["states"].widget.instance = self.instance
|
|
|
|
class Meta:
|
|
model = Document
|
|
|
|
class DocumentAdmin(admin.ModelAdmin):
|
|
list_display = ['name', 'rev', 'group', 'pages', 'intended_std_level', 'author_list', 'time']
|
|
search_fields = ['name']
|
|
list_filter = ['type']
|
|
raw_id_fields = ['authors', 'related', 'group', 'shepherd', 'ad']
|
|
inlines = [DocAliasInline, DocAuthorInline, ]
|
|
form = DocumentForm
|
|
|
|
def state(self, instance):
|
|
return self.get_state()
|
|
|
|
admin.site.register(Document, DocumentAdmin)
|
|
|
|
class DocHistoryAdmin(admin.ModelAdmin):
|
|
list_display = ['doc', 'rev', 'state', 'group', 'pages', 'intended_std_level', 'author_list', 'time']
|
|
search_fields = ['doc__name']
|
|
ordering = ['time', 'doc', 'rev']
|
|
raw_id_fields = ['doc', 'authors', 'related', 'group', 'shepherd', 'ad']
|
|
|
|
def state(self, instance):
|
|
return self.get_state()
|
|
|
|
admin.site.register(DocHistory, DocHistoryAdmin)
|
|
|
|
class DocAliasAdmin(admin.ModelAdmin):
|
|
list_display = ['name', 'document_link']
|
|
search_fields = ['name', 'document__name']
|
|
raw_id_fields = ['document']
|
|
admin.site.register(DocAlias, DocAliasAdmin)
|
|
|
|
class BallotTypeAdmin(admin.ModelAdmin):
|
|
list_display = ["slug", "doc_type", "name", "question"]
|
|
admin.site.register(BallotType, BallotTypeAdmin)
|
|
|
|
# events
|
|
|
|
class DocEventAdmin(admin.ModelAdmin):
|
|
def rev(obj):
|
|
return obj.doc.rev
|
|
list_display = ["doc", "type", "rev", "by", "time"]
|
|
search_fields = ["doc__name", "by__name"]
|
|
raw_id_fields = ["doc", "by"]
|
|
|
|
admin.site.register(DocEvent, DocEventAdmin)
|
|
|
|
admin.site.register(NewRevisionDocEvent, DocEventAdmin)
|
|
admin.site.register(BallotDocEvent, DocEventAdmin)
|
|
admin.site.register(WriteupDocEvent, DocEventAdmin)
|
|
admin.site.register(LastCallDocEvent, DocEventAdmin)
|
|
admin.site.register(TelechatDocEvent, DocEventAdmin)
|
|
|
|
class BallotPositionDocEventAdmin(DocEventAdmin):
|
|
raw_id_fields = ["doc", "by", "ad", "ballot"]
|
|
|
|
admin.site.register(BallotPositionDocEvent, BallotPositionDocEventAdmin)
|
|
|