mirror of
https://github.com/umap-project/umap.git
synced 2025-04-28 19:42:36 +02:00
feat: Expose teams in users’ CSV export (#2484)
This commit is contained in:
commit
fb03c6a39e
1 changed files with 15 additions and 2 deletions
|
@ -69,9 +69,18 @@ class PictogramAdmin(admin.ModelAdmin):
|
|||
list_filter = ("category",)
|
||||
|
||||
|
||||
class TeamAdmin(admin.ModelAdmin):
|
||||
class TeamAdmin(CSVExportMixin, admin.ModelAdmin):
|
||||
csv_fields = [
|
||||
"pk",
|
||||
"name",
|
||||
"users_count",
|
||||
]
|
||||
list_display = list(admin.ModelAdmin.list_display) + ["users_count"]
|
||||
filter_horizontal = ("users",)
|
||||
|
||||
def users_count(self, obj):
|
||||
return obj.users.count()
|
||||
|
||||
|
||||
class UserAdmin(CSVExportMixin, UserAdminBase):
|
||||
csv_fields = [
|
||||
|
@ -83,13 +92,17 @@ class UserAdmin(CSVExportMixin, UserAdminBase):
|
|||
"last_login",
|
||||
"date_joined",
|
||||
"maps_count",
|
||||
"user_teams",
|
||||
]
|
||||
list_display = list(UserAdminBase.list_display) + ["maps_count"]
|
||||
list_display = list(UserAdminBase.list_display) + ["maps_count", "user_teams"]
|
||||
|
||||
def maps_count(self, obj):
|
||||
# owner maps + maps as editor
|
||||
return obj.owned_maps.count() + obj.map_set.count()
|
||||
|
||||
def user_teams(self, obj):
|
||||
return " ; ".join(obj.teams.values_list("name", flat=True))
|
||||
|
||||
|
||||
admin.site.register(Map, MapAdmin)
|
||||
admin.site.register(DataLayer)
|
||||
|
|
Loading…
Reference in a new issue