almet/umap
1
0
Fork 0
mirror of https://github.com/umap-project/umap.git synced 2025-05-01 12:32:23 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

chore: display user maps in admin and add in csv export (#2169)

Browse source 
maps are owned maps + maps as editor

cf https://github.com/umap-project/umap/pull/2131#discussion_r1772775039
This commit is contained in:
Yohan Boniface 2024-10-04 16:42:53 +02:00 committed by GitHub
commit 4b4546991a
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 12 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
umap/admin.py
View file

@ -22,10 +22,15 @@ class CSVExportMixin:
headers={"Content-Disposition": f'attachment; filename="{filename}"'}, headers={"Content-Disposition": f'attachment; filename="{filename}"'},
) )
def get_cell(user, field):
if hasattr(self, field):
return getattr(self, field)(user)
return getattr(user, field)
writer = csv.writer(response) writer = csv.writer(response)
writer.writerow(self.csv_fields) writer.writerow(self.csv_fields)
for user in queryset: for user in queryset:
writer.writerow(getattr(user, field) for field in self.csv_fields) writer.writerow(get_cell(user, field) for field in self.csv_fields)
return response return response
@ -77,7 +82,13 @@ class UserAdmin(CSVExportMixin, UserAdminBase):
"last_name", "last_name",
"last_login", "last_login",
"date_joined", "date_joined",
"maps_count",
] ]
list_display = list(UserAdminBase.list_display) + ["maps_count"]
def maps_count(self, obj):
# owner maps + maps as editor
return obj.owned_maps.count() + obj.map_set.count()
admin.site.register(Map, MapAdmin) admin.site.register(Map, MapAdmin)