upgrades to admin: search_fields, list_filters and raw_id_field

Author: PetrDlouhy

oauth2_provider/admin.py

@@ -1,4 +1,5 @@
 from django.contrib import admin
+from django.contrib.auth import get_user_model
 
 from oauth2_provider.models import (
     get_access_token_admin_class,
@@ -13,6 +14,8 @@
     get_refresh_token_model,
 )
 
+has_email = hasattr(get_user_model(), 'email')
+
 
 class ApplicationAdmin(admin.ModelAdmin):
     list_display = ("id", "name", "user", "client_type", "authorization_grant_type")
@@ -28,6 +31,8 @@ class AccessTokenAdmin(admin.ModelAdmin):
     list_display = ("token", "user", "application", "expires")
     list_select_related = ("application", "user")
     raw_id_fields = ("user", "source_refresh_token")
+    search_fields = ("token",) + (("user__email",) if has_email else ())
+    list_filter = ("application",)
 
 
 class GrantAdmin(admin.ModelAdmin):
@@ -37,12 +42,18 @@ class GrantAdmin(admin.ModelAdmin):
 
 class IDTokenAdmin(admin.ModelAdmin):
     list_display = ("jti", "user", "application", "expires")
-    raw_id_fields = ("user",)
+    raw_id_fields = ("user", )
+    search_fields = ("token",) + (("user__email",) if has_email else ())
+    list_filter = ("application",)
+
+
 
 
 class RefreshTokenAdmin(admin.ModelAdmin):
     list_display = ("token", "user", "application")
     raw_id_fields = ("user", "access_token")
+    search_fields = ("token",) + (("user__email",) if has_email else ())
+    list_filter = ("application",)
 
 
 application_model = get_application_model()