admin.py

import reprlib

from django.contrib import admin
from django.forms import ModelForm, FileField
from django.db.models import QuerySet

from files.service import CDN, SelectelSwiftStorage
from files.models import UserFile


class UserFileForm(ModelForm):
    file = FileField(required=True)  # noqa VNE002

    class Meta:
        model = UserFile
        fields = "__all__"


@admin.register(UserFile)
class UserFileAdmin(admin.ModelAdmin):
    form = UserFileForm
    list_display = (
        "short_link",
        "filename",
        "user",
        "datetime_uploaded",
    )
    list_display_links = (
        "short_link",
        "filename",
        "user",
    )
    search_fields = (
        "link",
        "user__first_name",
        "user__last_name",
    )
    list_filter = (
        "user",
        "datetime_uploaded",
    )

    date_hierarchy = "datetime_uploaded"
    ordering = ("-datetime_uploaded",)

    cdn = CDN(storage=SelectelSwiftStorage())

    @admin.display(empty_value="Empty filename")
    def filename(self, obj):
        return obj.name

    @admin.display(empty_value="Empty link")
    def short_link(self, obj):
        link = obj.link.removeprefix("https://")
        return reprlib.repr(link).removesuffix("'").removeprefix("'")

    def get_fieldsets(self, request, obj=None):
        fieldsets = super().get_fieldsets(request, obj)
        file_field_index = fieldsets[0][1]["fields"].index("file")
        if obj:
            # remove file field if file is already uploaded
            del fieldsets[0][1]["fields"][file_field_index]
        else:
            # remove other field
            fieldsets[0][1]["fields"] = ["file"]
        return fieldsets

    def save_model(self, request, obj, form, change):
        info = self.cdn.upload(request.FILES["file"], request.user, quality=100)
        obj.link = info.url
        obj.user = request.user
        obj.name = info.name
        obj.size = info.size
        obj.extension = info.extension
        obj.mime_type = info.mime_type
        super().save_model(request, obj, form, change)

    def delete_model(self, request, obj):
        self.cdn.delete(obj.link)
        obj.delete()

    def delete_queryset(self, request, queryset):
        for obj in queryset:
            self.cdn.delete(obj.link)
        queryset.delete()

    def get_queryset(self, request) -> QuerySet[UserFile]:
        qs = super().get_queryset(request)
        if request.user.groups.filter(name="Руководитель программы").exists():
            return qs.filter(user=request.user)
        return qs