ArchiveBox/archivebox/core/admin.py
2020-07-02 03:13:35 -04:00

80 lines
2.9 KiB
Python

from django.contrib import admin
from django.utils.html import format_html
from util import htmldecode, urldecode
from core.models import Snapshot
from cli.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>'
'<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)