From fc782d9354b3e05d92bd2c209625588c178e8fb3 Mon Sep 17 00:00:00 2001
From: Sean Morley <zipsm15@gmail.com>
Date: Sun, 21 Jul 2024 09:03:25 -0400
Subject: [PATCH] chore: Register CollectionAdmin in admin.py

---
 backend/server/adventures/admin.py | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/backend/server/adventures/admin.py b/backend/server/adventures/admin.py
index 9291219..c736aa7 100644
--- a/backend/server/adventures/admin.py
+++ b/backend/server/adventures/admin.py
@@ -56,16 +56,24 @@ class CustomUserAdmin(UserAdmin):
             return mark_safe(f'<img src="{public_url}/media/{obj.profile_pic.name}" width="100px" height="100px"')
         else:
             return
+        
+class CollectionAdmin(admin.ModelAdmin):
+    def adventure_count(self, obj):
+        return obj.adventure_set.count()
+
+    adventure_count.short_description = 'Adventure Count'
+
+    list_display = ('name', 'user_id', 'adventure_count', 'is_public')
 
 admin.site.register(CustomUser, CustomUserAdmin)
 
 
 
 admin.site.register(Adventure, AdventureAdmin)
+admin.site.register(Collection, CollectionAdmin)
 admin.site.register(Country, CountryAdmin)
 admin.site.register(Region, RegionAdmin)
 admin.site.register(VisitedRegion)
-admin.site.register(Collection)
 
 admin.site.site_header = 'AdventureLog Admin'
 admin.site.site_title = 'AdventureLog Admin Site'