Skip to content

Commit

Permalink
PB-1350 Add filters for the Asset view in Django
Browse files Browse the repository at this point in the history
Add filters that help find assets that are either in progress of
uploading or that are missing on s3
  • Loading branch information
schtibe committed Jan 28, 2025
1 parent 13b6553 commit 772d932
Showing 1 changed file with 25 additions and 2 deletions.
27 changes: 25 additions & 2 deletions app/stac_api/admin.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,6 +6,7 @@

from django import forms
from django.contrib import messages
from django.contrib.admin import SimpleListFilter
from django.contrib.gis import admin
from django.contrib.gis.db import models
from django.contrib.postgres.fields import ArrayField
Expand Down Expand Up @@ -516,6 +517,25 @@ def clean_file(self):
return self.cleaned_data.get('file')


class NotUploadedYetFilter(SimpleListFilter):
"""Provide a filter for assets that helps with sorting out invalid items"""

title = 'Not uploaded yet'
parameter_name = 'file_size'

def lookups(self, request, model_admin):
return [
('0', _('Upload in progress')), ('none', _('Not found on s3'))
]

def queryset(self, request, queryset):
if self.value() == '0':
return queryset.filter(file_size=0)
if self.value() == 'none':
return queryset.filter(file_size=None)
return queryset


@admin.register(Asset)
class AssetAdmin(admin.ModelAdmin):
form = AssetAdminForm
Expand Down Expand Up @@ -543,11 +563,14 @@ class Media:
'update_interval',
'displayed_file_size'
]
list_display = ['name', 'item_name', 'collection_name', 'collection_published']
list_display = [
'name', 'item_name', 'collection_name', 'collection_published', 'created', 'updated'
]

list_filter = [
AutocompleteFilterFactory('Item name', 'item', use_pk_exact=True),
AutocompleteFilterFactory('Collection name', 'item__collection', use_pk_exact=True)
AutocompleteFilterFactory('Collection name', 'item__collection', use_pk_exact=True),
NotUploadedYetFilter
]

def get_search_results(self, request, queryset, search_term):
Expand Down

0 comments on commit 772d932

Please sign in to comment.