11from django .contrib import admin
2- from .models import Chat , Message , Profile , Bot , UserAccount , UsageLimitHit , AiModel , Device , RevenueCatWebhookEvent
2+ from .models import Chat , Message , Profile , Bot , UserAccount , UsageLimitHit , AiModel , Device , RevenueCatWebhookEvent , Deck , Flashcard
33from django .contrib .auth .models import User
44from django .contrib .auth .admin import UserAdmin as BaseUserAdmin
55
@@ -61,6 +61,20 @@ def get_readonly_fields(self, request, obj=None):
6161 def get_list_display (self , request ):
6262 return ['created_at' , 'raw_event' ] + list (super ().get_list_display (request ))
6363
64+ class DeckAdmin (admin .ModelAdmin ):
65+ def get_readonly_fields (self , request , obj = None ):
66+ return ['deck_id' , 'created_at' , 'updated_at' ]
67+
68+ def get_list_display (self , request ):
69+ return ['deck_id' , 'name' , 'profile' , 'chat' , 'card_count' , 'created_at' , 'updated_at' ] + list (super ().get_list_display (request ))
70+
71+ class FlashcardAdmin (admin .ModelAdmin ):
72+ def get_readonly_fields (self , request , obj = None ):
73+ return ['flashcard_id' , 'created_at' , 'updated_at' ]
74+
75+ def get_list_display (self , request ):
76+ return ['flashcard_id' , 'deck' , 'front' , 'order' , 'created_at' , 'updated_at' ] + list (super ().get_list_display (request ))
77+
6478class UserAccountAdmin (admin .ModelAdmin ):
6579 def get_list_display (self , request ):
6680 return ['user_id' , 'pin' , 'subscription_level' , 'timezone' ] + list (super ().get_list_display (request ))
@@ -78,5 +92,7 @@ class UserAdmin(BaseUserAdmin):
7892admin .site .register (AiModel , AiModelAdmin )
7993admin .site .register (UsageLimitHit , UsageLimitHitAdmin )
8094admin .site .register (RevenueCatWebhookEvent , RevenueCatWebhookEventAdmin )
95+ admin .site .register (Deck , DeckAdmin )
96+ admin .site .register (Flashcard , FlashcardAdmin )
8197admin .site .unregister (User )
8298admin .site .register (User , UserAdmin )
0 commit comments