internal-bot/intbot/core/admin.py at 1a52814861952e9eb9bf3188c115a2962b8fb164 · EuroPython/internal-bot · GitHub

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
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
import json

from core.models import DiscordMessage, PretalxData, PretixData, Webhook
from django.contrib import admin
from django.utils.html import format_html


class WebhookAdmin(admin.ModelAdmin):
    list_display = [
        "uuid",
        "source",
        "event",
        "created_at",
        "modified_at",
    ]
    list_filter = ["created_at"]
    readonly_fields = fields = [
        "uuid",
        "source",
        "event",
        "signature",
        "pretty_meta",
        "pretty_content",
        "created_at",
        "modified_at",
        "processed_at",
    ]

    def pretty_meta(self, obj: Webhook):
        return format_html("<pre>{}</pre>", json.dumps(obj.meta, indent=4))

    pretty_meta.short_description = "Meta"

    def pretty_content(self, obj: Webhook):
        return format_html("<pre>{}</pre>", json.dumps(obj.content, indent=4))

    pretty_content.short_description = "Content"


class DiscordMessageAdmin(admin.ModelAdmin):
    list_display = [
        "uuid",
        "channel_name",
        "content_short",
        "channel_id",
        "created_at",
        "modified_at",
        "sent_at",
    ]
    list_filter = [
        "created_at",
        "sent_at",
        "channel_name",
    ]
    readonly_fields = fields = [
        "uuid",
        "channel_id",
        "content",
        "created_at",
        "modified_at",
        "sent_at",
    ]

    def content_short(self, obj: DiscordMessage):
        # NOTE(artcz) This can create false shortcuts, but for most messages is
        # good enough, because most of them are longer than 20 chars
        return f"{obj.content[:10]}...{obj.content[-10:]}"


class PretalxDataAdmin(admin.ModelAdmin):
    list_display = [
        "uuid",
        "resource",
        "created_at",
        "modified_at",
    ]
    list_filter = [
        "created_at",
        "resource",
    ]
    readonly_fields = fields = [
        "uuid",
        "resource",
        "pretty_content",
        "created_at",
        "modified_at",
        "processed_at",
    ]

    def get_queryset(self, request):
        qs = super().get_queryset(request)
        qs = qs.defer("content")
        return qs

    def pretty_content(self, obj: PretalxData):
        return format_html("<pre>{}</pre>", json.dumps(obj.content, indent=4))

    pretty_content.short_description = "Content"


class PretixDataAdmin(admin.ModelAdmin):
    list_display = [
        "uuid",
        "resource",
        "created_at",
        "modified_at",
    ]
    list_filter = [
        "created_at",
        "resource",
    ]
    readonly_fields = fields = [
        "uuid",
        "resource",
        "pretty_content",
        "created_at",
        "modified_at",
        "processed_at",
    ]

    def get_queryset(self, request):
        qs = super().get_queryset(request)
        qs = qs.defer("content")
        return qs

    def pretty_content(self, obj: PretixData):
        return format_html("<pre>{}</pre>", json.dumps(obj.content, indent=4))

    pretty_content.short_description = "Content"


admin.site.register(Webhook, WebhookAdmin)
admin.site.register(DiscordMessage, DiscordMessageAdmin)
admin.site.register(PretalxData, PretalxDataAdmin)
admin.site.register(PretixData, PretixDataAdmin)