archival.ArchiveBox/archivebox/core/admin.py

from django.contrib import admin
from django.utils.html import format_html

from util import htmldecode, urldecode
from core.models import Snapshot
from archivebox.logging import printable_filesize

# TODO: https://stackoverflow.com/questions/40760880/add-custom-button-to-django-admin-panel


class SnapshotAdmin(admin.ModelAdmin):
    list_display = ('title_str', 'url_str', 'tags', 'files', 'added', 'updated')
    sort_fields = ('title_str', 'url_str', 'tags', 'added', 'updated')
    readonly_fields = ('id', 'num_outputs', 'is_archived', 'url_hash', 'added', 'updated')
    search_fields = ('url', 'timestamp', 'title', 'tags')
    fields = ('url', 'timestamp', 'title', 'tags', *readonly_fields)
    list_filter = ('added', 'updated', 'tags')
    ordering = ['-added']

    def id_str(self, obj):
        return format_html(
            '<code style="font-size: 10px">{}</code>',
            obj.url_hash[:8],
        )

    def title_str(self, obj):
        canon = obj.as_link().canonical_outputs()
        return format_html(
            '<a href="/{}">'
            '<img src="/{}/{}" style="height: 20px; width: 20px;" onerror="this.remove()">'
            ' &nbsp; &nbsp; '
            '</a>'
            '<a href="/{}/{}">'
            '<b>{}</b></a>',
            obj.archive_path,
            obj.archive_path, canon['google_favicon_path'],
            obj.archive_path, canon['wget_path'] or '',
            urldecode(htmldecode(obj.latest_title or obj.title or '-'))[:128],
        )

    def files(self, obj):
        canon = obj.as_link().canonical_outputs()
        return format_html(
            '<span style="font-size: 1.2em; opacity: 0.8">'
            '<a href="/{}/{}">🌐 </a> '
            '<a href="/{}/{}">📄</a> '
            '<a href="/{}/{}">🖥 </a> '
            '<a href="/{}/{}">🅷 </a> '
            '<a href="/{}/{}">📼 </a> '
            '<a href="/{}/{}">📦 </a> '
            '<a href="/{}/{}">🏛 </a> '
            '</span><br/>'
            '<a href="/{}">{}</a>',
            obj.archive_path, canon['wget_path'] or '',
            obj.archive_path, canon['pdf_path'],
            obj.archive_path, canon['screenshot_path'],
            obj.archive_path, canon['dom_path'],
            obj.archive_path, canon['media_path'],
            obj.archive_path, canon['git_path'],
            obj.archive_path, canon['archive_org_path'],
            obj.archive_path,
            printable_filesize(obj.archive_size) if obj.archive_size else 'pending',
        )

    def url_str(self, obj):
        return format_html(
            '<a href="{}"><code>{}</code></a>',
            obj.url,
            obj.url.split('://www.', 1)[-1].split('://', 1)[-1][:64],
        )

    id_str.short_description = 'ID'
    title_str.short_description = 'Title'
    url_str.short_description = 'URL'

    id_str.admin_order_field = 'id'
    title_str.admin_order_field = 'title'
    url_str.admin_order_field = 'url'

admin.site.register(Snapshot, SnapshotAdmin)