from django.conf import settings from django.contrib import admin from django.contrib.auth.models import User, Group from django.core.urlresolvers import reverse from django.templatetags.static import static from .models import Correspondent, Tag, Document, Log class MonthListFilter(admin.SimpleListFilter): title = "Month" # Parameter for the filter that will be used in the URL query. parameter_name = "month" def lookups(self, request, model_admin): r = [] for document in Document.objects.all(): r.append(( document.created.strftime("%Y-%m"), document.created.strftime("%B %Y") )) return sorted(set(r), key=lambda x: x[0], reverse=True) def queryset(self, request, queryset): if not self.value(): return None year, month = self.value().split("-") return queryset.filter(created__year=year, created__month=month) class CommonAdmin(admin.ModelAdmin): list_per_page = settings.PAPERLESS_LIST_PER_PAGE class CorrespondentAdmin(CommonAdmin): list_display = ("name", "match", "matching_algorithm") list_filter = ("matching_algorithm",) list_editable = ("match", "matching_algorithm") class TagAdmin(CommonAdmin): list_display = ("name", "colour", "match", "matching_algorithm") list_filter = ("colour", "matching_algorithm") list_editable = ("colour", "match", "matching_algorithm") class DocumentAdmin(CommonAdmin): class Media: css = { "all": ("paperless.css",) } search_fields = ("correspondent__name", "title", "content") list_display = ("title", "created", "thumbnail", "correspondent", "tags_") list_filter = ("tags", "correspondent", MonthListFilter) ordering = ["-created", "correspondent"] def has_add_permission(self, request): return False def created_(self, obj): return obj.created.date().strftime("%Y-%m-%d") created_.short_description = "Created" def thumbnail(self, obj): if settings.FORCE_SCRIPT_NAME: src_link = "{}/fetch/thumb/{}".format(settings.FORCE_SCRIPT_NAME, obj.id) else: src_link = "/fetch/thumb/{}".format(obj.id) png_img = self._html_tag( "img", src=src_link, width=180, alt="Thumbnail of {}".format(obj.file_name), title=obj.file_name ) return self._html_tag("a", png_img, href=obj.download_url) thumbnail.allow_tags = True def tags_(self, obj): r = "" for tag in obj.tags.all(): colour = tag.get_colour_display() r += self._html_tag( "a", tag.slug, **{ "class": "tag", "style": "background-color: {};".format(colour), "href": "{}?tags__id__exact={}".format( reverse("admin:documents_document_changelist"), tag.pk ) } ) return r tags_.allow_tags = True def document(self, obj): return self._html_tag( "a", self._html_tag( "img", src=static("documents/img/{}.png".format(obj.file_type)), width=22, height=22, alt=obj.file_type, title=obj.file_name ), href=obj.download_url ) document.allow_tags = True @staticmethod def _html_tag(kind, inside=None, **kwargs): attributes = [] for lft, rgt in kwargs.items(): attributes.append('{}="{}"'.format(lft, rgt)) if inside is not None: return "<{kind} {attributes}>{inside}".format( kind=kind, attributes=" ".join(attributes), inside=inside) return "<{} {}/>".format(kind, " ".join(attributes)) class LogAdmin(CommonAdmin): list_display = ("created", "message", "level",) list_filter = ("level", "created",) admin.site.register(Correspondent, CorrespondentAdmin) admin.site.register(Tag, TagAdmin) admin.site.register(Document, DocumentAdmin) admin.site.register(Log, LogAdmin) # Unless we implement multi-user, these default registrations don't make sense. admin.site.unregister(Group) admin.site.unregister(User)