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

Browse source 
maps are owner maps + maps as editor

cf https://github.com/umap-project/umap/pull/2131#discussion_r1772775039
This commit is contained in:
Yohan Boniface 2024-09-24 11:07:47 +02:00
parent a1b35ddff0
commit 3719e88643
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}"'},
)
def get_cell(user, field):
if hasattr(self, field):
return getattr(self, field)(user)
return getattr(user, field)
writer = csv.writer(response)
writer.writerow(self.csv_fields)
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
@ -77,7 +82,13 @@ class UserAdmin(CSVExportMixin, UserAdminBase):
"last_name",
"last_login",
"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)