|
1 | 1 | #coding: utf-8 |
| 2 | +from django import forms |
2 | 3 | from django.contrib import admin |
3 | | -from ietf.ipr.models import IprContact, IprDetail, IprDocAlias, IprNotification, IprUpdate |
4 | | - |
5 | | -class IprContactAdmin(admin.ModelAdmin): |
6 | | - list_display=('__str__', 'ipr') |
7 | | -admin.site.register(IprContact, IprContactAdmin) |
8 | | - |
9 | | -class IprDetailAdmin(admin.ModelAdmin): |
10 | | - list_display = ['title', 'submitted_date', 'docs', 'status'] |
| 4 | +from ietf.name.models import DocRelationshipName |
| 5 | +from ietf.ipr.models import (IprNotification, IprDisclosureBase, IprDocRel, IprEvent, |
| 6 | + RelatedIpr, HolderIprDisclosure, ThirdPartyIprDisclosure, GenericIprDisclosure, |
| 7 | + NonDocSpecificIprDisclosure) |
| 8 | + |
| 9 | +# ------------------------------------------------------ |
| 10 | +# ModelAdmins |
| 11 | +# ------------------------------------------------------ |
| 12 | +class IprDocRelAdminForm(forms.ModelForm): |
| 13 | + class Meta: |
| 14 | + model = IprDocRel |
| 15 | + widgets = { |
| 16 | + 'sections':forms.TextInput, |
| 17 | + } |
| 18 | + |
| 19 | +class IprDocRelInline(admin.TabularInline): |
| 20 | + model = IprDocRel |
| 21 | + form = IprDocRelAdminForm |
| 22 | + raw_id_fields = ['document'] |
| 23 | + extra = 1 |
| 24 | + |
| 25 | +class RelatedIprInline(admin.TabularInline): |
| 26 | + model = RelatedIpr |
| 27 | + raw_id_fields = ['target'] |
| 28 | + fk_name = 'source' |
| 29 | + extra = 1 |
| 30 | + |
| 31 | + def formfield_for_foreignkey(self, db_field, request, **kwargs): |
| 32 | + if db_field.name == "relationship": |
| 33 | + kwargs["queryset"] = DocRelationshipName.objects.filter(slug='updates') |
| 34 | + return super(RelatedIprInline, self).formfield_for_foreignkey(db_field, request, **kwargs) |
| 35 | + |
| 36 | +class IprDisclosureBaseAdmin(admin.ModelAdmin): |
| 37 | + list_display = ['title', 'time', 'related_docs', 'state'] |
11 | 38 | search_fields = ['title', 'legal_name'] |
| 39 | + raw_id_fields = ['by'] |
| 40 | + inlines = [IprDocRelInline,RelatedIprInline] |
| 41 | + |
| 42 | + def related_docs(self, obj): |
| 43 | + return u", ".join(a.formatted_name() for a in IprDocRel.objects.filter(disclosure=obj).order_by("id").select_related("document")) |
| 44 | + |
| 45 | +admin.site.register(IprDisclosureBase, IprDisclosureBaseAdmin) |
| 46 | + |
| 47 | +class HolderIprDisclosureAdmin(admin.ModelAdmin): |
| 48 | + list_display = ['title', 'time', 'related_docs', 'state'] |
| 49 | + raw_id_fields = ["by"] |
| 50 | + inlines = [IprDocRelInline,RelatedIprInline] |
| 51 | + |
| 52 | + def related_docs(self, obj): |
| 53 | + return u", ".join(a.formatted_name() for a in IprDocRel.objects.filter(disclosure=obj).order_by("id").select_related("document")) |
| 54 | + |
| 55 | +admin.site.register(HolderIprDisclosure, HolderIprDisclosureAdmin) |
12 | 56 |
|
13 | | - def docs(self, ipr): |
14 | | - return u", ".join(a.formatted_name() for a in IprDocAlias.objects.filter(ipr=ipr).order_by("id").select_related("doc_alias")) |
| 57 | +class ThirdPartyIprDisclosureAdmin(admin.ModelAdmin): |
| 58 | + list_display = ['title', 'time', 'related_docs', 'state'] |
| 59 | + raw_id_fields = ["by"] |
| 60 | + inlines = [IprDocRelInline,RelatedIprInline] |
| 61 | + |
| 62 | + def related_docs(self, obj): |
| 63 | + return u", ".join(a.formatted_name() for a in IprDocRel.objects.filter(disclosure=obj).order_by("id").select_related("document")) |
15 | 64 |
|
16 | | -admin.site.register(IprDetail, IprDetailAdmin) |
| 65 | +admin.site.register(ThirdPartyIprDisclosure, ThirdPartyIprDisclosureAdmin) |
| 66 | + |
| 67 | +class GenericIprDisclosureAdmin(admin.ModelAdmin): |
| 68 | + list_display = ['title', 'time', 'related_docs', 'state'] |
| 69 | + raw_id_fields = ["by"] |
| 70 | + inlines = [RelatedIprInline] |
| 71 | + |
| 72 | + def related_docs(self, obj): |
| 73 | + return u", ".join(a.formatted_name() for a in IprDocRel.objects.filter(disclosure=obj).order_by("id").select_related("document")) |
| 74 | + |
| 75 | +admin.site.register(GenericIprDisclosure, GenericIprDisclosureAdmin) |
| 76 | + |
| 77 | +class NonDocSpecificIprDisclosureAdmin(admin.ModelAdmin): |
| 78 | + list_display = ['title', 'time', 'related_docs', 'state'] |
| 79 | + raw_id_fields = ["by"] |
| 80 | + inlines = [RelatedIprInline] |
| 81 | + |
| 82 | + def related_docs(self, obj): |
| 83 | + return u", ".join(a.formatted_name() for a in IprDocRel.objects.filter(disclosure=obj).order_by("id").select_related("document")) |
| 84 | + |
| 85 | +admin.site.register(NonDocSpecificIprDisclosure, NonDocSpecificIprDisclosureAdmin) |
17 | 86 |
|
18 | 87 | class IprNotificationAdmin(admin.ModelAdmin): |
19 | 88 | pass |
20 | 89 | admin.site.register(IprNotification, IprNotificationAdmin) |
21 | 90 |
|
22 | | -class IprUpdateAdmin(admin.ModelAdmin): |
23 | | - pass |
24 | | -admin.site.register(IprUpdate, IprUpdateAdmin) |
| 91 | +class IprDocRelAdmin(admin.ModelAdmin): |
| 92 | + raw_id_fields = ["disclosure", "document"] |
| 93 | +admin.site.register(IprDocRel, IprDocRelAdmin) |
| 94 | + |
| 95 | +class RelatedIprAdmin(admin.ModelAdmin): |
| 96 | + list_display = ['source', 'target', 'relationship', ] |
| 97 | + search_fields = ['source__name', 'target__name', 'target__document__name', ] |
| 98 | + raw_id_fields = ['source', 'target', ] |
| 99 | +admin.site.register(RelatedIpr, RelatedIprAdmin) |
25 | 100 |
|
26 | | -class IprDocAliasAdmin(admin.ModelAdmin): |
27 | | - raw_id_fields = ["ipr", "doc_alias"] |
28 | | -admin.site.register(IprDocAlias, IprDocAliasAdmin) |
| 101 | +class IprEventAdmin(admin.ModelAdmin): |
| 102 | + list_display = ["disclosure", "type", "by", "time"] |
| 103 | + list_filter = ["time", "type"] |
| 104 | + search_fields = ["disclosure__title", "by__name"] |
| 105 | + raw_id_fields = ["disclosure", "by", "message", "in_reply_to"] |
| 106 | +admin.site.register(IprEvent, IprEventAdmin) |
0 commit comments