from django.contrib import admin
from django.template.defaultfilters import filesizeformat

from .models import Torrent


@admin.register(Torrent)
class TorrentAdmin(admin.ModelAdmin):
    list_display = ["name", "user", "display_size", "len_files"]
    list_filter = ["user"]

    def display_size(self, obj):
        return filesizeformat(obj.size)