-
Notifications
You must be signed in to change notification settings - Fork 655
/
admin.py
101 lines (73 loc) · 2.88 KB
/
admin.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
from datetime import datetime
from typing import Any, List, Optional, TypeVar
from django.contrib import admin
from django.contrib.auth.models import AbstractBaseUser
from django.db.models import QuerySet
from django.utils.translation import gettext_lazy as _
from rest_framework.request import Request
from ..models import TokenUser
from .models import BlacklistedToken, OutstandingToken
AuthUser = TypeVar("AuthUser", AbstractBaseUser, TokenUser)
class OutstandingTokenAdmin(admin.ModelAdmin):
list_display = (
"jti",
"user",
"created_at",
"expires_at",
)
search_fields = (
"user__id",
"jti",
)
ordering = ("user",)
def get_queryset(self, *args, **kwargs) -> QuerySet:
qs = super().get_queryset(*args, **kwargs)
return qs.select_related("user")
# Read-only behavior defined below
actions = None
def get_readonly_fields(self, *args, **kwargs) -> List[Any]:
return [f.name for f in self.model._meta.fields]
def has_add_permission(self, *args, **kwargs) -> bool:
return False
def has_delete_permission(self, *args, **kwargs) -> bool:
return False
def has_change_permission(
self, request: Request, obj: Optional[object] = None
) -> bool:
return request.method in ["GET", "HEAD"] and super().has_change_permission(
request, obj
)
admin.site.register(OutstandingToken, OutstandingTokenAdmin)
class BlacklistedTokenAdmin(admin.ModelAdmin):
list_display = (
"token_jti",
"token_user",
"token_created_at",
"token_expires_at",
"blacklisted_at",
)
search_fields = (
"token__user__id",
"token__jti",
)
ordering = ("token__user",)
def get_queryset(self, *args, **kwargs) -> QuerySet:
qs = super().get_queryset(*args, **kwargs)
return qs.select_related("token__user")
def token_jti(self, obj: BlacklistedToken) -> str:
return obj.token.jti
token_jti.short_description = _("jti") # type: ignore
token_jti.admin_order_field = "token__jti" # type: ignore
def token_user(self, obj: BlacklistedToken) -> AuthUser:
return obj.token.user
token_user.short_description = _("user") # type: ignore
token_user.admin_order_field = "token__user" # type: ignore
def token_created_at(self, obj: BlacklistedToken) -> datetime:
return obj.token.created_at
token_created_at.short_description = _("created at") # type: ignore
token_created_at.admin_order_field = "token__created_at" # type: ignore
def token_expires_at(self, obj: BlacklistedToken) -> datetime:
return obj.token.expires_at
token_expires_at.short_description = _("expires at") # type: ignore
token_expires_at.admin_order_field = "token__expires_at" # type: ignore
admin.site.register(BlacklistedToken, BlacklistedTokenAdmin)