FIR/incidents/admin.py at master · security-geeks/FIR · 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
from incidents.models import (
    Incident,
    BusinessLine,
    BaleCategory,
    Comments,
    LabelGroup,
    Label,
    IncidentCategory,
    IncidentStatus,
    Log,
    Profile,
    IncidentTemplate,
    Attribute,
    ValidAttribute,
    SeverityChoice,
    Tlp,
    AccessControlEntry,
)
from incidents.forms import IncidentStatusAdminForm
from django.contrib import admin
from django.contrib.auth import admin as auth_admin, get_user_model
from treebeard.admin import TreeAdmin
from treebeard.forms import movenodeform_factory

User = get_user_model()


class ACENestedAdmin(admin.TabularInline):
    model = AccessControlEntry


class UserAdmin(auth_admin.UserAdmin):
    inlines = [
        ACENestedAdmin,
    ]
    list_display = (
        "username",
        "email",
        "first_name",
        "last_name",
        "is_staff",
        "is_active",
    )


class BusinessLineAdmin(TreeAdmin):
    search_fields = ("name",)
    form = movenodeform_factory(BusinessLine)


class IncidentAdmin(admin.ModelAdmin):
    exclude = ("artifacts",)


class IncidentStatusAdmin(admin.ModelAdmin):
    form = IncidentStatusAdminForm


class LogAdmin(admin.ModelAdmin):
    list_display = (
        "__str__",
        "incident",
        "comment",
        "who",
        "what",
        "when",
    )
    search_fields = (
        "incident__id",
        "incident__subject",
        "comment__id",
        "who__username",
        "when",
        "what",
    )


admin.site.register(Incident, IncidentAdmin)
admin.site.register(BusinessLine, BusinessLineAdmin)
admin.site.register(BaleCategory)
admin.site.register(Comments)
admin.site.register(LabelGroup)
admin.site.register(Label)
admin.site.register(IncidentCategory)
admin.site.register(IncidentStatus, IncidentStatusAdmin)
admin.site.register(Log, LogAdmin)
admin.site.register(Profile)
admin.site.register(IncidentTemplate)
admin.site.register(Attribute)
admin.site.register(ValidAttribute)
admin.site.register(SeverityChoice)
admin.site.register(Tlp)
admin.site.unregister(User)
admin.site.register(User, UserAdmin)