Blame view

src/core/pyros_django/common/admin.py 6.68 KB
5ce2836f   Alexis Koralewski   update models (ad...
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
from django import forms
from django.contrib import admin
from django.contrib.auth.models import User

# EP
from django.conf import settings

from common.models import *


# EP added
class ReadOnlyModelAdmin(admin.ModelAdmin):
    """ModelAdmin class that prevents modifications through the admin.
    The changelist and the detail view work, but a 403 is returned
    if one actually tries to edit an object.
    Source: https://gist.github.com/aaugustin/1388243
    """

    actions = None

    def get_readonly_fields(self, request, obj=None):
        return self.fields or [f.name for f in self.model._meta.fields]

    def has_add_permission(self, request):
        return False

    # Allow viewing objects but not actually changing them
    def has_change_permission(self, request, obj=None):
        if request.method not in ('GET', 'HEAD'):
            return False
        return super(ReadOnlyModelAdmin, self).has_change_permission(request, obj)

    def has_delete_permission(self, request, obj=None):
        return False


# EP added

# Edit mode
# DEBUG = False
# View only mode
# DEBUG = True

''' Uncomment for production '''

# if settings.DEBUG:
# class PyrosModelAdmin(ReadOnlyModelAdmin):
#     pass
# else:
#     class PyrosModelAdmin(admin.ModelAdmin):
#         pass


class PyrosModelAdmin(admin.ModelAdmin):
    pass

# Many To Many interface adapter


class PyrosUserAndSPInline(admin.TabularInline):
    #model = ScientificProgram.pyros_users.through
    pass


class SequenceAndScheduleInline(admin.TabularInline):
    model = Schedule.sequences.through

class PyrosUserAndUserLevelInline(admin.TabularInline):
    # add admin representation for m2m relation between PyrosUser and UserLevel
    model = UserLevel.pyros_users.through

class ScheduleAdmin(admin.ModelAdmin):
    inlines = [
        SequenceAndScheduleInline,
    ]
    exclude = ('sequences',)


# One To Many interface adapters

class SequenceInline(admin.TabularInline):
    model = Sequence
    readonly_fields = ("name",)
    fields = ("name",)
    show_change_link = True


class RequestInline(admin.TabularInline):
    model = Request
    readonly_fields = ("name",)
    fields = ("name",)
    show_change_link = True


class AlbumInline(admin.TabularInline):
    model = Album
    readonly_fields = ("name",)
    fields = ("name",)
    show_change_link = True


class PlanInline(admin.TabularInline):
    model = Plan
3b81a22b   Alexis Koralewski   Rework on Request...
104
105
    #readonly_fields = ("name",)
    #fields = ("name",)
5ce2836f   Alexis Koralewski   update models (ad...
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
136
137
138
139
140
141
142
143
144
145
146
    show_change_link = True


class ImageInline(admin.TabularInline):
    model = Image
    readonly_fields = ("name",)
    fields = ("name",)
    show_change_link = True


class DetectorInline(admin.TabularInline):
    model = Detector
    readonly_fields = ("device_name",)
    fields = ("device_name",)
    show_change_link = True


class PyrosUserInline(admin.TabularInline):
    model = PyrosUser
    readonly_fields = ("user_username",)
    fields = ("user_username",)
    show_change_link = True


class FilterInline(admin.TabularInline):
    model = Filter
    readonly_fields = ("device_name",)
    fields = ("device_name",)
    show_change_link = True


class AlertInline(admin.TabularInline):
    model = Alert
    readonly_fields = ("request_name",)
    fields = ("request_name",)
    show_change_link = True


# Admin model classes

class RequestAdmin(PyrosModelAdmin):
3b81a22b   Alexis Koralewski   Rework on Request...
147
148
149
150
    pass
    # inlines = [
    #     SequenceInline,
    # ]
5ce2836f   Alexis Koralewski   update models (ad...
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202


class SequenceAdmin(PyrosModelAdmin):
    inlines = [
        AlbumInline,
        SequenceAndScheduleInline,  # for M2M interface
    ]


class PyrosUserAdmin(PyrosModelAdmin):
    list_display = ("user_username","is_active","laboratory")
    list_filter = ("is_active",)
    list_editable = ("is_active",)
    inlines = [
        RequestInline,
        # A user has many SPs
#        PyrosUserAndSPInline,  # for M2M interface   
    ]


class StrategyObsAdmin(PyrosModelAdmin):
    inlines = [
        AlertInline,
    ]


class ScientificProgramAdmin(PyrosModelAdmin):
    inlines = [
        RequestInline,
        # A SP has many users:
    #    PyrosUserAndSPInline,  # for M2M interface
    ]
    exclude = ('pyros_users',)  # for M2M interface


class CountryAdmin(PyrosModelAdmin):
    inlines = [
        PyrosUserInline,
    ]


class UserLevelAdmin(PyrosModelAdmin):
    inlines = [
        #PyrosUserInline,
        PyrosUserAndUserLevelInline,
    ]
    list_display = ("name","priority",)
    # we need to exclude pyros_users which represents the m2m relation between UserLevel and PyrosUser
    exclude = ("pyros_users",)


class FilterAdmin(PyrosModelAdmin):
3b81a22b   Alexis Koralewski   Rework on Request...
203
204
205
206
    # inlines = [
    #     PlanInline,
    # ]
    pass
5ce2836f   Alexis Koralewski   update models (ad...
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221


class FilterWheelAdmin(PyrosModelAdmin):
    inlines = [
        FilterInline,
    ]


class NrtAnalysisAdmin(PyrosModelAdmin):
    inlines = [
        ImageInline,
    ]


class DetectorAdmin(PyrosModelAdmin):
3b81a22b   Alexis Koralewski   Rework on Request...
222
223
224
225
    pass
    # inlines = [
    #     AlbumInline,
    # ]
5ce2836f   Alexis Koralewski   update models (ad...
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284


class TelescopeAdmin(PyrosModelAdmin):
    inlines = [
        DetectorInline,
    ]


class PlanAdmin(PyrosModelAdmin):
    inlines = [
        ImageInline,
    ]


# class AlbumAdmin(admin.ModelAdmin):
class AlbumAdmin(PyrosModelAdmin):
    inlines = [
        PlanInline,
    ]


# Link the models to the admin interface

# (EP added 10/7/19)
admin.site.register(AgentCmd)
admin.site.register(AgentLogs)
admin.site.register(AgentSurvey)
admin.site.register(AgentDeviceStatus)


admin.site.register(Album, AlbumAdmin)
admin.site.register(Alert)
admin.site.register(Country, CountryAdmin)
admin.site.register(Detector, DetectorAdmin)
admin.site.register(Filter, FilterAdmin)
admin.site.register(FilterWheel, FilterWheelAdmin)
admin.site.register(Image)
admin.site.register(Log)
admin.site.register(NrtAnalysis, NrtAnalysisAdmin)
admin.site.register(Plan, PlanAdmin)
admin.site.register(Request, RequestAdmin)
admin.site.register(Schedule, ScheduleAdmin)
admin.site.register(ScheduleHasSequences)
admin.site.register(ScientificProgram, ScientificProgramAdmin)
admin.site.register(Sequence, SequenceAdmin)
admin.site.register(SiteWatch)
admin.site.register(SiteWatchHistory)
admin.site.register(StrategyObs, StrategyObsAdmin)
admin.site.register(TaskId)
admin.site.register(Telescope, TelescopeAdmin)
admin.site.register(PyrosUser, PyrosUserAdmin)
admin.site.register(UserLevel, UserLevelAdmin)
admin.site.register(Version)
admin.site.register(WeatherWatch)
admin.site.register(WeatherWatchHistory)
admin.site.register(PlcDeviceStatus)
admin.site.register(PlcDevice)
admin.site.register(Config)
admin.site.register(Institute)